wmfs.c 16 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626
  1. /*
  2. * wmfs2 by Martin Duquesnoy <xorg62@gmail.com> { for(i = 2011; i < 2111; ++i) ©(i); }
  3. * For license, see COPYING.
  4. */
  5. #include <string.h>
  6. #include <getopt.h>
  7. #include <signal.h>
  8. #include <sys/wait.h>
  9. #include <X11/keysym.h>
  10. #include <X11/cursorfont.h>
  11. #ifdef HAVE_IMLIB2
  12. #include <Imlib2.h>
  13. #endif /* HAVE_IMLIB2 */
  14. #include "wmfs.h"
  15. #include "event.h"
  16. #include "ewmh.h"
  17. #include "screen.h"
  18. #include "infobar.h"
  19. #include "util.h"
  20. #include "config.h"
  21. #include "client.h"
  22. #include "layout.h"
  23. int
  24. wmfs_error_handler(Display *d, XErrorEvent *event)
  25. {
  26. char mess[256];
  27. /* Check if there is another WM running */
  28. if(event->error_code == BadAccess
  29. && W->root == event->resourceid)
  30. errl(EXIT_FAILURE, "Another Window Manager is already running.");
  31. /* Ignore focus change error for unmapped client
  32. * 42 = X_SetInputFocus
  33. * 28 = X_GrabButton
  34. */
  35. if(client_gb_win(event->resourceid))
  36. if(event->error_code == BadWindow
  37. || event->request_code == 42
  38. || event->request_code == 28)
  39. return 0;
  40. if(XGetErrorText(d, event->error_code, mess, 128))
  41. warnxl("%s(%d) opcodes %d/%d\n resource #%lx\n",
  42. mess,
  43. event->error_code,
  44. event->request_code,
  45. event->minor_code,
  46. event->resourceid);
  47. return 1;
  48. }
  49. int
  50. wmfs_error_handler_dummy(Display *d, XErrorEvent *event)
  51. {
  52. (void)d;
  53. (void)event;
  54. return 0;
  55. }
  56. void
  57. wmfs_numlockmask(void)
  58. {
  59. int i, j;
  60. XModifierKeymap *mm = XGetModifierMapping(W->dpy);
  61. for(i = 0; i < 8; i++)
  62. for(j = 0; j < mm->max_keypermod; ++j)
  63. if(mm->modifiermap[i * mm->max_keypermod + j]
  64. == XKeysymToKeycode(W->dpy, XK_Num_Lock))
  65. W->numlockmask = (1 << i);
  66. XFreeModifiermap(mm);
  67. }
  68. void
  69. wmfs_init_font(char *font, struct theme *t)
  70. {
  71. XFontStruct **xfs = NULL;
  72. char **misschar, **names, *defstring;
  73. int d;
  74. if(!(t->font.fontset = XCreateFontSet(W->dpy, font, &misschar, &d, &defstring)))
  75. {
  76. warnxl("Can't load font '%s'", font);
  77. t->font.fontset = XCreateFontSet(W->dpy, "fixed", &misschar, &d, &defstring);
  78. }
  79. XExtentsOfFontSet(t->font.fontset);
  80. XFontsOfFontSet(t->font.fontset, &xfs, &names);
  81. t->font.as = xfs[0]->max_bounds.ascent;
  82. t->font.de = xfs[0]->max_bounds.descent;
  83. t->font.width = xfs[0]->max_bounds.width;
  84. t->font.height = t->font.as + t->font.de;
  85. if(misschar)
  86. XFreeStringList(misschar);
  87. }
  88. static void
  89. wmfs_xinit(void)
  90. {
  91. XGCValues xgc =
  92. {
  93. .function = GXinvert,
  94. .subwindow_mode = IncludeInferiors,
  95. .line_width = 1
  96. };
  97. XSetWindowAttributes at =
  98. {
  99. .event_mask = (KeyMask | ButtonMask | MouseMask
  100. | PropertyChangeMask | SubstructureRedirectMask
  101. | SubstructureNotifyMask | StructureNotifyMask),
  102. .cursor = XCreateFontCursor(W->dpy, XC_left_ptr)
  103. };
  104. /*
  105. * X Error handler
  106. */
  107. XSetErrorHandler(wmfs_error_handler);
  108. /*
  109. * X var
  110. */
  111. W->xscreen = DefaultScreen(W->dpy);
  112. W->xdepth = DefaultDepth(W->dpy, W->xscreen);
  113. W->gc = DefaultGC(W->dpy, W->xscreen);
  114. W->xmaxw = DisplayWidth(W->dpy, W->xscreen);
  115. W->xmaxh = DisplayHeight(W->dpy, W->xscreen);
  116. /*
  117. * Keys
  118. */
  119. wmfs_numlockmask();
  120. /*
  121. * Root window/cursor
  122. */
  123. W->root = RootWindow(W->dpy, W->xscreen);
  124. XChangeWindowAttributes(W->dpy, W->root, CWEventMask | CWCursor, &at);
  125. W->rgc = XCreateGC(W->dpy, W->root, GCFunction | GCSubwindowMode | GCLineWidth, &xgc);
  126. /*
  127. * Locale (font encode)
  128. */
  129. setlocale(LC_CTYPE, "");
  130. /*
  131. * Barwin linked list
  132. */
  133. SLIST_INIT(&W->h.barwin);
  134. /*
  135. * Optional dep init
  136. */
  137. #ifdef HAVE_IMLIB2
  138. imlib_context_set_display(W->dpy);
  139. imlib_context_set_visual(DefaultVisual(W->dpy, W->xscreen));
  140. imlib_context_set_colormap(DefaultColormap(W->dpy, W->xscreen));
  141. #endif /* HAVE_IMLIB2 */
  142. W->flags |= WMFS_RUNNING;
  143. }
  144. void
  145. wmfs_grab_keys(void)
  146. {
  147. KeyCode c;
  148. struct keybind *k;
  149. wmfs_numlockmask();
  150. XUngrabKey(W->dpy, AnyKey, AnyModifier, W->root);
  151. SLIST_FOREACH(k, &W->h.keybind, next)
  152. if((c = XKeysymToKeycode(W->dpy, k->keysym)))
  153. {
  154. XGrabKey(W->dpy, c, k->mod, W->root, True, GrabModeAsync, GrabModeAsync);
  155. XGrabKey(W->dpy, c, k->mod | LockMask, W->root, True, GrabModeAsync, GrabModeAsync);
  156. XGrabKey(W->dpy, c, k->mod | W->numlockmask, W->root, True, GrabModeAsync, GrabModeAsync);
  157. XGrabKey(W->dpy, c, k->mod | LockMask | W->numlockmask, W->root, True, GrabModeAsync, GrabModeAsync);
  158. }
  159. }
  160. /** Scan xprops of previous session to set it back
  161. * Check if there are windows on X (previous sessions windows)
  162. */
  163. static void
  164. wmfs_scan(void)
  165. {
  166. struct geo g;
  167. struct client *c, *cc, *fc;
  168. struct screen *s;
  169. int i, n, rf, nscreen = 0;
  170. int tag = -1, screen = -1, flags = -1;
  171. unsigned long ir, il;
  172. long *ret = NULL, *tret = NULL;
  173. bool getg = false;
  174. XWindowAttributes wa;
  175. Window usl, usl2, *w = NULL, tm, focus;
  176. Atom rt;
  177. SLIST_INIT(&W->h.client);
  178. W->flags |= WMFS_SCAN;
  179. /* Get previous selected tag to apply it at the end */
  180. if(XGetWindowProperty(W->dpy, W->root, W->net_atom[wmfs_current_tag], 0, 32,
  181. False, XA_CARDINAL, &rt, &rf, &ir, &il,
  182. (unsigned char**)&tret)
  183. == Success && tret)
  184. {
  185. nscreen = (int)ir;
  186. }
  187. /* Previous focused client before reload */
  188. if(XGetWindowProperty(W->dpy, W->root, W->net_atom[wmfs_focus], 0, 32,
  189. False, XA_WINDOW, &rt, &rf, &ir, &il,
  190. (unsigned char**)&ret)
  191. == Success && ret)
  192. {
  193. focus = *ret;
  194. XFree(ret);
  195. }
  196. if(XQueryTree(W->dpy, W->root, &usl, &usl2, &w, (unsigned int*)&n))
  197. for(i = n - 1; i != -1; --i)
  198. {
  199. XGetWindowAttributes(W->dpy, w[i], &wa);
  200. if(!wa.override_redirect && wa.map_state == IsViewable)
  201. {
  202. if(XGetWindowProperty(W->dpy, w[i], ATOM("_WMFS_TAG"), 0, 32,
  203. False, XA_CARDINAL, &rt, &rf, &ir, &il,
  204. (unsigned char**)&ret)
  205. == Success && ret)
  206. {
  207. tag = *ret;
  208. XFree(ret);
  209. }
  210. if(XGetWindowProperty(W->dpy, w[i], ATOM("_WMFS_SCREEN"), 0, 32,
  211. False, XA_CARDINAL, &rt, &rf, &ir, &il,
  212. (unsigned char**)&ret)
  213. == Success && ret)
  214. {
  215. screen = *ret;
  216. XFree(ret);
  217. }
  218. if(XGetWindowProperty(W->dpy, w[i], ATOM("_WMFS_FLAGS"), 0, 32,
  219. False, XA_CARDINAL, &rt, &rf, &ir, &il,
  220. (unsigned char**)&ret)
  221. == Success && ret)
  222. {
  223. flags = *ret;
  224. flags &= ~(CLIENT_TABBED | CLIENT_REMOVEALL);
  225. XFree(ret);
  226. }
  227. if(XGetWindowProperty(W->dpy, w[i], ATOM("_WMFS_GEO"), 0, 32,
  228. False, XA_CARDINAL, &rt, &rf, &ir, &il,
  229. (unsigned char**)&ret)
  230. == Success && ret)
  231. {
  232. g.x = ret[0];
  233. g.y = ret[1];
  234. g.w = ret[2];
  235. g.h = ret[3];
  236. getg = true;
  237. XFree(ret);
  238. }
  239. if(XGetWindowProperty(W->dpy, w[i], ATOM("_WMFS_TABMASTER"), 0, 32,
  240. False, XA_WINDOW, &rt, &rf, &ir, &il,
  241. (unsigned char**)&ret)
  242. == Success && ret)
  243. {
  244. tm = *ret;
  245. XFree(ret);
  246. }
  247. c = client_new(w[i], &wa, true);
  248. if(tm != c->win)
  249. c->tmp = tm;
  250. tm = 0;
  251. if(flags != -1)
  252. c->flags |= flags;
  253. if(tag != -1 && screen != -1)
  254. {
  255. c->screen = screen_gb_id(screen);
  256. if(getg)
  257. c->flags |= CLIENT_IGNORE_LAYOUT;
  258. client_map(c);
  259. tag_client(tag_gb_id(c->screen, tag), c);
  260. if(getg)
  261. client_moveresize(c, &g);
  262. client_get_name(c);
  263. }
  264. }
  265. }
  266. if(!nscreen)
  267. {
  268. SLIST_FOREACH(s, &W->h.screen, next)
  269. if(!TAILQ_EMPTY(&s->tags))
  270. tag_screen(s, TAILQ_FIRST(&s->tags));
  271. }
  272. else
  273. {
  274. /* Set back selected tag */
  275. for(i = 0; i < nscreen; ++i)
  276. {
  277. s = screen_gb_id(i);
  278. tag_screen(s, tag_gb_id(s, tret[i]));
  279. }
  280. }
  281. /* Re-adjust tabbed clients */
  282. SLIST_FOREACH(c, &W->h.client, next)
  283. if((cc = client_gb_win(c->tmp)) && cc != c)
  284. _client_tab(c, cc);
  285. if((fc = client_gb_win(focus)) && fc != W->client)
  286. client_focus(fc);
  287. SLIST_FOREACH(c, &W->h.client, next)
  288. layout_fix_hole(c);
  289. W->flags &= ~WMFS_SCAN;
  290. if(tret)
  291. XFree(tret);
  292. XFree(w);
  293. XSync(W->dpy, false);
  294. }
  295. static inline void
  296. wmfs_sigchld(void)
  297. {
  298. if(W->flags & WMFS_SIGCHLD)
  299. {
  300. while(waitpid(-1, NULL, WNOHANG) > 0);
  301. W->flags ^= WMFS_SIGCHLD;
  302. }
  303. }
  304. static void
  305. wmfs_loop(void)
  306. {
  307. XEvent ev;
  308. while((W->flags & WMFS_RUNNING) && !XNextEvent(W->dpy, &ev))
  309. {
  310. /* Manage SIGCHLD event here, X is not safe with it */
  311. wmfs_sigchld();
  312. EVENT_HANDLE(&ev);
  313. }
  314. }
  315. static inline void
  316. wmfs_init(void)
  317. {
  318. log_init();
  319. wmfs_xinit();
  320. ewmh_init();
  321. screen_init();
  322. event_init();
  323. config_init();
  324. }
  325. void
  326. wmfs_quit(void)
  327. {
  328. struct keybind *k;
  329. struct rule *r;
  330. struct theme *t;
  331. struct client *c;
  332. struct mousebind *m;
  333. struct launcher *l;
  334. ewmh_update_wmfs_props();
  335. XFreeGC(W->dpy, W->rgc);
  336. while(!SLIST_EMPTY(&W->h.client))
  337. {
  338. c = SLIST_FIRST(&W->h.client);
  339. client_update_props(c, CPROP_LOC | CPROP_FLAG | CPROP_GEO);
  340. c->flags |= (CLIENT_IGNORE_LAYOUT | CLIENT_REMOVEALL);
  341. client_remove(c);
  342. }
  343. /* Will free:
  344. *
  345. * Screens -> tags
  346. * -> Infobars -> Elements
  347. */
  348. screen_free();
  349. /* Conf stuffs */
  350. while(!SLIST_EMPTY(&W->h.theme))
  351. {
  352. t = SLIST_FIRST(&W->h.theme);
  353. SLIST_REMOVE_HEAD(&W->h.theme, next);
  354. XFreeFontSet(W->dpy, t->font.fontset);
  355. status_free_ctx(&t->tags_n_sl);
  356. status_free_ctx(&t->tags_s_sl);
  357. status_free_ctx(&t->tags_o_sl);
  358. status_free_ctx(&t->tags_u_sl);
  359. status_free_ctx(&t->client_n_sl);
  360. status_free_ctx(&t->client_s_sl);
  361. free(t);
  362. }
  363. while(!SLIST_EMPTY(&W->h.keybind))
  364. {
  365. k = SLIST_FIRST(&W->h.keybind);
  366. SLIST_REMOVE_HEAD(&W->h.keybind, next);
  367. free((void*)k->cmd);
  368. free(k);
  369. }
  370. while(!SLIST_EMPTY(&W->h.mousebind))
  371. {
  372. m = SLIST_FIRST(&W->h.mousebind);
  373. SLIST_REMOVE_HEAD(&W->h.mousebind, globnext);
  374. free((void*)m->cmd);
  375. free(m);
  376. }
  377. while(!SLIST_EMPTY(&W->h.launcher))
  378. {
  379. l = SLIST_FIRST(&W->h.launcher);
  380. SLIST_REMOVE_HEAD(&W->h.launcher, next);
  381. free((void*)l->name);
  382. free((void*)l->prompt);
  383. free((void*)l->command);
  384. free(l);
  385. }
  386. while(!SLIST_EMPTY(&W->h.rule))
  387. {
  388. r = SLIST_FIRST(&W->h.rule);
  389. SLIST_REMOVE_HEAD(&W->h.rule, next);
  390. free(r->class);
  391. free(r->instance);
  392. free(r->role);
  393. free(r->name);
  394. free(r);
  395. }
  396. /* close log */
  397. if(W->log)
  398. fclose(W->log), W->log = NULL;
  399. W->flags &= ~WMFS_RUNNING;
  400. }
  401. /** Reload WMFS binary
  402. */
  403. void
  404. uicb_reload(Uicb cmd)
  405. {
  406. (void)cmd;
  407. W->flags &= ~WMFS_RUNNING;
  408. W->flags |= WMFS_RELOAD;
  409. }
  410. void
  411. uicb_quit(Uicb cmd)
  412. {
  413. (void)cmd;
  414. W->flags &= ~WMFS_RUNNING;
  415. }
  416. static void
  417. exec_uicb_function(Display *dpy, Window root, char *func, char *cmd)
  418. {
  419. Atom utf8s = XInternAtom(dpy, "UTF8_STRING", false);
  420. XClientMessageEvent e = {
  421. .type = ClientMessage,
  422. .message_type = XInternAtom(dpy, "_WMFS_FUNCTION", false),
  423. .window = root,
  424. .format = 32,
  425. .data.l[4] = true
  426. };
  427. XChangeProperty(dpy,root, XInternAtom(dpy, "_WMFS_FUNCTION", false), utf8s,
  428. 8, PropModeReplace, (unsigned char*)func, strlen(func));
  429. if(!cmd)
  430. cmd = "";
  431. XChangeProperty(dpy, root, XInternAtom(dpy, "_WMFS_CMD", false), utf8s,
  432. 8, PropModeReplace, (unsigned char*)cmd, strlen(cmd));
  433. XSendEvent(dpy, root, false, StructureNotifyMask, (XEvent*)&e);
  434. XSync(dpy, False);
  435. }
  436. static void
  437. signal_handle(int sig)
  438. {
  439. switch (sig)
  440. {
  441. case SIGQUIT:
  442. case SIGTERM:
  443. W->flags &= ~WMFS_RUNNING;
  444. break;
  445. case SIGCHLD:
  446. W->flags |= WMFS_SIGCHLD;
  447. break;
  448. }
  449. }
  450. int
  451. main(int argc, char **argv)
  452. {
  453. int i;
  454. bool r;
  455. Display *dpy;
  456. char path[MAX_PATH_LEN] = { 0 };
  457. struct sigaction sa;
  458. (void)argc;
  459. sprintf(path, "%s/"CONFIG_DEFAULT_PATH, getenv("HOME"));
  460. /* Opt */
  461. while((i = getopt(argc, argv, "hvC:c:")) != -1)
  462. {
  463. switch(i)
  464. {
  465. default:
  466. case 'h':
  467. printf("usage: %s [-hv] [-c <func> <cmd] [-C <file>]\n"
  468. " -h Show this page\n"
  469. " -v Show WMFS version\n"
  470. " -c <func> <cmd> Execute a specified UICB function\n"
  471. " -C <file> Launch WMFS with a specified configuration file\n", argv[0]);
  472. exit(EXIT_SUCCESS);
  473. break;
  474. case 'v':
  475. printf("wmfs("WMFS_VERSION") 2 beta\n");
  476. exit(EXIT_SUCCESS);
  477. break;
  478. case 'c':
  479. if(!(dpy = XOpenDisplay(NULL)))
  480. {
  481. fprintf(stderr, "%s: Can't open X server\n", argv[0]);
  482. exit(EXIT_FAILURE);
  483. }
  484. exec_uicb_function(dpy, DefaultRootWindow(dpy), optarg, argv[optind]);
  485. XCloseDisplay(dpy);
  486. exit(EXIT_SUCCESS);
  487. break;
  488. case 'C':
  489. strncpy(path, optarg, sizeof(W->confpath));
  490. break;
  491. }
  492. }
  493. W = (struct wmfs*)xcalloc(1, sizeof(struct wmfs));
  494. /* Default path ~/.config/wmfs/wmfsrc */
  495. W->confpath = path;
  496. /* Get X display */
  497. if(!(W->dpy = XOpenDisplay(NULL)))
  498. {
  499. fprintf(stderr, "%s: Can't open X server\n", argv[0]);
  500. exit(EXIT_FAILURE);
  501. }
  502. /* Set signal handler */
  503. memset(&sa, 0, sizeof(sa));
  504. sa.sa_handler = signal_handle;
  505. sigemptyset(&sa.sa_mask);
  506. sigaction(SIGQUIT, &sa, NULL);
  507. sigaction(SIGTERM, &sa, NULL);
  508. sigaction(SIGCHLD, &sa, NULL);
  509. /* Core */
  510. wmfs_init();
  511. wmfs_scan();
  512. wmfs_loop();
  513. wmfs_quit();
  514. r = (W->flags & WMFS_RELOAD);
  515. free(W);
  516. if(r)
  517. execvp(argv[0], argv);
  518. XCloseDisplay(W->dpy);
  519. return 1;
  520. }