wmfs.c 16 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627
  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. SLIST_INIT(&W->h.vbarwin);
  135. /*
  136. * Optional dep init
  137. */
  138. #ifdef HAVE_IMLIB2
  139. imlib_context_set_display(W->dpy);
  140. imlib_context_set_visual(DefaultVisual(W->dpy, W->xscreen));
  141. imlib_context_set_colormap(DefaultColormap(W->dpy, W->xscreen));
  142. #endif /* HAVE_IMLIB2 */
  143. W->flags |= WMFS_RUNNING;
  144. }
  145. void
  146. wmfs_grab_keys(void)
  147. {
  148. KeyCode c;
  149. struct keybind *k;
  150. wmfs_numlockmask();
  151. XUngrabKey(W->dpy, AnyKey, AnyModifier, W->root);
  152. SLIST_FOREACH(k, &W->h.keybind, next)
  153. if((c = XKeysymToKeycode(W->dpy, k->keysym)))
  154. {
  155. XGrabKey(W->dpy, c, k->mod, W->root, True, GrabModeAsync, GrabModeAsync);
  156. XGrabKey(W->dpy, c, k->mod | LockMask, W->root, True, GrabModeAsync, GrabModeAsync);
  157. XGrabKey(W->dpy, c, k->mod | W->numlockmask, W->root, True, GrabModeAsync, GrabModeAsync);
  158. XGrabKey(W->dpy, c, k->mod | LockMask | W->numlockmask, W->root, True, GrabModeAsync, GrabModeAsync);
  159. }
  160. }
  161. /** Scan xprops of previous session to set it back
  162. * Check if there are windows on X (previous sessions windows)
  163. */
  164. static void
  165. wmfs_scan(void)
  166. {
  167. struct geo g;
  168. struct client *c, *cc, *fc;
  169. struct screen *s;
  170. int i, n, rf, nscreen = 0;
  171. int tag = -1, screen = -1, flags = -1;
  172. unsigned long ir, il;
  173. long *ret = NULL, *tret = NULL;
  174. bool getg = false;
  175. XWindowAttributes wa;
  176. Window usl, usl2, *w = NULL, tm, focus;
  177. Atom rt;
  178. SLIST_INIT(&W->h.client);
  179. W->flags |= WMFS_SCAN;
  180. /* Get previous selected tag to apply it at the end */
  181. if(XGetWindowProperty(W->dpy, W->root, W->net_atom[wmfs_current_tag], 0, 32,
  182. False, XA_CARDINAL, &rt, &rf, &ir, &il,
  183. (unsigned char**)&tret)
  184. == Success && tret)
  185. {
  186. nscreen = (int)ir;
  187. }
  188. /* Previous focused client before reload */
  189. if(XGetWindowProperty(W->dpy, W->root, W->net_atom[wmfs_focus], 0, 32,
  190. False, XA_WINDOW, &rt, &rf, &ir, &il,
  191. (unsigned char**)&ret)
  192. == Success && ret)
  193. {
  194. focus = *ret;
  195. XFree(ret);
  196. }
  197. if(XQueryTree(W->dpy, W->root, &usl, &usl2, &w, (unsigned int*)&n))
  198. for(i = n - 1; i != -1; --i)
  199. {
  200. XGetWindowAttributes(W->dpy, w[i], &wa);
  201. if(!wa.override_redirect && wa.map_state == IsViewable)
  202. {
  203. if(XGetWindowProperty(W->dpy, w[i], ATOM("_WMFS_TAG"), 0, 32,
  204. False, XA_CARDINAL, &rt, &rf, &ir, &il,
  205. (unsigned char**)&ret)
  206. == Success && ret)
  207. {
  208. tag = *ret;
  209. XFree(ret);
  210. }
  211. if(XGetWindowProperty(W->dpy, w[i], ATOM("_WMFS_SCREEN"), 0, 32,
  212. False, XA_CARDINAL, &rt, &rf, &ir, &il,
  213. (unsigned char**)&ret)
  214. == Success && ret)
  215. {
  216. screen = *ret;
  217. XFree(ret);
  218. }
  219. if(XGetWindowProperty(W->dpy, w[i], ATOM("_WMFS_FLAGS"), 0, 32,
  220. False, XA_CARDINAL, &rt, &rf, &ir, &il,
  221. (unsigned char**)&ret)
  222. == Success && ret)
  223. {
  224. flags = *ret;
  225. flags &= ~(CLIENT_TABBED | CLIENT_REMOVEALL);
  226. XFree(ret);
  227. }
  228. if(XGetWindowProperty(W->dpy, w[i], ATOM("_WMFS_GEO"), 0, 32,
  229. False, XA_CARDINAL, &rt, &rf, &ir, &il,
  230. (unsigned char**)&ret)
  231. == Success && ret)
  232. {
  233. g.x = ret[0];
  234. g.y = ret[1];
  235. g.w = ret[2];
  236. g.h = ret[3];
  237. getg = true;
  238. XFree(ret);
  239. }
  240. if(XGetWindowProperty(W->dpy, w[i], ATOM("_WMFS_TABMASTER"), 0, 32,
  241. False, XA_WINDOW, &rt, &rf, &ir, &il,
  242. (unsigned char**)&ret)
  243. == Success && ret)
  244. {
  245. tm = *ret;
  246. XFree(ret);
  247. }
  248. c = client_new(w[i], &wa, true);
  249. if(tm != c->win)
  250. c->tmp = tm;
  251. tm = 0;
  252. if(flags != -1)
  253. c->flags |= flags;
  254. if(tag != -1 && screen != -1)
  255. {
  256. c->screen = screen_gb_id(screen);
  257. if(getg)
  258. c->flags |= CLIENT_IGNORE_LAYOUT;
  259. client_map(c);
  260. tag_client(tag_gb_id(c->screen, tag), c);
  261. if(getg)
  262. client_moveresize(c, &g);
  263. client_get_name(c);
  264. }
  265. }
  266. }
  267. if(!nscreen)
  268. {
  269. SLIST_FOREACH(s, &W->h.screen, next)
  270. if(!TAILQ_EMPTY(&s->tags))
  271. tag_screen(s, TAILQ_FIRST(&s->tags));
  272. }
  273. else
  274. {
  275. /* Set back selected tag */
  276. for(i = 0; i < nscreen; ++i)
  277. {
  278. s = screen_gb_id(i);
  279. tag_screen(s, tag_gb_id(s, tret[i]));
  280. }
  281. }
  282. /* Re-adjust tabbed clients */
  283. SLIST_FOREACH(c, &W->h.client, next)
  284. if((cc = client_gb_win(c->tmp)) && cc != c)
  285. _client_tab(c, cc);
  286. if((fc = client_gb_win(focus)) && fc != W->client)
  287. client_focus(fc);
  288. SLIST_FOREACH(c, &W->h.client, next)
  289. layout_fix_hole(c);
  290. W->flags &= ~WMFS_SCAN;
  291. if(tret)
  292. XFree(tret);
  293. XFree(w);
  294. XSync(W->dpy, false);
  295. }
  296. static inline void
  297. wmfs_sigchld(void)
  298. {
  299. if(W->flags & WMFS_SIGCHLD)
  300. {
  301. while(waitpid(-1, NULL, WNOHANG) > 0);
  302. W->flags ^= WMFS_SIGCHLD;
  303. }
  304. }
  305. static void
  306. wmfs_loop(void)
  307. {
  308. XEvent ev;
  309. while((W->flags & WMFS_RUNNING) && !XNextEvent(W->dpy, &ev))
  310. {
  311. /* Manage SIGCHLD event here, X is not safe with it */
  312. wmfs_sigchld();
  313. EVENT_HANDLE(&ev);
  314. }
  315. }
  316. static inline void
  317. wmfs_init(void)
  318. {
  319. log_init();
  320. wmfs_xinit();
  321. ewmh_init();
  322. screen_init();
  323. event_init();
  324. config_init();
  325. }
  326. void
  327. wmfs_quit(void)
  328. {
  329. struct keybind *k;
  330. struct rule *r;
  331. struct theme *t;
  332. struct client *c;
  333. struct mousebind *m;
  334. struct launcher *l;
  335. ewmh_update_wmfs_props();
  336. XFreeGC(W->dpy, W->rgc);
  337. while(!SLIST_EMPTY(&W->h.client))
  338. {
  339. c = SLIST_FIRST(&W->h.client);
  340. client_update_props(c, CPROP_LOC | CPROP_FLAG | CPROP_GEO);
  341. c->flags |= (CLIENT_IGNORE_LAYOUT | CLIENT_REMOVEALL);
  342. client_remove(c);
  343. }
  344. /* Will free:
  345. *
  346. * Screens -> tags
  347. * -> Infobars -> Elements
  348. */
  349. screen_free();
  350. /* Conf stuffs */
  351. while(!SLIST_EMPTY(&W->h.theme))
  352. {
  353. t = SLIST_FIRST(&W->h.theme);
  354. SLIST_REMOVE_HEAD(&W->h.theme, next);
  355. XFreeFontSet(W->dpy, t->font.fontset);
  356. status_free_ctx(&t->tags_n_sl);
  357. status_free_ctx(&t->tags_s_sl);
  358. status_free_ctx(&t->tags_o_sl);
  359. status_free_ctx(&t->tags_u_sl);
  360. status_free_ctx(&t->client_n_sl);
  361. status_free_ctx(&t->client_s_sl);
  362. free(t);
  363. }
  364. while(!SLIST_EMPTY(&W->h.keybind))
  365. {
  366. k = SLIST_FIRST(&W->h.keybind);
  367. SLIST_REMOVE_HEAD(&W->h.keybind, next);
  368. free((void*)k->cmd);
  369. free(k);
  370. }
  371. while(!SLIST_EMPTY(&W->h.mousebind))
  372. {
  373. m = SLIST_FIRST(&W->h.mousebind);
  374. SLIST_REMOVE_HEAD(&W->h.mousebind, globnext);
  375. free((void*)m->cmd);
  376. free(m);
  377. }
  378. while(!SLIST_EMPTY(&W->h.launcher))
  379. {
  380. l = SLIST_FIRST(&W->h.launcher);
  381. SLIST_REMOVE_HEAD(&W->h.launcher, next);
  382. free((void*)l->name);
  383. free((void*)l->prompt);
  384. free((void*)l->command);
  385. free(l);
  386. }
  387. while(!SLIST_EMPTY(&W->h.rule))
  388. {
  389. r = SLIST_FIRST(&W->h.rule);
  390. SLIST_REMOVE_HEAD(&W->h.rule, next);
  391. free(r->class);
  392. free(r->instance);
  393. free(r->role);
  394. free(r->name);
  395. free(r);
  396. }
  397. /* close log */
  398. if(W->log)
  399. fclose(W->log), W->log = NULL;
  400. W->flags &= ~WMFS_RUNNING;
  401. }
  402. /** Reload WMFS binary
  403. */
  404. void
  405. uicb_reload(Uicb cmd)
  406. {
  407. (void)cmd;
  408. W->flags &= ~WMFS_RUNNING;
  409. W->flags |= WMFS_RELOAD;
  410. }
  411. void
  412. uicb_quit(Uicb cmd)
  413. {
  414. (void)cmd;
  415. W->flags &= ~WMFS_RUNNING;
  416. }
  417. static void
  418. exec_uicb_function(Display *dpy, Window root, char *func, char *cmd)
  419. {
  420. Atom utf8s = XInternAtom(dpy, "UTF8_STRING", false);
  421. XClientMessageEvent e = {
  422. .type = ClientMessage,
  423. .message_type = XInternAtom(dpy, "_WMFS_FUNCTION", false),
  424. .window = root,
  425. .format = 32,
  426. .data.l[4] = true
  427. };
  428. XChangeProperty(dpy,root, XInternAtom(dpy, "_WMFS_FUNCTION", false), utf8s,
  429. 8, PropModeReplace, (unsigned char*)func, strlen(func));
  430. if(!cmd)
  431. cmd = "";
  432. XChangeProperty(dpy, root, XInternAtom(dpy, "_WMFS_CMD", false), utf8s,
  433. 8, PropModeReplace, (unsigned char*)cmd, strlen(cmd));
  434. XSendEvent(dpy, root, false, StructureNotifyMask, (XEvent*)&e);
  435. XSync(dpy, False);
  436. }
  437. static void
  438. signal_handle(int sig)
  439. {
  440. switch (sig)
  441. {
  442. case SIGQUIT:
  443. case SIGTERM:
  444. W->flags &= ~WMFS_RUNNING;
  445. break;
  446. case SIGCHLD:
  447. W->flags |= WMFS_SIGCHLD;
  448. break;
  449. }
  450. }
  451. int
  452. main(int argc, char **argv)
  453. {
  454. int i;
  455. bool r;
  456. Display *dpy;
  457. char path[MAX_PATH_LEN] = { 0 };
  458. struct sigaction sa;
  459. (void)argc;
  460. sprintf(path, "%s/"CONFIG_DEFAULT_PATH, getenv("HOME"));
  461. /* Opt */
  462. while((i = getopt(argc, argv, "hvC:c:")) != -1)
  463. {
  464. switch(i)
  465. {
  466. default:
  467. case 'h':
  468. printf("usage: %s [-hv] [-c <func> <cmd] [-C <file>]\n"
  469. " -h Show this page\n"
  470. " -v Show WMFS version\n"
  471. " -c <func> <cmd> Execute a specified UICB function\n"
  472. " -C <file> Launch WMFS with a specified configuration file\n", argv[0]);
  473. exit(EXIT_SUCCESS);
  474. break;
  475. case 'v':
  476. printf("wmfs("WMFS_VERSION") 2 beta\n");
  477. exit(EXIT_SUCCESS);
  478. break;
  479. case 'c':
  480. if(!(dpy = XOpenDisplay(NULL)))
  481. {
  482. fprintf(stderr, "%s: Can't open X server\n", argv[0]);
  483. exit(EXIT_FAILURE);
  484. }
  485. exec_uicb_function(dpy, DefaultRootWindow(dpy), optarg, argv[optind]);
  486. XCloseDisplay(dpy);
  487. exit(EXIT_SUCCESS);
  488. break;
  489. case 'C':
  490. strncpy(path, optarg, sizeof(W->confpath));
  491. break;
  492. }
  493. }
  494. W = (struct wmfs*)xcalloc(1, sizeof(struct wmfs));
  495. /* Default path ~/.config/wmfs/wmfsrc */
  496. W->confpath = path;
  497. /* Get X display */
  498. if(!(W->dpy = XOpenDisplay(NULL)))
  499. {
  500. fprintf(stderr, "%s: Can't open X server\n", argv[0]);
  501. exit(EXIT_FAILURE);
  502. }
  503. /* Set signal handler */
  504. memset(&sa, 0, sizeof(sa));
  505. sa.sa_handler = signal_handle;
  506. sigemptyset(&sa.sa_mask);
  507. sigaction(SIGQUIT, &sa, NULL);
  508. sigaction(SIGTERM, &sa, NULL);
  509. sigaction(SIGCHLD, &sa, NULL);
  510. /* Core */
  511. wmfs_init();
  512. wmfs_scan();
  513. wmfs_loop();
  514. wmfs_quit();
  515. r = (W->flags & WMFS_RELOAD);
  516. free(W);
  517. if(r)
  518. execvp(argv[0], argv);
  519. XCloseDisplay(W->dpy);
  520. return 1;
  521. }