Advertisement
Guest User

g15macro.c

a guest
Apr 11th, 2013
190
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C 31.83 KB | None | 0 0
  1. /*
  2.         This file is part of g15daemon.
  3.  
  4.         g15daemon is free software; you can redistribute it and/or modify
  5.         it under the terms of the GNU General Public License as published by
  6.         the Free Software Foundation; either version 2 of the License, or
  7.         (at your option) any later version.
  8.  
  9.         g15daemon is distributed in the hope that it will be useful,
  10.         but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.         MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12.         GNU General Public License for more details.
  13.  
  14.         You should have received a copy of the GNU General Public License
  15.         along with g15daemon; if not, write to the Free Software
  16.         Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  17.  
  18.         (c) 2006-2007 Mike Lampard
  19.  
  20.         $Revision: 445 $ -  $Date: 2008-01-23 18:56:23 +1030 (Wed, 23 Jan 2008) $ $Author: mlampard $
  21.  
  22.         This daemon listens on localhost port 15550 for client connections,
  23.         and arbitrates LCD display.  Allows for multiple simultaneous clients.
  24.         Client screens can be cycled through by pressing the 'L1' key.
  25.  
  26.         This is a macro recorder and playback utility for the G15 and g15daemon.
  27. */
  28.  
  29. #include <unistd.h>
  30. #include <stdio.h>
  31. #include <stdlib.h>
  32. #include <string.h>
  33. #include <sys/types.h>
  34. #include <sys/stat.h>
  35. #include <fcntl.h>
  36. #include <poll.h>
  37. #include <arpa/inet.h>
  38. #include <signal.h>
  39. #include <pwd.h>
  40. #include <pthread.h>
  41. #include <sys/time.h>
  42. #include <config.h>
  43. #include <X11/Xlib.h>
  44. #include <stdarg.h>
  45. #ifdef HAVE_X11_EXTENSIONS_XTEST_H
  46. #include <X11/extensions/XTest.h>
  47. #endif
  48. #include <X11/XF86keysym.h>
  49.  
  50. #include <g15daemon_client.h>
  51. #include <libg15.h>
  52. #include <libg15render.h>
  53. #include "config.h"
  54.  
  55. #define  XK_MISCELLANY
  56. #define XK_LATIN1
  57. #define XK_LATIN2
  58. #include <X11/keysymdef.h>
  59.        
  60. int g15screen_fd;
  61. int config_fd = 0;
  62. g15canvas *canvas;
  63.  
  64. static Display *dpy;
  65. static Window root_win;
  66.  
  67. pthread_mutex_t x11mutex;
  68. pthread_mutex_t config_mutex;
  69.  
  70. int leaving = 0;
  71. int display_timeout=500;
  72. int have_xtest = False;
  73. int debug = 0;
  74.  
  75. unsigned char recstring[1024];
  76.  
  77. static unsigned int mled_state = G15_LED_M1;
  78. static int mkey_state = 0;
  79. static int recording = 0;
  80.  
  81. typedef struct keypress_s {
  82.     unsigned long keycode;
  83.     unsigned long time_ms;
  84.     unsigned char pressed;
  85.     unsigned long modifiers;
  86.     unsigned int mouse_x;
  87.     unsigned int mouse_y;
  88.     unsigned int buttons;
  89. }keypress_t;
  90.  
  91.  
  92. #define MAX_KEYSTEPS 1024
  93.  
  94. typedef struct keysequence_s {
  95.     keypress_t recorded_keypress[MAX_KEYSTEPS];
  96.     unsigned int record_steps;
  97. } keysequence_t;
  98.  
  99. typedef struct gkeys_s{
  100.     unsigned int recorded;
  101.     keysequence_t keysequence;
  102. }gkeys_t;
  103.  
  104. typedef struct mstates_s {
  105.     gkeys_t gkeys[18];
  106. }mstates_t;
  107.  
  108. mstates_t *mstates[3];
  109.  
  110. struct current_recording {
  111.     keypress_t recorded_keypress[MAX_KEYSTEPS];
  112. }current_recording;
  113.  
  114. unsigned int rec_index=0;
  115.  
  116. const char *gkeystring[] = { "G1","G2","G3","G4","G5","G6","G7","G8","G9","G10","G11","G12","G13","G14","G15","G16","G17","G18","Unknown" };
  117.  
  118. const int gkeycodes[] = { 177,152,190,208,129,130,231,209,210,136,220,143,246,251,137,138,133,183 };
  119.  
  120. const int mmedia_codes[] = {164, 162, 144, 153, 174, 176};
  121.  
  122. const long mmedia_defaults[] = {
  123.     XF86XK_AudioStop,
  124.     XF86XK_AudioPlay,
  125.     XF86XK_AudioPrev,
  126.     XF86XK_AudioNext,
  127.     XF86XK_AudioLowerVolume,
  128.     XF86XK_AudioRaiseVolume
  129. };
  130. /* because this is an X11 client, we can work around the kernel limitations on key numbers */
  131. const long gkeydefaults[] = {
  132.     /* M1 palette */
  133.     XF86XK_Launch4,
  134.     XF86XK_Launch5,
  135.     XF86XK_Launch6,
  136.     XF86XK_Launch7,
  137.     XF86XK_Launch8,
  138.     XF86XK_Launch9,
  139.     XF86XK_LaunchA,
  140.     XF86XK_LaunchB,
  141.     XF86XK_LaunchC,  
  142.     XF86XK_LaunchD,
  143.     XF86XK_LaunchE,
  144.     XF86XK_LaunchF,
  145.     XF86XK_iTouch,
  146.     XF86XK_Calculater,
  147.     XF86XK_Support,
  148.     XF86XK_Word,
  149.     XF86XK_Messenger,
  150.     XF86XK_WebCam,
  151.     /* M2 palette */
  152.     XK_F13,
  153.     XK_F14,
  154.     XK_F15,
  155.     XK_F16,
  156.     XK_F17,
  157.     XK_F18,
  158.     XK_F19,
  159.     XK_F20,
  160.     XK_F21,
  161.     XK_F22,
  162.     XK_F23,
  163.     XK_F24,
  164.     XK_F25,
  165.     XK_F26,
  166.     XK_F27,
  167.     XK_F28,
  168.     XK_F29,
  169.     XK_F30,
  170.     /* M3 palette */
  171.     XK_Tcedilla,
  172.     XK_racute,
  173.     XK_abreve,
  174.     XK_lacute,
  175.     XK_cacute,
  176.     XK_ccaron,
  177.     XK_eogonek,
  178.     XK_ecaron,
  179.     XK_dcaron,
  180.     XK_dstroke,
  181.     XK_nacute,
  182.     XK_ncaron,
  183.     XK_odoubleacute,
  184.     XK_udoubleacute,
  185.     XK_rcaron,
  186.     XK_uring,
  187.     XK_scaron,
  188.     XK_abovedot
  189. };
  190.  
  191. int map_gkey(keystate){
  192.     int retval = -1;
  193.     switch(keystate){
  194.         case G15_KEY_G1:  retval = 0;   break;
  195.         case G15_KEY_G2:  retval = 1;   break;
  196.         case G15_KEY_G3:  retval = 2;   break;
  197.         case G15_KEY_G4:  retval = 3;   break;
  198.         case G15_KEY_G5:  retval = 4;   break;
  199.         case G15_KEY_G6:  retval = 5;   break;
  200.         case G15_KEY_G7:  retval = 6;   break;
  201.         case G15_KEY_G8:  retval = 7;   break;
  202.         case G15_KEY_G9:  retval = 8;   break;
  203.         case G15_KEY_G10: retval = 9;   break;
  204.         case G15_KEY_G11: retval = 10;   break;
  205.         case G15_KEY_G12: retval = 11;   break;
  206.         case G15_KEY_G13: retval = 12;   break;
  207.         case G15_KEY_G14: retval = 13;   break;
  208.         case G15_KEY_G15: retval = 14;   break;
  209.         case G15_KEY_G16: retval = 15;   break;
  210.         case G15_KEY_G17: retval = 16;   break;
  211.         case G15_KEY_G18: retval = 17;   break;
  212.     }
  213.     return retval;
  214. }
  215.  
  216. /* debugging wrapper */
  217. static int g15macro_log (const char *fmt, ...) {
  218.  
  219.     if (debug) {
  220.         printf("G15Macro: ");
  221.         va_list argp;
  222.         va_start (argp, fmt);
  223.         vprintf(fmt,argp);
  224.         va_end (argp);
  225.     }
  226.  
  227.     return 0;
  228. }
  229.  
  230. void fake_keyevent(int keycode,int keydown,unsigned long modifiers){
  231.   if(have_xtest && !recording) {
  232.     #ifdef HAVE_X11_EXTENSIONS_XTEST_H        
  233.     pthread_mutex_lock(&x11mutex);
  234.        XTestFakeKeyEvent(dpy, keycode,keydown, CurrentTime);
  235.        XSync(dpy,False);
  236.     pthread_mutex_unlock(&x11mutex);
  237.     usleep(1500);
  238.      #endif        
  239.   } else {
  240.     XKeyEvent event;
  241.     Window current_focus;
  242.     int dummy = 0;
  243.     int key = 0;
  244.  
  245.     pthread_mutex_lock(&x11mutex);
  246.       XGetInputFocus(dpy,&current_focus, &dummy);
  247.       key = XKeycodeToKeysym(dpy,keycode,0);
  248.       if(keydown)
  249.         event.type=KeyPress;
  250.       else
  251.         event.type=KeyRelease;
  252.  
  253.       event.keycode = keycode;
  254.       event.serial = 0;
  255.       event.send_event = False;
  256.       event.display = dpy;
  257.       event.x = event.y = event.x_root = event.y_root = 0;
  258.       event.time = CurrentTime;
  259.       event.same_screen = True;
  260.       event.subwindow = None;
  261.       event.window = current_focus;
  262.       event.root = root_win;
  263.       event.state = modifiers;
  264.       XSendEvent(dpy,current_focus,False,0,(XEvent*)&event);
  265.       XSync(dpy,False);
  266.     pthread_mutex_unlock(&x11mutex);
  267.   }
  268. }
  269.  
  270. void record_cleanup(){
  271.     g15_send_cmd (g15screen_fd,G15DAEMON_MKEYLEDS,mled_state);
  272.     memset(recstring,0,strlen((char*)recstring));
  273.     rec_index = 0;
  274.     recording = 0;
  275.     pthread_mutex_lock(&x11mutex);
  276.     XUngrabKeyboard(dpy,CurrentTime);
  277.     XFlush(dpy);
  278.     pthread_mutex_unlock(&x11mutex);
  279. }
  280.  
  281. void record_cancel(){
  282.     memset(canvas->buffer,0,G15_BUFFER_LEN);
  283.     g15r_renderString (canvas, (unsigned char *)"Recording", 0, G15_TEXT_LARGE, 80-((strlen("Recording")/2)*8), 4);
  284.     g15r_renderString (canvas, (unsigned char *)"Canceled", 0, G15_TEXT_LARGE, 80-((strlen("Canceled")/2)*8), 18);
  285.     g15_send(g15screen_fd,(char *)canvas->buffer,G15_BUFFER_LEN);
  286.     record_cleanup();
  287. }
  288.  
  289. void record_complete(unsigned long keystate)
  290. {
  291.     char tmpstr[1024];
  292.     int gkey = map_gkey(keystate);
  293.  
  294.     pthread_mutex_lock(&config_mutex);
  295.    
  296.     if(!rec_index) // nothing recorded - delete prior recording
  297.         memset(mstates[mkey_state]->gkeys[gkey].keysequence.recorded_keypress,0,sizeof(keysequence_t));
  298.     else
  299.         memcpy(mstates[mkey_state]->gkeys[gkey].keysequence.recorded_keypress, &current_recording, sizeof(keysequence_t));
  300.  
  301.     mstates[mkey_state]->gkeys[gkey].keysequence.record_steps=rec_index;
  302.     pthread_mutex_unlock(&config_mutex);
  303.    
  304.     memset(canvas->buffer,0,G15_BUFFER_LEN);
  305.     if(rec_index){
  306.         strcpy(tmpstr,"For key ");
  307.         strcat(tmpstr,gkeystring[map_gkey(keystate)]);
  308.         g15macro_log("Recording Complete %s\n",tmpstr);
  309.         g15r_renderString (canvas, (unsigned char *)"Recording", 0, G15_TEXT_LARGE, 80-((strlen("Recording")/2)*8), 4);
  310.         g15r_renderString (canvas, (unsigned char *)"Complete", 0, G15_TEXT_LARGE, 80-((strlen("Complete")/2)*8), 18);
  311.  
  312.     }else{
  313.         strcpy(tmpstr,"From Key ");
  314.         strcat(tmpstr,gkeystring[map_gkey(keystate)]);
  315.         g15macro_log("Macro deleted %s\n",tmpstr);
  316.         g15r_renderString (canvas, (unsigned char *)"Macro", 0, G15_TEXT_LARGE, 80-((strlen("Macro")/2)*8), 4);
  317.         g15r_renderString (canvas, (unsigned char *)"Deleted", 0, G15_TEXT_LARGE, 80-((strlen("Deleted")/2)*8), 18);
  318.     }
  319.     g15r_renderString (canvas, (unsigned char *)tmpstr, 0, G15_TEXT_LARGE, 80-((strlen(tmpstr)/2)*8), 32);
  320.    
  321.     g15_send(g15screen_fd,(char *)canvas->buffer,G15_BUFFER_LEN);
  322.  
  323.     record_cleanup();
  324. }
  325.  
  326. int calc_mkey_offset() {
  327.         int mkey_offset=0;
  328.         switch(mkey_state){
  329.           case 0:
  330.             mkey_offset = 0;
  331.             break;
  332.           case 1:
  333.             mkey_offset = 18;
  334.             break;
  335.           case 2:
  336.             mkey_offset = 36;
  337.             break;
  338.           default:
  339.             mkey_offset=0;
  340.         }
  341.       return mkey_offset;
  342. }
  343.  
  344. void macro_playback(unsigned long keystate)
  345. {
  346.     int i = 0;
  347.     KeySym key;
  348.     int keyevent;
  349.     int gkey = map_gkey(keystate);
  350.     if(gkey<0)
  351.       return;
  352.    
  353.     /* if no macro has been recorded for this key, send the g15daemon default keycode */
  354.     if(mstates[mkey_state]->gkeys[gkey].keysequence.record_steps==0){
  355.         int mkey_offset=0;
  356.  
  357.         mkey_offset = calc_mkey_offset();
  358.  
  359.         pthread_mutex_lock(&x11mutex);
  360.         keyevent=XKeysymToKeycode(dpy, gkeydefaults[gkey+mkey_offset]);
  361.         pthread_mutex_unlock(&x11mutex);
  362.  
  363.         fake_keyevent(keyevent,1,None);
  364.         fake_keyevent(keyevent,0,None);
  365.         g15macro_log("Key: \t%s\n",XKeysymToString(gkeydefaults[gkey+mkey_offset]));
  366.         return;
  367.     }
  368.     g15macro_log("Macro Playback: for key %s\n",gkeystring[gkey]);
  369.     pthread_mutex_lock(&config_mutex);
  370.     for(i=0;i<mstates[mkey_state]->gkeys[gkey].keysequence.record_steps;i++){
  371.  
  372.         fake_keyevent(mstates[mkey_state]->gkeys[gkey].keysequence.recorded_keypress[i].keycode,
  373.                           mstates[mkey_state]->gkeys[gkey].keysequence.recorded_keypress[i].pressed,
  374.                           mstates[mkey_state]->gkeys[gkey].keysequence.recorded_keypress[i].modifiers);
  375.  
  376.         pthread_mutex_lock(&x11mutex);
  377.         key = XKeycodeToKeysym(dpy,mstates[mkey_state]->gkeys[gkey].keysequence.recorded_keypress[i].keycode,0);
  378.         pthread_mutex_unlock(&x11mutex);
  379.         g15macro_log("\t%s %s\n",XKeysymToString(key),mstates[mkey_state]->gkeys[gkey].keysequence.recorded_keypress[i].pressed?"Down":"Up");
  380.        
  381.         switch (key) {
  382.             case XK_Control_L:
  383.             case XK_Control_R:
  384.             case XK_Meta_L:
  385.             case XK_Meta_R:
  386.             case XK_Alt_L:
  387.             case XK_Alt_R:
  388.             case XK_Super_L:
  389.             case XK_Super_R:
  390.             case XK_Hyper_L:
  391.             case XK_Hyper_R:
  392.              usleep(mstates[mkey_state]->gkeys[gkey].keysequence.recorded_keypress[i].time_ms*1000);
  393.               break;
  394.             default:
  395.              usleep(1000);
  396.         }
  397.     }
  398.     pthread_mutex_unlock(&config_mutex);
  399.     g15macro_log("Macro Playback Complete\n");
  400. }
  401.  
  402. /* WARNING:  uses global mkey state */
  403. void dump_config(FILE *configfile)
  404. {
  405.     int i=0,gkey=0;
  406.     KeySym key;
  407.     pthread_mutex_lock(&config_mutex);
  408.     int orig_mkeystate=mkey_state;
  409.     for(mkey_state=0;mkey_state<3;mkey_state++){
  410.       fprintf(configfile,"\n\nCodes for MKey %i\n",mkey_state+1);
  411.       for(gkey=0;gkey<18;gkey++){
  412.         fprintf(configfile,"Key %s:",gkeystring[gkey]);
  413.         /* if no macro has been recorded for this key, dump the g15daemon default keycode */
  414.         if(mstates[mkey_state]->gkeys[gkey].keysequence.record_steps==0){
  415.           int mkey_offset=0;
  416.           mkey_offset = calc_mkey_offset();
  417.           fprintf(configfile,"\t%s\n",XKeysymToString(gkeydefaults[gkey+mkey_offset]));
  418.         }else{
  419.           fprintf(configfile,"\n");
  420.           for(i=0;i<mstates[mkey_state]->gkeys[gkey].keysequence.record_steps;i++){
  421.             key = XKeycodeToKeysym(dpy,mstates[mkey_state]->gkeys[gkey].keysequence.recorded_keypress[i].keycode,0);
  422.             fprintf(configfile,"\t%s %s %u\n",XKeysymToString(key),mstates[mkey_state]->gkeys[gkey].keysequence.recorded_keypress[i].pressed?"Down":"Up",(unsigned int)mstates[mkey_state]->gkeys[gkey].keysequence.recorded_keypress[i].modifiers);
  423.         }
  424.       }
  425.      }
  426.     }
  427.     mkey_state=orig_mkeystate;
  428.     pthread_mutex_unlock(&config_mutex);
  429. }
  430.  
  431. void save_macros(char *filename){
  432.   FILE *configfile;
  433.   configfile=fopen(filename,"w");
  434.  
  435.   dump_config(configfile);
  436.  
  437.   fclose(configfile);
  438. }
  439.  
  440. void restore_config(char *filename) {
  441.   FILE *f;
  442.   char tmpstring[1024];
  443.   unsigned int key=0;
  444.   unsigned int mkey=0;
  445.   unsigned int i=0;
  446.   unsigned int keycode;
  447.   f=fopen(filename,"r");
  448.   printf("restoring codes\n");
  449.   pthread_mutex_lock(&config_mutex);
  450.   do{
  451.     memset(tmpstring,0,1024);
  452.     fgets(tmpstring,1024,f);
  453.  
  454.     if(tmpstring[0]=='C'){
  455.       sscanf(tmpstring,"Codes for MKey %i\n",&mkey);
  456.       mkey--;
  457.       i=0;
  458.     }
  459.     if(tmpstring[0]=='K'){
  460.       sscanf(tmpstring,"Key G%i:",&key);
  461.       key--;
  462.       i=0;
  463.     }
  464.     if(tmpstring[0]=='\t'){
  465.       char codestr[64];
  466.       char pressed[20];
  467.       unsigned int modifiers = 0;
  468.       sscanf(tmpstring,"\t%s %s %i\n",(char*)&codestr,(char*)&pressed,&modifiers);
  469.       keycode = XKeysymToKeycode(dpy,XStringToKeysym(codestr));
  470.       mstates[mkey]->gkeys[key].keysequence.recorded_keypress[i].keycode = keycode;
  471.       mstates[mkey]->gkeys[key].keysequence.recorded_keypress[i].pressed = strncmp(pressed,"Up",2)?1:0;
  472.       mstates[mkey]->gkeys[key].keysequence.recorded_keypress[i].modifiers = modifiers;
  473.       mstates[mkey]->gkeys[key].keysequence.record_steps=++i;    
  474.     }
  475.   }  while(!feof(f));
  476.   pthread_mutex_unlock(&config_mutex);
  477.   fclose(f);
  478. }
  479.  
  480. void change_keymap(int offset){
  481.     int i=0,j=0;
  482.     pthread_mutex_lock(&x11mutex);
  483.     for(i=offset;i<offset+18;i++,j++)
  484.     {
  485.       KeySym newmap[1];
  486.       newmap[0]=gkeydefaults[i];
  487.       XChangeKeyboardMapping (dpy, gkeycodes[j], 1, newmap, 1);
  488.     }
  489.     XFlush(dpy);
  490.     pthread_mutex_unlock(&x11mutex);
  491.  
  492. }
  493.  
  494. /* ensure that the multimedia keys are configured */
  495. void configure_mmediakeys(){
  496.  
  497.    KeySym newmap[1];
  498.    int i=0;
  499.    pthread_mutex_lock(&x11mutex);
  500.    for(i=0;i<6;i++){
  501.      newmap[0]=mmedia_defaults[i];
  502.      XChangeKeyboardMapping (dpy, mmedia_codes[i], 1, newmap, 1);
  503.    }
  504.    XFlush(dpy);
  505.    pthread_mutex_unlock(&x11mutex);
  506.    
  507. }
  508.  
  509. void handle_mkey_switch(unsigned int mkey) {
  510.     int mkey_offset = 0;  
  511.     switch(mkey) {
  512.       case G15_KEY_M1:
  513.         mled_state=G15_LED_M1;
  514.         mkey_state=0;  
  515.         break;
  516.       case G15_KEY_M2:
  517.         mled_state=G15_LED_M2;  
  518.         mkey_state=1;
  519.         break;
  520.       case G15_KEY_M3:
  521.         mled_state=G15_LED_M3;  
  522.         mkey_state=2;
  523.         break;
  524.     }
  525.     mkey_offset = calc_mkey_offset();
  526.     if(recording)
  527.       record_cancel();
  528.     g15_send_cmd (g15screen_fd,G15DAEMON_MKEYLEDS,mled_state);
  529.     change_keymap(mkey_offset);
  530. }
  531.  
  532. void *Lkeys_thread() {
  533.     unsigned long keystate = 0;
  534.     struct pollfd fds;
  535.     char ver[5];
  536.     int foo = 0;
  537.     strncpy(ver,G15DAEMON_VERSION,3);
  538.     float g15v;
  539.     sscanf(ver,"%f",&g15v);
  540.  
  541.     g15macro_log("Using version %.2f as keypress protocol\n",g15v);
  542.  
  543.     while(!leaving){
  544.  
  545.         /* g15daemon series 1.2 need key request packets */
  546.         if((g15v*10)<=18) {
  547.             keystate = g15_send_cmd (g15screen_fd, G15DAEMON_GET_KEYSTATE, foo);
  548.         } else {
  549.             fds.fd = g15screen_fd;
  550.             fds.events = POLLIN;
  551.             fds.revents=0;
  552.             keystate=0;
  553.             if ((poll(&fds, 1, 1000)) > 0) {
  554.                 read (g15screen_fd, &keystate, sizeof (keystate));    
  555.             }
  556.         }
  557.  
  558.         if (keystate!=0)
  559.         {
  560.             g15macro_log("Received Keystate == %lu\n",keystate);
  561.                      
  562.             switch (keystate)
  563.             {
  564.                 case G15_KEY_L5:{
  565.                     int fg_check = g15_send_cmd (g15screen_fd, G15DAEMON_IS_FOREGROUND, foo);
  566.                     if(fg_check)
  567.                       leaving = 1;
  568.                     break;
  569.                 }
  570.                 case G15_KEY_MR: {
  571.                     if(!recording) {
  572.                       if(0==g15_send_cmd (g15screen_fd, G15DAEMON_IS_FOREGROUND, foo)){
  573.                         usleep(1000);
  574.                         g15_send_cmd (g15screen_fd, G15DAEMON_SWITCH_PRIORITIES, foo);
  575.                         g15macro_log("Switching to LCD foreground\n");
  576.                       }
  577.                       usleep(1000);
  578.                       g15_send_cmd (g15screen_fd,G15DAEMON_MKEYLEDS, G15_LED_MR | mled_state);
  579.                       g15r_initCanvas (canvas);
  580.                       g15r_renderString (canvas, (unsigned char *)"Recording", 0, G15_TEXT_LARGE, 80-((strlen("Recording")/2)*8), 1);
  581.                       g15_send(g15screen_fd,(char *)canvas->buffer,G15_BUFFER_LEN);
  582.                       g15macro_log("Recording Enabled\n");
  583.                       recording = 1;
  584.                       pthread_mutex_lock(&x11mutex);
  585.                       XGrabKeyboard(dpy, root_win, True, GrabModeAsync, GrabModeAsync, CurrentTime);
  586.                       pthread_mutex_unlock(&x11mutex);
  587.                       memset(&current_recording,0,sizeof(current_recording));
  588.                     } else {
  589.                       record_cancel();
  590.                     }
  591.                     break;
  592.                   }
  593.                 case G15_KEY_M1:
  594.                     handle_mkey_switch(G15_KEY_M1);
  595.                     break;
  596.                 case G15_KEY_M2:
  597.                     handle_mkey_switch(G15_KEY_M2);
  598.                     break;
  599.                 case G15_KEY_M3:
  600.                     handle_mkey_switch(G15_KEY_M3);
  601.                     break;
  602.                 default:
  603.                     if(keystate >=G15_KEY_G1 && keystate <=G15_KEY_G18){
  604.                         if(recording==1){
  605.                             record_complete(keystate);
  606.                         } else {
  607.                             macro_playback(keystate);
  608.                         }
  609.                     }
  610.                     break;
  611.             }
  612.             keystate = 0;
  613.         }
  614.     }
  615.     return NULL;
  616. }
  617.  
  618. unsigned int g15daemon_gettime_ms(){
  619.     struct timeval tv;
  620.     gettimeofday(&tv,NULL);
  621.     return (tv.tv_sec*1000+tv.tv_usec/1000);
  622. }
  623.  
  624. void xkey_handler(XEvent *event) {
  625.  
  626.     static unsigned long lasttime;
  627.     unsigned char keytext[256];
  628.     unsigned int keycode = event->xkey.keycode;
  629.     int press = True;
  630.  
  631.     if(event->type==KeyRelease){ // we only do keyreleases for some keys
  632.       pthread_mutex_lock(&x11mutex);
  633.         KeySym key = XKeycodeToKeysym(dpy, keycode, 0);
  634.       pthread_mutex_unlock(&x11mutex);
  635.         switch (key) {
  636.             case XK_Shift_L:
  637.             case XK_Shift_R:
  638.             case XK_Control_L:
  639.             case XK_Control_R:
  640.             case XK_Caps_Lock:
  641.             case XK_Shift_Lock:
  642.             case XK_Meta_L:
  643.             case XK_Meta_R:
  644.             case XK_Alt_L:
  645.             case XK_Alt_R:
  646.             case XK_Super_L:
  647.             case XK_Super_R:
  648.             case XK_Hyper_L:
  649.             case XK_Hyper_R:
  650.             default:
  651.                 press = False;
  652.         }
  653.     }
  654.     if(recording){
  655.         current_recording.recorded_keypress[rec_index].keycode = keycode;
  656.         current_recording.recorded_keypress[rec_index].pressed = press;
  657.         current_recording.recorded_keypress[rec_index].modifiers = event->xkey.state;
  658.         if(rec_index==0)
  659.             current_recording.recorded_keypress[rec_index].time_ms=0;
  660.         else
  661.             current_recording.recorded_keypress[rec_index].time_ms=g15daemon_gettime_ms() - lasttime;
  662.         if(rec_index < MAX_KEYSTEPS) {
  663.           rec_index++;
  664.         /* now the default stuff */
  665.         pthread_mutex_lock(&x11mutex);        
  666.           XUngrabKeyboard(dpy,CurrentTime);
  667.        pthread_mutex_unlock(&x11mutex);
  668.  
  669.         fake_keyevent(keycode,press,event->xkey.state);
  670.        
  671.         pthread_mutex_lock(&x11mutex);
  672.         XGrabKeyboard(dpy, root_win, True, GrabModeAsync, GrabModeAsync, CurrentTime);
  673.         XFlush(dpy);
  674.         strcpy((char*)keytext,XKeysymToString(XKeycodeToKeysym(dpy, keycode, 0)));
  675.         pthread_mutex_unlock(&x11mutex);        
  676.         if(0==strcmp((char*)keytext,"space"))
  677.             strcpy((char*)keytext," ");
  678.         if(0==strcmp((char*)keytext,"period"))
  679.             strcpy((char*)keytext,".");
  680.         if(press==True){
  681.           strcat((char*)recstring,(char*)keytext);
  682.           g15macro_log("Adding %s to Macro\n",keytext);
  683.           g15r_renderString (canvas, (unsigned char *)recstring, 0, G15_TEXT_MED, 80-((strlen((char*)recstring)/2)*5), 22);
  684.           g15_send(g15screen_fd,(char *)canvas->buffer,G15_BUFFER_LEN);
  685.         }
  686.       } else {
  687.           pthread_mutex_lock(&x11mutex);        
  688.             XUngrabKeyboard(dpy,CurrentTime);
  689.           pthread_mutex_unlock(&x11mutex);
  690.           recording = 0;
  691.           rec_index = 0;
  692.         }
  693.  
  694.     }else
  695.         rec_index=0;
  696.  
  697.         lasttime = g15daemon_gettime_ms();
  698. }
  699.  
  700. static void* xevent_thread()
  701. {
  702.     XEvent event;
  703.     long event_mask = KeyPressMask|KeyReleaseMask|FocusChangeMask|SubstructureNotifyMask;
  704.     int retval=0;
  705.     pthread_mutex_lock(&x11mutex);
  706.     XSelectInput(dpy, root_win, event_mask);
  707.     pthread_mutex_unlock(&x11mutex);
  708.     while(!leaving){
  709.         pthread_mutex_lock(&x11mutex);
  710.         memset(&event,0,sizeof(XEvent));
  711.         retval = XCheckMaskEvent(dpy, event_mask, &event);
  712.         pthread_mutex_unlock(&x11mutex);
  713.         if(retval == True){
  714.             switch(event.type) {
  715.                 case KeyPress:
  716.                     xkey_handler(&event);
  717.                     break;
  718.                 case KeyRelease:
  719.                     xkey_handler(&event);
  720.                     break;
  721.                 case FocusIn:
  722.                 case FocusOut:
  723.                 case EnterNotify:
  724.                 case LeaveNotify:
  725.                 case MapNotify:
  726.                 case UnmapNotify:
  727.                 case MapRequest:
  728.                 case ConfigureNotify:
  729.                 case CreateNotify:
  730.                 case DestroyNotify:
  731.                     break;
  732.                 case ReparentNotify: {
  733.                         if(recording) {
  734.                             pthread_mutex_lock(&x11mutex);
  735.                             XGrabKeyboard(dpy, root_win, True, GrabModeAsync, GrabModeAsync, CurrentTime);
  736.                             XFlush(dpy);
  737.                             pthread_mutex_unlock(&x11mutex);
  738.                         }
  739.                         break;
  740.                     }
  741.                 default:
  742.                     g15macro_log("Unhandled event (%i) received\n",event.type);
  743.             }
  744.         }else
  745.         usleep(25000);
  746.     }
  747.     return NULL;
  748. }
  749.  
  750. int myx_error_handler(Display *dpy, XErrorEvent *err){
  751.     return 0;
  752. }
  753.  
  754. void g15macro_sighandler(int sig) {
  755.     switch(sig){
  756.          case SIGINT:
  757.          case SIGTERM:
  758.          case SIGQUIT:
  759.          case SIGPIPE:
  760.               leaving = 1;
  761.                break;
  762.     }
  763. }
  764.  
  765. void helptext() {
  766.   printf("G15Macro %s\n",PACKAGE_VERSION);
  767.   printf("\n--user (-u) \"username\" run as user \"username\"\n");
  768.   printf("--dump (-d) dump current configuration to stdout\n");
  769.   printf("--debug (-g) print debugging information\n");
  770.   printf("--version (-v) print version and exit\n");
  771.   printf("--keysonly (-k) configure multimedia and extra keys then exit\n");
  772.   printf("--help (-h) this help text\n\n");
  773. }
  774.  
  775. int main(int argc, char **argv)
  776. {
  777.     pthread_t Xkeys;
  778.     pthread_t Lkeys;
  779. #ifdef USE_XTEST
  780.     int xtest_major_version = 0;
  781.     int xtest_minor_version = 0;
  782. #endif    
  783.     struct sigaction new_action;
  784.     int dummy=0,i=0;
  785.     unsigned char user[256];
  786.     struct passwd *username;
  787.     char configpath[1024];
  788.     char splashpath[1024];
  789.     unsigned int dump = 0;
  790.     unsigned int keysonly = 0;
  791.     FILE *config;
  792.     unsigned int convert = 0;
  793.     strncpy(configpath,getenv("HOME"),1024);
  794.  
  795.     memset(user,0,256);
  796.     for(i=0;i<argc;i++){
  797.         if (!strncmp(argv[i], "-u",2) || !strncmp(argv[i], "--user",6)) {
  798.            if(argv[i+1]!=NULL){
  799.              strncpy((char*)user,argv[i+1],128);
  800.              i++;
  801.            }
  802.         }
  803.  
  804.         if (!strncmp(argv[i], "-d",2) || !strncmp(argv[i], "--dump",6)) {
  805.           dump = 1;
  806.         }
  807.  
  808.         if (!strncmp(argv[i], "-h",2) || !strncmp(argv[i], "--help",6)) {
  809.           helptext();
  810.           exit(0);
  811.         }
  812.  
  813.         if (!strncmp(argv[i], "-k",2) || !strncmp(argv[i], "--keysonly",10)) {
  814.           keysonly = 1;
  815.         }
  816.  
  817.         if (!strncmp(argv[i], "-g",2) || !strncmp(argv[i], "--debug",7)) {
  818.           printf("Debugging Enabled\n");
  819.           debug = 1;
  820.         }
  821.  
  822.         if (!strncmp(argv[i], "-v",2) || !strncmp(argv[i], "--version",9)) {
  823.           printf("G15Macro version %s\n\n",PACKAGE_VERSION);
  824.           exit(0);
  825.         }
  826.  
  827.     }
  828.  
  829.     if(strlen((char*)user)){
  830.       username = getpwnam((char*)user);
  831.         if (username==NULL) {
  832.             username = getpwuid(geteuid());
  833.             printf("BEWARE: running as effective uid %i\n",username->pw_uid);
  834.         }
  835.         else {
  836.            if(0==setuid(username->pw_uid)) {
  837.              setgid(username->pw_gid);
  838.              strncpy(configpath,username->pw_dir,1024);
  839.              printf("running as user %s\n",username->pw_name);
  840.            }
  841.            else
  842.              printf("Unable to run as user \"%s\" - you dont have permissions for that.\nRunning as \"%s\"\n",username->pw_name,getenv("USER"));
  843.         }
  844.     }
  845.  
  846.     do {
  847.       dpy = XOpenDisplay(getenv("DISPLAY"));
  848.       if (!dpy) {
  849.         printf("Unable to open display %s - retrying\n",getenv("DISPLAY"));
  850.         sleep(2);
  851.         }
  852.     }while(!dpy);
  853.  
  854.     /* completely ignore errors and carry on */
  855.     XSetErrorHandler(myx_error_handler);
  856.     configure_mmediakeys();
  857.     change_keymap(0);
  858.     XFlush(dpy);
  859.    
  860.     if(keysonly>0)
  861.       goto close_and_exit;
  862.  
  863.     /* old binary config format */
  864.     strncat(configpath,"/.g15macro",1024-strlen(configpath));
  865.     strncat(configpath,"/g15macro-data",1024-strlen(configpath));
  866.     config_fd = open(configpath,O_RDONLY|O_SYNC);
  867.      
  868.     mstates[0] = malloc(sizeof(mstates_t));
  869.     mstates[1] = (mstates_t*)malloc(sizeof(mstates_t));
  870.     mstates[2] = (mstates_t*)malloc(sizeof(mstates_t));
  871.                    
  872.     if(config_fd>0) {
  873.         printf("Converting old data\n");
  874.         read(config_fd,mstates[0],sizeof(mstates_t));
  875.         read(config_fd,mstates[1],sizeof(mstates_t));
  876.         read(config_fd,mstates[2],sizeof(mstates_t));
  877.         close(config_fd);
  878.         strncpy(configpath,getenv("HOME"),1024);
  879.         strncat(configpath,"/.g15macro",1024-strlen(configpath));
  880.         char configbak[1024];
  881.         strcpy(configbak,configpath);
  882.         strncat(configpath,"/g15macro-data",1024-strlen(configpath));
  883.         strncat(configbak,"/g15macro-data.old",1024-strlen(configpath));
  884.         rename(configpath,configbak);
  885.         convert = 1;
  886.     }else {
  887.       memset(mstates[0],0,sizeof(mstates));
  888.       memset(mstates[1],0,sizeof(mstates));
  889.       memset(mstates[2],0,sizeof(mstates));
  890.     }
  891.     /* new format */
  892.     strncpy(configpath,getenv("HOME"),1024);
  893.     strncat(configpath,"/.g15macro",1024-strlen(configpath));
  894.     mkdir(configpath,0777);
  895.     strncat(configpath,"/g15macro.conf",1024-strlen(configpath));
  896.     config=fopen(configpath,"a");
  897.     fclose(config);
  898.    
  899.     do {
  900.       if((g15screen_fd = new_g15_screen(G15_G15RBUF))<0){
  901.         printf("Sorry, cant connect to the G15daemon - retrying\n");
  902.         sleep(2);
  903.       }
  904.     }while(g15screen_fd<0);
  905.  
  906.     if(!convert)
  907.       restore_config(configpath);
  908.  
  909.     if(dump){
  910.         printf("G15Macro Dumping Codes...");
  911.         dump_config(stderr);
  912.         exit(0);
  913.     }
  914.      
  915.     g15_send_cmd (g15screen_fd,G15DAEMON_KEY_HANDLER, dummy);
  916.     usleep(1000);
  917.     g15_send_cmd (g15screen_fd,G15DAEMON_MKEYLEDS,mled_state);
  918.     usleep(1000);
  919.     canvas = (g15canvas *) malloc (sizeof (g15canvas));
  920.  
  921.     if (canvas != NULL) {
  922.         g15r_initCanvas(canvas);
  923.     } else {
  924.         printf("Unable to initialise the libg15render canvas\nExiting\n");
  925.         return 1;
  926.     }
  927.  
  928.     root_win = DefaultRootWindow(dpy);
  929.     if (!root_win) {
  930.         printf("Cant find root window\n");
  931.         return 1;
  932.     }
  933.  
  934.     have_xtest = False;
  935. #ifdef HAVE_XTEST
  936. #ifdef USE_XTEST
  937.     have_xtest = XTestQueryExtension(dpy, &dummy, &dummy, &xtest_major_version, &xtest_minor_version);
  938.     if(have_xtest == False || xtest_major_version < 2 || (xtest_major_version <= 2 && xtest_minor_version < 2))
  939.     {
  940.         printf("Warning: XTEST extension not supported by Xserver.  This is not fatal.\nReverting to XSendEvent for keypress emulation\n");
  941.     }
  942. #else
  943.   printf("XTest disabled by configure option.  Using XSendEvent instead.\n");
  944. #endif
  945. #else
  946.   printf("XTest disabled by configure: no devel package was found.  Using XSendEvent instead.\n");
  947. #endif
  948.  
  949.      
  950.     new_action.sa_handler = g15macro_sighandler;
  951.     new_action.sa_flags = 0;
  952.     sigaction(SIGINT, &new_action, NULL);
  953.     sigaction(SIGQUIT, &new_action, NULL);
  954.     sigaction(SIGTERM, &new_action, NULL);
  955.     sigaction(SIGPIPE, &new_action, NULL);
  956.    
  957.     snprintf((char*)splashpath,1024,"%s/%s",DATADIR,"g15macro/splash/g15macro.wbmp");
  958.     g15r_loadWbmpSplash(canvas, splashpath);
  959.     g15_send(g15screen_fd,(char *)canvas->buffer,G15_BUFFER_LEN);
  960.     #ifdef G15DAEMON_NEVER_SELECT
  961.         g15_send_cmd (g15screen_fd, G15DAEMON_NEVER_SELECT, dummy);
  962.     #endif
  963.    
  964.     usleep(1000);
  965.     pthread_mutex_init(&x11mutex,NULL);
  966.     pthread_mutex_init(&config_mutex,NULL);
  967.     pthread_attr_t attr;
  968.     pthread_attr_init(&attr);
  969.     int thread_policy=SCHED_FIFO;
  970.     pthread_attr_setschedpolicy(&attr,thread_policy);      
  971.     pthread_attr_setstacksize(&attr,32*1024); /* set stack to 32k - dont need 8Mb !! */
  972.  
  973.     pthread_create(&Xkeys, &attr, xevent_thread, NULL);
  974.     pthread_create(&Lkeys, &attr, Lkeys_thread, NULL);
  975.     do{
  976.         if(display_timeout<-1)
  977.           display_timeout=-1;
  978.         else
  979.           display_timeout--;
  980.  
  981.         if(recording)
  982.             display_timeout=500;
  983.  
  984.         if(display_timeout<=0){
  985.             int fg_check = g15_send_cmd (g15screen_fd, G15DAEMON_IS_FOREGROUND, dummy);
  986.             if (fg_check==1) { // foreground
  987.                     do {
  988.                       g15_send_cmd (g15screen_fd, G15DAEMON_SWITCH_PRIORITIES, dummy);
  989.                     } while(g15_send_cmd (g15screen_fd, G15DAEMON_IS_FOREGROUND, dummy)==1);
  990.             }
  991.  
  992.            usleep(500*1000);
  993.         }
  994.     }while(!usleep(1000) &&  !leaving);
  995.  
  996.     if(recording){
  997.         recording = 0;
  998.         XUngrabKeyboard(dpy,CurrentTime);
  999.     }
  1000.  
  1001.     save_macros(configpath);
  1002.     g15_send_cmd (g15screen_fd,G15DAEMON_MKEYLEDS,0);
  1003.  
  1004.     pthread_join(Xkeys,NULL);
  1005.     pthread_join(Lkeys,NULL);
  1006.     pthread_mutex_destroy(&x11mutex);
  1007.     pthread_mutex_destroy(&config_mutex);
  1008.     /* revert the keymap to g15daemon default on exit */
  1009.     change_keymap(0);
  1010.     close(g15screen_fd);
  1011. close_and_exit:
  1012.     /*    XCloseDisplay(dpy);  */
  1013.     return 0;
  1014. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement