]> git.sur5r.net Git - i3/i3/blob - src/main.c
fix some memory leaks when user passes command line arguments twice (Thanks Tiago)
[i3/i3] / src / main.c
1 /*
2  * vim:ts=4:sw=4:expandtab
3  */
4 #include <ev.h>
5 #include <fcntl.h>
6 #include <limits.h>
7 #include "all.h"
8
9 static int xkb_event_base;
10
11 int xkb_current_group;
12
13 extern Con *focused;
14
15 char **start_argv;
16
17 xcb_connection_t *conn;
18 xcb_event_handlers_t evenths;
19 xcb_property_handlers_t prophs;
20 xcb_atom_t atoms[NUM_ATOMS];
21
22 xcb_window_t root;
23 uint8_t root_depth;
24
25 xcb_key_symbols_t *keysyms;
26
27 /* Those are our connections to X11 for use with libXcursor and XKB */
28 Display *xlibdpy, *xkbdpy;
29
30 /* The list of key bindings */
31 struct bindings_head *bindings;
32
33 /* The list of exec-lines */
34 struct autostarts_head autostarts = TAILQ_HEAD_INITIALIZER(autostarts);
35
36 /* The list of assignments */
37 struct assignments_head assignments = TAILQ_HEAD_INITIALIZER(assignments);
38
39 /* We hope that those are supported and set them to true */
40 bool xcursor_supported = true;
41 bool xkb_supported = true;
42
43 /*
44  * This callback is only a dummy, see xcb_prepare_cb and xcb_check_cb.
45  * See also man libev(3): "ev_prepare" and "ev_check" - customise your event loop
46  *
47  */
48 static void xcb_got_event(EV_P_ struct ev_io *w, int revents) {
49     /* empty, because xcb_prepare_cb and xcb_check_cb are used */
50 }
51
52 /*
53  * Flush before blocking (and waiting for new events)
54  *
55  */
56 static void xcb_prepare_cb(EV_P_ ev_prepare *w, int revents) {
57     xcb_flush(conn);
58 }
59
60 /*
61  * Instead of polling the X connection socket we leave this to
62  * xcb_poll_for_event() which knows better than we can ever know.
63  *
64  */
65 static void xcb_check_cb(EV_P_ ev_check *w, int revents) {
66     xcb_generic_event_t *event;
67
68     while ((event = xcb_poll_for_event(conn)) != NULL) {
69             xcb_event_handle(&evenths, event);
70             free(event);
71     }
72 }
73
74 int main(int argc, char *argv[]) {
75     //parse_cmd("[ foo ] attach, attach ; focus");
76     int screens;
77     char *override_configpath = NULL;
78     bool autostart = true;
79     char *layout_path = NULL;
80     bool delete_layout_path;
81     bool only_check_config = false;
82     bool force_xinerama = false;
83     xcb_intern_atom_cookie_t atom_cookies[NUM_ATOMS];
84     static struct option long_options[] = {
85         {"no-autostart", no_argument, 0, 'a'},
86         {"config", required_argument, 0, 'c'},
87         {"version", no_argument, 0, 'v'},
88         {"help", no_argument, 0, 'h'},
89         {"layout", required_argument, 0, 'L'},
90         {"restart", required_argument, 0, 0},
91         {"force-xinerama", no_argument, 0, 0},
92         {0, 0, 0, 0}
93     };
94     int option_index = 0, opt;
95
96     setlocale(LC_ALL, "");
97
98     /* Disable output buffering to make redirects in .xsession actually useful for debugging */
99     if (!isatty(fileno(stdout)))
100         setbuf(stdout, NULL);
101
102     start_argv = argv;
103
104     while ((opt = getopt_long(argc, argv, "c:CvaL:hld:V", long_options, &option_index)) != -1) {
105         switch (opt) {
106             case 'a':
107                 LOG("Autostart disabled using -a\n");
108                 autostart = false;
109                 break;
110             case 'L':
111                 FREE(layout_path);
112                 layout_path = sstrdup(optarg);
113                 delete_layout_path = false;
114                 break;
115             case 'c':
116                 FREE(override_configpath);
117                 override_configpath = sstrdup(optarg);
118                 break;
119             case 'C':
120                 LOG("Checking configuration file only (-C)\n");
121                 only_check_config = true;
122                 break;
123             case 'v':
124                 printf("i3 version " I3_VERSION " © 2009 Michael Stapelberg and contributors\n");
125                 exit(EXIT_SUCCESS);
126             case 'V':
127                 set_verbosity(true);
128                 break;
129             case 'd':
130                 LOG("Enabling debug loglevel %s\n", optarg);
131                 add_loglevel(optarg);
132                 break;
133             case 'l':
134                 /* DEPRECATED, ignored for the next 3 versions (3.e, 3.f, 3.g) */
135                 break;
136             case 0:
137                 if (strcmp(long_options[option_index].name, "force-xinerama") == 0) {
138                     force_xinerama = true;
139                     ELOG("Using Xinerama instead of RandR. This option should be "
140                          "avoided at all cost because it does not refresh the list "
141                          "of screens, so you cannot configure displays at runtime. "
142                          "Please check if your driver really does not support RandR "
143                          "and disable this option as soon as you can.\n");
144                     break;
145                 } else if (strcmp(long_options[option_index].name, "restart") == 0) {
146                     FREE(layout_path);
147                     layout_path = sstrdup(optarg);
148                     delete_layout_path = true;
149                     break;
150                 }
151                 /* fall-through */
152             default:
153                 fprintf(stderr, "Usage: %s [-c configfile] [-d loglevel] [-a] [-v] [-V] [-C]\n", argv[0]);
154                 fprintf(stderr, "\n");
155                 fprintf(stderr, "-a: disable autostart\n");
156                 fprintf(stderr, "-L <layoutfile>: load the layout from <layoutfile>\n");
157                 fprintf(stderr, "-v: display version and exit\n");
158                 fprintf(stderr, "-V: enable verbose mode\n");
159                 fprintf(stderr, "-d <loglevel>: enable debug loglevel <loglevel>\n");
160                 fprintf(stderr, "-c <configfile>: use the provided configfile instead\n");
161                 fprintf(stderr, "-C: check configuration file and exit\n");
162                 fprintf(stderr, "--force-xinerama: Use Xinerama instead of RandR. This "
163                                 "option should only be used if you are stuck with the "
164                                 "nvidia closed source driver which does not support RandR.\n");
165                 exit(EXIT_FAILURE);
166         }
167     }
168
169     LOG("i3 (tree) version " I3_VERSION " starting\n");
170
171     conn = xcb_connect(NULL, &screens);
172     if (xcb_connection_has_error(conn))
173         errx(EXIT_FAILURE, "Cannot open display\n");
174
175     load_configuration(conn, override_configpath, false);
176     if (only_check_config) {
177         LOG("Done checking configuration file. Exiting.\n");
178         exit(0);
179     }
180
181     xcb_screen_t *root_screen = xcb_aux_get_screen(conn, screens);
182     root = root_screen->root;
183     root_depth = root_screen->root_depth;
184
185     uint32_t mask = XCB_CW_EVENT_MASK;
186     uint32_t values[] = { XCB_EVENT_MASK_SUBSTRUCTURE_REDIRECT |
187                           XCB_EVENT_MASK_STRUCTURE_NOTIFY |         /* when the user adds a screen (e.g. video
188                                                                            projector), the root window gets a
189                                                                            ConfigureNotify */
190                           XCB_EVENT_MASK_POINTER_MOTION |
191                           XCB_EVENT_MASK_PROPERTY_CHANGE |
192                           XCB_EVENT_MASK_ENTER_WINDOW };
193     xcb_void_cookie_t cookie;
194     cookie = xcb_change_window_attributes_checked(conn, root, mask, values);
195     check_error(conn, cookie, "Another window manager seems to be running");
196
197     /* Place requests for the atoms we need as soon as possible */
198     #define REQUEST_ATOM(name) atom_cookies[name] = xcb_intern_atom(conn, 0, strlen(#name), #name);
199
200     REQUEST_ATOM(_NET_SUPPORTED);
201     REQUEST_ATOM(_NET_WM_STATE_FULLSCREEN);
202     REQUEST_ATOM(_NET_SUPPORTING_WM_CHECK);
203     REQUEST_ATOM(_NET_WM_NAME);
204     REQUEST_ATOM(_NET_WM_STATE);
205     REQUEST_ATOM(_NET_WM_WINDOW_TYPE);
206     REQUEST_ATOM(_NET_WM_DESKTOP);
207     REQUEST_ATOM(_NET_WM_WINDOW_TYPE_DOCK);
208     REQUEST_ATOM(_NET_WM_WINDOW_TYPE_DIALOG);
209     REQUEST_ATOM(_NET_WM_WINDOW_TYPE_UTILITY);
210     REQUEST_ATOM(_NET_WM_WINDOW_TYPE_TOOLBAR);
211     REQUEST_ATOM(_NET_WM_WINDOW_TYPE_SPLASH);
212     REQUEST_ATOM(_NET_WM_STRUT_PARTIAL);
213     REQUEST_ATOM(WM_PROTOCOLS);
214     REQUEST_ATOM(WM_DELETE_WINDOW);
215     REQUEST_ATOM(UTF8_STRING);
216     REQUEST_ATOM(WM_STATE);
217     REQUEST_ATOM(WM_CLIENT_LEADER);
218     REQUEST_ATOM(_NET_CURRENT_DESKTOP);
219     REQUEST_ATOM(_NET_ACTIVE_WINDOW);
220     REQUEST_ATOM(_NET_WORKAREA);
221
222     /* Initialize the Xlib connection */
223     xlibdpy = xkbdpy = XOpenDisplay(NULL);
224
225     /* Try to load the X cursors and initialize the XKB extension */
226     if (xlibdpy == NULL) {
227         ELOG("ERROR: XOpenDisplay() failed, disabling libXcursor/XKB support\n");
228         xcursor_supported = false;
229         xkb_supported = false;
230     } else if (fcntl(ConnectionNumber(xlibdpy), F_SETFD, FD_CLOEXEC) == -1) {
231         ELOG("Could not set FD_CLOEXEC on xkbdpy\n");
232         return 1;
233     } else {
234         xcursor_load_cursors();
235         /*init_xkb();*/
236     }
237
238     memset(&evenths, 0, sizeof(xcb_event_handlers_t));
239     memset(&prophs, 0, sizeof(xcb_property_handlers_t));
240
241     xcb_event_handlers_init(conn, &evenths);
242     xcb_property_handlers_init(&prophs, &evenths);
243     xcb_event_set_key_press_handler(&evenths, handle_key_press, NULL);
244
245     xcb_event_set_button_press_handler(&evenths, handle_button_press, NULL);
246
247     xcb_event_set_map_request_handler(&evenths, handle_map_request, NULL);
248
249     xcb_event_set_unmap_notify_handler(&evenths, handle_unmap_notify_event, NULL);
250     xcb_event_set_destroy_notify_handler(&evenths, handle_destroy_notify_event, NULL);
251
252     xcb_event_set_expose_handler(&evenths, handle_expose_event, NULL);
253
254     xcb_event_set_motion_notify_handler(&evenths, handle_motion_notify, NULL);
255
256     /* Enter window = user moved his mouse over the window */
257     xcb_event_set_enter_notify_handler(&evenths, handle_enter_notify, NULL);
258
259     /* Client message are sent to the root window. The only interesting client message
260        for us is _NET_WM_STATE, we honour _NET_WM_STATE_FULLSCREEN */
261     xcb_event_set_client_message_handler(&evenths, handle_client_message, NULL);
262
263     /* Configure request = window tried to change size on its own */
264     xcb_event_set_configure_request_handler(&evenths, handle_configure_request, NULL);
265
266     /* Setup NetWM atoms */
267     #define GET_ATOM(name) \
268         do { \
269             xcb_intern_atom_reply_t *reply = xcb_intern_atom_reply(conn, atom_cookies[name], NULL); \
270             if (!reply) { \
271                 ELOG("Could not get atom " #name "\n"); \
272                 exit(-1); \
273             } \
274             atoms[name] = reply->atom; \
275             free(reply); \
276         } while (0)
277
278     GET_ATOM(_NET_SUPPORTED);
279     GET_ATOM(_NET_WM_STATE_FULLSCREEN);
280     GET_ATOM(_NET_SUPPORTING_WM_CHECK);
281     GET_ATOM(_NET_WM_NAME);
282     GET_ATOM(_NET_WM_STATE);
283     GET_ATOM(_NET_WM_WINDOW_TYPE);
284     GET_ATOM(_NET_WM_DESKTOP);
285     GET_ATOM(_NET_WM_WINDOW_TYPE_DOCK);
286     GET_ATOM(_NET_WM_WINDOW_TYPE_DIALOG);
287     GET_ATOM(_NET_WM_WINDOW_TYPE_UTILITY);
288     GET_ATOM(_NET_WM_WINDOW_TYPE_TOOLBAR);
289     GET_ATOM(_NET_WM_WINDOW_TYPE_SPLASH);
290     GET_ATOM(_NET_WM_STRUT_PARTIAL);
291     GET_ATOM(WM_PROTOCOLS);
292     GET_ATOM(WM_DELETE_WINDOW);
293     GET_ATOM(UTF8_STRING);
294     GET_ATOM(WM_STATE);
295     GET_ATOM(WM_CLIENT_LEADER);
296     GET_ATOM(_NET_CURRENT_DESKTOP);
297     GET_ATOM(_NET_ACTIVE_WINDOW);
298     GET_ATOM(_NET_WORKAREA);
299
300     /* Watch _NET_WM_NAME (title of the window encoded in UTF-8) */
301     xcb_property_set_handler(&prophs, atoms[_NET_WM_NAME], 128, handle_windowname_change, NULL);
302
303     /* Watch WM_HINTS (contains the urgent property) */
304     xcb_property_set_handler(&prophs, WM_HINTS, UINT_MAX, handle_hints, NULL);
305
306     /* Watch WM_NAME (title of the window encoded in COMPOUND_TEXT) */
307     xcb_watch_wm_name(&prophs, 128, handle_windowname_change_legacy, NULL);
308
309     /* Watch WM_NORMAL_HINTS (aspect ratio, size increments, …) */
310     xcb_property_set_handler(&prophs, WM_NORMAL_HINTS, UINT_MAX, handle_normal_hints, NULL);
311
312     /* Watch WM_CLIENT_LEADER (= logical parent window for toolbars etc.) */
313     xcb_property_set_handler(&prophs, atoms[WM_CLIENT_LEADER], UINT_MAX, handle_clientleader_change, NULL);
314
315     /* Watch WM_TRANSIENT_FOR property (to which client this popup window belongs) */
316     xcb_property_set_handler(&prophs, WM_TRANSIENT_FOR, UINT_MAX, handle_transient_for, NULL);
317
318     /* Set up the atoms we support */
319     xcb_change_property(conn, XCB_PROP_MODE_REPLACE, root, atoms[_NET_SUPPORTED], ATOM, 32, 7, atoms);
320     /* Set up the window manager’s name */
321     xcb_change_property(conn, XCB_PROP_MODE_REPLACE, root, atoms[_NET_SUPPORTING_WM_CHECK], WINDOW, 32, 1, &root);
322     xcb_change_property(conn, XCB_PROP_MODE_REPLACE, root, atoms[_NET_WM_NAME], atoms[UTF8_STRING], 8, strlen("i3"), "i3");
323
324     keysyms = xcb_key_symbols_alloc(conn);
325
326     xcb_get_numlock_mask(conn);
327
328     translate_keysyms();
329     grab_all_keys(conn, false);
330
331     bool needs_tree_init = true;
332     if (layout_path) {
333         LOG("Trying to restore the layout from %s...", layout_path);
334         needs_tree_init = !tree_restore(layout_path);
335         if (delete_layout_path)
336             unlink(layout_path);
337         free(layout_path);
338     }
339     if (needs_tree_init)
340         tree_init();
341
342     int randr_base;
343     if (force_xinerama) {
344         xinerama_init();
345     } else {
346         DLOG("Checking for XRandR...\n");
347         randr_init(&randr_base);
348
349         xcb_event_set_handler(&evenths,
350                               randr_base + XCB_RANDR_SCREEN_CHANGE_NOTIFY,
351                               handle_screen_change,
352                               NULL);
353     }
354
355     tree_render();
356
357     struct ev_loop *loop = ev_loop_new(0);
358     if (loop == NULL)
359             die("Could not initialize libev. Bad LIBEV_FLAGS?\n");
360
361     /* Create the UNIX domain socket for IPC */
362     if (config.ipc_socket_path != NULL) {
363         int ipc_socket = ipc_create_socket(config.ipc_socket_path);
364         if (ipc_socket == -1) {
365             ELOG("Could not create the IPC socket, IPC disabled\n");
366         } else {
367             struct ev_io *ipc_io = scalloc(sizeof(struct ev_io));
368             ev_io_init(ipc_io, ipc_new_client, ipc_socket, EV_READ);
369             ev_io_start(loop, ipc_io);
370         }
371     }
372
373     struct ev_io *xcb_watcher = scalloc(sizeof(struct ev_io));
374     struct ev_check *xcb_check = scalloc(sizeof(struct ev_check));
375     struct ev_prepare *xcb_prepare = scalloc(sizeof(struct ev_prepare));
376
377     ev_io_init(xcb_watcher, xcb_got_event, xcb_get_file_descriptor(conn), EV_READ);
378     ev_io_start(loop, xcb_watcher);
379
380     ev_check_init(xcb_check, xcb_check_cb);
381     ev_check_start(loop, xcb_check);
382
383     ev_prepare_init(xcb_prepare, xcb_prepare_cb);
384     ev_prepare_start(loop, xcb_prepare);
385
386     xcb_flush(conn);
387
388     manage_existing_windows(root);
389
390     setup_signal_handler();
391
392     /* Ignore SIGPIPE to survive errors when an IPC client disconnects
393      * while we are sending him a message */
394     signal(SIGPIPE, SIG_IGN);
395
396     /* Autostarting exec-lines */
397     if (autostart) {
398         struct Autostart *exec;
399         TAILQ_FOREACH(exec, &autostarts, autostarts) {
400             LOG("auto-starting %s\n", exec->command);
401             start_application(exec->command);
402         }
403     }
404
405     ev_loop(loop, 0);
406 }