use char not a long, since we're telling the SM that its 1 byte lone
[dana/openbox.git] / openbox / session.c
1 /* This session code is largely inspired by metacity code. */
2
3 #ifndef USE_SM
4
5 void session_load(char *path) {}
6 void session_startup(int argc, char **argv) {}
7 void session_shutdown() {}
8
9 #else
10
11 #include "debug.h"
12 #include "openbox.h"
13 #include "session.h"
14 #include "client.h"
15 #include "prop.h"
16 #include "parser/parse.h"
17
18 #include <time.h>
19 #include <errno.h>
20 #include <stdio.h>
21
22 #ifdef HAVE_UNISTD_H
23 #  include <sys/types.h>
24 #  include <unistd.h>
25 #endif
26
27 #include <X11/SM/SMlib.h>
28
29 static SmcConn     sm_conn;
30 static gchar      *save_file;
31 static gint        sm_argc;
32 static gchar     **sm_argv;
33
34 static gboolean session_save();
35
36 static void sm_save_yourself(SmcConn conn, SmPointer data, int save_type,
37                              Bool shutdown, int interact_style, Bool fast);
38 static void sm_die(SmcConn conn, SmPointer data);
39 static void sm_save_complete(SmcConn conn, SmPointer data);
40 static void sm_shutdown_cancelled(SmcConn conn, SmPointer data);
41
42 static void save_commands()
43 {
44     SmProp *props[2];
45     SmProp prop_cmd = { SmCloneCommand, SmLISTofARRAY8, 1, };
46     SmProp prop_res = { SmRestartCommand, SmLISTofARRAY8, };
47     gint i, j, n;
48     gboolean has_id = FALSE, has_file = FALSE;
49
50     for (i = 1; !has_id && !has_file && i < sm_argc - 1; ++i) {
51         if (!has_id && strcmp(sm_argv[i], "--sm-client-id") == 0)
52             has_id = TRUE;
53         if (!has_file && strcmp(sm_argv[i], "--sm-save-file") == 0)
54             has_file = TRUE;
55     }
56
57     n = (has_file ? sm_argc-2 : sm_argc);
58     n = (has_id ? n-2 : n);
59     prop_cmd.vals = g_new(SmPropValue, n);
60     prop_cmd.num_vals = n;
61     for (i = 0, j = 0; i < sm_argc; ++i, ++j) {
62         if (strcmp (sm_argv[i], "--sm-client-id") == 0 ||
63             strcmp (sm_argv[i], "--sm-save-file") == 0) {
64             ++i, --j; /* skip the next as well, keep j where it is */
65         } else {
66             prop_cmd.vals[j].value = sm_argv[i];
67             prop_cmd.vals[j].length = strlen(sm_argv[i]);
68         }
69     }
70
71     n = (has_file ? sm_argc : sm_argc+2);
72     n = (has_id ? n-2 : n);
73     prop_res.vals = g_new(SmPropValue, n);
74     prop_res.num_vals = n;
75     for (i = 0, j = 0; i < sm_argc; ++i, ++j) { 
76         if (strcmp (sm_argv[i], "--sm-client-id") == 0 ||
77             strcmp (sm_argv[i], "--sm-save-file") == 0) {
78             ++i, --j; /* skip the next as well, keep j where it is */
79         } else {
80             prop_res.vals[j].value = sm_argv[i];
81             prop_res.vals[j].length = strlen(sm_argv[i]);
82         }
83     }
84
85     if (save_file) {
86         prop_res.vals[j].value = "--sm-save-file";
87         prop_res.vals[j++].length = strlen("--sm-save-file");
88         prop_res.vals[j].value = save_file;
89         prop_res.vals[j++].length = strlen(save_file);
90     } else {
91         prop_res.vals[j].value = "--sm-client-id";
92         prop_res.vals[j++].length = strlen("--sm-client-id");
93         prop_res.vals[j].value = ob_sm_id;
94         prop_res.vals[j++].length = strlen(ob_sm_id);
95     }
96
97     props[0] = &prop_res;
98     props[1] = &prop_cmd;
99     SmcSetProperties(sm_conn, 1, props);
100
101     g_free(prop_res.vals);
102     g_free(prop_cmd.vals);
103 }
104
105 void session_startup(int argc, char **argv)
106 {
107 #define SM_ERR_LEN 1024
108
109     SmcCallbacks cb;
110     char sm_err[SM_ERR_LEN];
111
112     sm_argc = argc;
113     sm_argv = argv;
114
115     cb.save_yourself.callback = sm_save_yourself;
116     cb.save_yourself.client_data = NULL;
117
118     cb.die.callback = sm_die;
119     cb.die.client_data = NULL;
120
121     cb.save_complete.callback = sm_save_complete;
122     cb.save_complete.client_data = NULL;
123
124     cb.shutdown_cancelled.callback = sm_shutdown_cancelled;
125     cb.shutdown_cancelled.client_data = NULL;
126
127     sm_conn = SmcOpenConnection(NULL, NULL, 1, 0,
128                                 SmcSaveYourselfProcMask |
129                                 SmcDieProcMask |
130                                 SmcSaveCompleteProcMask |
131                                 SmcShutdownCancelledProcMask,
132                                 &cb, ob_sm_id, &ob_sm_id,
133                                 SM_ERR_LEN, sm_err);
134     if (sm_conn == NULL)
135         g_warning("Failed to connect to session manager: %s", sm_err);
136     else {
137         SmPropValue val_prog;
138         SmPropValue val_uid;
139         SmPropValue val_hint; 
140         SmPropValue val_pri;
141         SmPropValue val_pid;
142         SmProp prop_prog = { SmProgram, SmARRAY8, 1, };
143         SmProp prop_uid = { SmUserID, SmARRAY8, 1, };
144         SmProp prop_hint = { SmRestartStyleHint, SmCARD8, 1, };
145         SmProp prop_pid = { SmProcessID, SmARRAY8, 1, };
146         SmProp prop_pri = { "_GSM_Priority", SmCARD8, 1, };
147         SmProp *props[6];
148         gchar hint, pri;
149         gchar pid[32];
150
151         val_prog.value = argv[0];
152         val_prog.length = strlen(argv[0]);
153
154         val_uid.value = g_strdup(g_get_user_name());
155         val_uid.length = strlen(val_uid.value);
156
157         hint = SmRestartImmediately;
158         val_hint.value = &hint;
159         val_hint.length = 1;
160
161         sprintf(pid, "%ld", (long)getpid());
162         val_pid.value = pid;
163         val_pid.length = strlen(pid);
164
165         /* priority with gnome-session-manager, low to run before other apps */
166         pri = 20;
167         val_pri.value = &pri;
168         val_pri.length = 1;
169
170         prop_prog.vals = &val_prog;
171         prop_uid.vals = &val_uid;
172         prop_hint.vals = &val_hint;
173         prop_pid.vals = &val_pid;
174         prop_pri.vals = &val_pri;
175
176         props[0] = &prop_prog;
177         props[1] = &prop_uid;
178         props[2] = &prop_hint;
179         props[3] = &prop_pid;
180         props[4] = &prop_pri;
181
182         SmcSetProperties(sm_conn, 5, props);
183
184         g_free(val_uid.value);
185
186         save_commands();
187
188         ob_debug("Connected to session manager with id %s\n", ob_sm_id);
189     }
190 }
191
192 void session_shutdown()
193 {
194     g_free(save_file);
195
196     if (sm_conn) {
197         SmPropValue val_hint;
198         SmProp prop_hint = { SmRestartStyleHint, SmCARD8, 1, };
199         SmProp *props[1];
200         gulong hint;
201
202         /* when we exit, we want to reset this to a more friendly state */
203         hint = SmRestartIfRunning;
204         val_hint.value = &hint;
205         val_hint.length = 1;
206
207         prop_hint.vals = &val_hint;
208
209         props[0] = &prop_hint;
210
211         SmcSetProperties(sm_conn, 1, props);
212
213         SmcCloseConnection(sm_conn, 0, NULL);
214     }
215 }
216
217 static void sm_save_yourself(SmcConn conn, SmPointer data, int save_type,
218                              Bool shutdown, int interact_style, Bool fast)
219 {
220     gboolean success;
221
222     ob_debug("got SAVE YOURSELF from session manager\n");
223
224     success = session_save();
225     save_commands();
226
227     SmcSaveYourselfDone(conn, success);
228 }
229
230 static void sm_die(SmcConn conn, SmPointer data)
231 {
232     ob_exit();
233     ob_debug("got DIE from session manager\n");
234 }
235
236 static void sm_save_complete(SmcConn conn, SmPointer data)
237 {
238     ob_debug("got SAVE COMPLETE from session manager\n");
239 }
240
241 static void sm_shutdown_cancelled(SmcConn conn, SmPointer data)
242 {
243     ob_debug("got SHUTDOWN CANCELLED from session manager\n");
244 }
245
246 static gboolean session_save()
247 {
248     gchar *filename;
249     FILE *f;
250     GList *it;
251     gboolean success = TRUE;
252
253     /* this algo is from metacity */
254     filename = g_strdup_printf("%d-%d-%u.obs",
255                                (int) time(NULL),
256                                (int) getpid(),
257                                g_random_int());
258     save_file = g_build_filename(g_get_home_dir(), ".openbox", "sessions",
259                                  filename, NULL);
260     g_free(filename);
261
262     f = fopen(save_file, "w");
263     if (!f) {
264         success = FALSE;
265         g_warning("unable to save the session to %s: %s",
266                   save_file, strerror(errno));
267     } else {
268         fprintf(f, "<?xml version=\"1.0\"?>\n\n");
269         fprintf(f, "<openbox_session id=\"%s\">\n\n", ob_sm_id);
270
271         for (it = client_list; it; it = g_list_next(it)) {
272             guint num;
273             gint32 *dimensions;
274             gint prex, prey, prew, preh;
275             ObClient *c = it->data;
276
277             if (!client_normal(c))
278                 continue;
279
280             prex = c->area.x;
281             prey = c->area.y;
282             prew = c->area.width;
283             preh = c->area.height;
284             if (PROP_GETA32(c->window, openbox_premax, cardinal,
285                             (guint32**)&dimensions, &num)) {
286                 if (num == 4) {
287                     prex = dimensions[0];
288                     prey = dimensions[1];
289                     prew = dimensions[2];
290                     preh = dimensions[3];
291                 }
292                 g_free(dimensions);
293             }
294
295             fprintf(f, "<window id=\"%s\">\n",
296                     g_markup_escape_text("XXX", -1));
297             fprintf(f, "\t<name>%s</name>\n",
298                     g_markup_escape_text(c->name, -1));
299             fprintf(f, "\t<class>%s</class>\n",
300                     g_markup_escape_text(c->class, -1));
301             fprintf(f, "\t<role>%s</role>\n",
302                     g_markup_escape_text(c->role, -1));
303             fprintf(f, "\t<desktop>%d</desktop>\n", c->desktop);
304             fprintf(f, "\t<x>%d</x>\n", prex);
305             fprintf(f, "\t<y>%d</y>\n", prey);
306             fprintf(f, "\t<width>%d</width>\n", prew);
307             fprintf(f, "\t<height>%d</height>\n", preh);
308             if (c->shaded)
309                 fprintf(f, "\t<shaded />\n");
310             if (c->iconic)
311                 fprintf(f, "\t<iconic />\n");
312             if (c->skip_pager)
313                 fprintf(f, "\t<skip_pager />\n");
314             if (c->skip_taskbar)
315                 fprintf(f, "\t<skip_taskbar />\n");
316             if (c->fullscreen)
317                 fprintf(f, "\t<fullscreen />\n");
318             if (c->above)
319                 fprintf(f, "\t<above />\n");
320             if (c->below)
321                 fprintf(f, "\t<below />\n");
322             if (c->max_horz)
323                 fprintf(f, "\t<max_horz />\n");
324             if (c->max_vert)
325                 fprintf(f, "\t<max_vert />\n");
326             fprintf(f, "</window>\n\n");
327         }
328
329         fprintf(f, "</openbox_session>\n");
330
331         if (fflush(f)) {
332             success = FALSE;
333             g_warning("error while saving the session to %s: %s",
334                       save_file, strerror(errno));
335         }
336         fclose(f);
337     }
338
339     return success;
340 }
341
342 void session_load(char *path)
343 {
344     xmlDocPtr doc;
345     xmlNodePtr node, n;
346     gchar *sm_id;
347
348     if (!parse_load(path, "openbox_session", &doc, &node))
349         return;
350
351     if (!parse_attr_string("id", node, &sm_id))
352         return;
353     ob_sm_id = g_strdup(sm_id);
354
355     node = parse_find_node("window", node->xmlChildrenNode);
356     while (node) {
357         gchar *id, *name, *class, *role;
358         guint desktop;
359         gint x, y, w, h;
360         gboolean shaded, iconic, skip_pager, skip_taskbar, fullscreen;
361         gboolean above, below, max_horz, max_vert;
362
363         if (!parse_attr_string("id", node, &id))
364             goto session_load_bail;
365         if (!(n = parse_find_node("name", node->xmlChildrenNode)))
366             goto session_load_bail;
367         name = parse_string(doc, n);
368         if (!(n = parse_find_node("class", node->xmlChildrenNode)))
369             goto session_load_bail;
370         class = parse_string(doc, n);
371         if (!(n = parse_find_node("role", node->xmlChildrenNode)))
372             goto session_load_bail;
373         role = parse_string(doc, n);
374         if (!(n = parse_find_node("desktop", node->xmlChildrenNode)))
375             goto session_load_bail;
376         desktop = parse_int(doc, n);
377         if (!(n = parse_find_node("x", node->xmlChildrenNode)))
378             goto session_load_bail;
379         x = parse_int(doc, n);
380         if (!(n = parse_find_node("y", node->xmlChildrenNode)))
381             goto session_load_bail;
382         y = parse_int(doc, n);
383         if (!(n = parse_find_node("width", node->xmlChildrenNode)))
384             goto session_load_bail;
385         w = parse_int(doc, n);
386         if (!(n = parse_find_node("height", node->xmlChildrenNode)))
387             goto session_load_bail;
388         h = parse_int(doc, n);
389
390         shaded = parse_find_node("shaded", node->xmlChildrenNode) != NULL;
391         iconic = parse_find_node("iconic", node->xmlChildrenNode) != NULL;
392         skip_pager = parse_find_node("skip_pager", node->xmlChildrenNode)
393             != NULL;
394         skip_taskbar = parse_find_node("skip_taskbar", node->xmlChildrenNode)
395             != NULL;
396         fullscreen = parse_find_node("fullscreen", node->xmlChildrenNode)
397             != NULL;
398         above = parse_find_node("above", node->xmlChildrenNode) != NULL;
399         below = parse_find_node("below", node->xmlChildrenNode) != NULL;
400         max_horz = parse_find_node("max_horz", node->xmlChildrenNode) != NULL;
401         max_vert = parse_find_node("max_vert", node->xmlChildrenNode) != NULL;
402         
403         g_message("read session window %s", name);
404
405         /* XXX save this */
406
407     session_load_bail:
408
409         node = parse_find_node("window", node->next);
410     }
411
412     xmlFreeDoc(doc);
413
414     unlink(path);
415 }
416
417 #endif