com32/cmenu/test.c

   1 /* -*- c -*- ------------------------------------------------------------- *
   2  *
   3  *   Copyright 2004-2006 Murali Krishnan Ganapathy - All Rights Reserved
   4  *
   5  *   This program is free software; you can redistribute it and/or modify
   6  *   it under the terms of the GNU General Public License as published by
   7  *   the Free Software Foundation, Inc., 53 Temple Place Ste 330,
   8  *   Boston MA 02111-1307, USA; either version 2 of the License, or
   9  *   (at your option) any later version; incorporated herein by reference.
  10  *
  11  * ----------------------------------------------------------------------- */
  12
  13 #ifndef NULL
  14 #define NULL ((void *) 0)
  15 #endif
  16
  17 #include "menu.h"
  18 #include "help.h"
  19 #include "passwords.h"
  20 #include "com32io.h"
  21 #include <string.h>
  22 #include <stdlib.h>
  23 #include <unistd.h>
  24
  25 #define MAX_CMD_LINE_LENGTH 514
  26
  27 typedef struct s_xtra {
  28   long ipappend; // Stores the ipappend flag to send (useful for PXELINUX only)
  29   char *argsmenu; // Stores the name of menu which contains options for the given RUN item
  30   char *perms; // stores the permissions required to activate the item
  31 } t_xtra;
  32
  33 typedef t_xtra *pt_xtra; // Pointer to extra datastructure
  34
  35 // Types of dot commands for which caller takes responsibility of handling
  36 // In some case some commands may not make sense, it is up to the caller
  37 // to handle cases which do not make sense
  38 typedef enum {QUIT_CMD, REPEAT_CMD, ENTER_CMD, ESCAPE_CMD} t_dotcmd;
  39
  40
  41 /*----------------- Global Variables */
  42
  43 // default user
  44 #define GUEST_USER "guest"
  45
  46 // for local commands. return value of execdotcmd
  47 #define QUIT_CMD 0
  48 #define RPT_CMD 1
  49
  50 char username[12]; // Name of user currently using the system
  51
  52 int PWD_ROW; // Line number where user authentication happens
  53 int EDIT_ROW; // row where User Tab
  54
  55 char loginstr[] = "<L>ogin  ";
  56 char logoutstr[30];
  57
  58 int vmode; // The video mode we want to be in
  59 char timeoutcmd[MAX_CMD_LINE_LENGTH]; // Command to execute on timeout
  60 char totaltimeoutcmd[MAX_CMD_LINE_LENGTH]; // Command to execute on totaltimeout
  61
  62 char QUITSTR[] = ".quit"; // same as exit
  63 char IGNORESTR[]=".ignore"; // same as repeat, wait
  64
  65 /*----------------  End globals */
  66
  67 // returns pointer to first non-space char
  68 // and advances end of str by removing trailing spaces
  69 char * strip(char *str)
  70 {
  71   char *p,*s,*e;
  72   if (!str) return NULL;
  73   p = str;
  74   s = NULL;
  75   e = NULL;
  76   while (*p) {
  77     if (*p != ' ') {
  78        // mark start of string or record the last visited non-space char
  79        if (!s) s=p; else e=p;
  80     }
  81     p++;
  82   }
  83   *(++e)='\0'; // kill string earlier
  84   return s;
  85 }
  86
  87 // executes a list of % separated commands
  88 // non-dot commands are assumed to be syslinux commands
  89 // All syslinux commands are appended with the contents of kerargs
  90 // If it fails (kernel not found) then the next one is tried in the
  91 // list
  92 // returns QUIT_CMD or RPT_CMD
  93 t_dotcmd execdotcmd(const char *cmd, char *defcmd, const char *kerargs)
  94 {
  95    char cmdline[MAX_CMD_LINE_LENGTH];
  96    char dotcmd[MAX_CMD_LINE_LENGTH];
  97    char *curr,*next,*p,*args;
  98    char ctr;
  99
 100    strcpy(dotcmd,cmd);
 101    next = dotcmd;
 102    cmdline[0] = '\0';
 103    while (*next) { // if something to do
 104       curr = next;
 105       p = strchr(next,'%');
 106       if (p) {
 107          *p--='\0'; next=p+2;
 108          while (*p == ' ') p--;
 109          *(++p)='\0'; // remove trailing spaces
 110       } else {
 111         if (*defcmd) { // execute defcmd next
 112             next=defcmd;
 113             defcmd=NULL; // exec def cmd only once
 114         } else next=NULL;
 115       }
 116       // now we just need to execute the command "curr"
 117       curr = strip(curr);
 118       if (curr[0] != '.') { // just run the kernel
 119          strcpy(cmdline,curr);
 120          if (kerargs) strcat(cmdline,kerargs);
 121          runsyslinuximage(cmdline,0); // No IPAppend
 122       } else { // We have a DOT command
 123         // split command into command and args (may be empty)
 124         args = curr;
 125         while ( (*args != ' ') && (*args != '\0') ) args++;
 126         if (*args) { // found a space
 127            *args++ = '\0';
 128            while (*args == ' ') args++; // skip over spaces
 129         }
 130         if ( (strcmp(curr,".exit")==0) ||
 131              (strcmp(curr,".quit")==0)
 132            )
 133            return QUIT_CMD;
 134         if ( (strcmp(curr,".repeat")==0) ||
 135              (strcmp(curr,".ignore")==0) ||
 136              (strcmp(curr,".wait")==0)
 137            )
 138            return RPT_CMD;
 139         if (strcmp(curr,".beep")==0) {
 140            if ((args) && ('0' <= args[0]) && (args[0] <= '9'))
 141               ctr = args[0]-'0';
 142            else ctr=1;
 143            for (;ctr>0; ctr--) beep();
 144         }
 145         if (strcmp(curr,".help")==0) runhelp(args);
 146       }
 147    }
 148    return RPT_CMD; // by default we do not quit
 149 }
 150
 151
 152 TIMEOUTCODE timeout(const char *cmd)
 153 {
 154   t_dotcmd c;
 155   c = execdotcmd(cmd,".wait",NULL);
 156   switch(c) {
 157     case ENTER_CMD:
 158          return CODE_ENTER;
 159     case ESCAPE_CMD:
 160          return CODE_ESCAPE;
 161     default:
 162          return CODE_WAIT;
 163   }
 164 }
 165
 166 TIMEOUTCODE ontimeout(void)
 167 {
 168    return timeout(timeoutcmd);
 169 }
 170
 171 TIMEOUTCODE ontotaltimeout(void)
 172 {
 173    return timeout(totaltimeoutcmd);
 174 }
 175
 176 void keys_handler(t_menusystem * ms __attribute__ (( unused )), t_menuitem * mi, int scancode)
 177 {
 178    int nc, nr;
 179
 180    if (getscreensize(1, &nr, &nc)) {
 181        /* Unknown screen size? */
 182        nc = 80;
 183        nr = 24;
 184    }
 185
 186    if ( (scancode == KEY_F1) && (mi->helpid != 0xFFFF) ) { // If scancode of F1 and non-trivial helpid
 187       runhelpsystem(mi->helpid);
 188    }
 189
 190    // If user hit TAB, and item is an "executable" item
 191    // and user has privileges to edit it, edit it in place.
 192    if ((scancode == KEY_TAB) && (mi->action == OPT_RUN) &&
 193        (EDIT_ROW < nr) && (EDIT_ROW > 0) &&
 194        (isallowed(username,"editcmd") || isallowed(username,"root"))) {
 195      // User typed TAB and has permissions to edit command line
 196      gotoxy(EDIT_ROW,1);
 197      csprint("Command line:",0x07);
 198      editstring(mi->data,ACTIONLEN);
 199      gotoxy(EDIT_ROW,1);
 200      cprint(' ',0x07,nc-1);
 201    }
 202 }
 203
 204 t_handler_return login_handler(t_menusystem *ms, t_menuitem *mi)
 205 {
 206   (void)mi; // Unused
 207   char pwd[40];
 208   char login[40];
 209   int nc, nr;
 210   t_handler_return rv;
 211
 212   (void)ms;
 213
 214   rv = ACTION_INVALID;
 215   if (PWD_ROW < 0) return rv; // No need to authenticate
 216
 217   if (mi->item == loginstr) { /* User wants to login */
 218     if (getscreensize(1, &nr, &nc)) {
 219         /* Unknown screen size? */
 220         nc = 80;
 221         nr = 24;
 222     }
 223
 224     gotoxy(PWD_ROW,1);
 225     csprint("Enter Username: ",0x07);
 226     getstring(login, sizeof username);
 227     gotoxy(PWD_ROW,1);
 228     cprint(' ',0x07,nc);
 229     csprint("Enter Password: ",0x07);
 230     getpwd(pwd, sizeof pwd);
 231     gotoxy(PWD_ROW,1);
 232     cprint(' ',0x07,nc);
 233
 234     if (authenticate_user(login,pwd))
 235     {
 236       strcpy(username,login);
 237       strcpy(logoutstr,"<L>ogout ");
 238       strcat(logoutstr,username);
 239       mi->item = logoutstr; // Change item to read "Logout"
 240       rv.refresh = 1; // refresh the screen (as item contents changed)
 241     }
 242     else strcpy(username,GUEST_USER);
 243   }
 244   else // User needs to logout
 245   {
 246     strcpy(username,GUEST_USER);
 247     strcpy(logoutstr,"");
 248     mi->item = loginstr;
 249   }
 250
 251   return rv;
 252 }
 253
 254 t_handler_return check_perms(t_menusystem *ms, t_menuitem *mi)
 255 {
 256    char *perms;
 257    pt_xtra x;
 258
 259    (void) ms; // To keep compiler happy
 260
 261    x = (pt_xtra) mi->extra_data;
 262    perms = ( x ? x->perms : NULL);
 263    if (!perms) return ACTION_VALID;
 264
 265    if (isallowed(username,"root") || isallowed(username,perms)) // If allowed
 266       return ACTION_VALID;
 267    else return ACTION_INVALID;
 268 }
 269
 270 // Compute the full command line to add and if non-trivial
 271 // prepend the string prepend to final command line
 272 // Assume cmdline points to buffer long enough to hold answer
 273 void gencommand(pt_menuitem mi, char *cmdline)
 274 {
 275    pt_xtra x;
 276    cmdline[0] = '\0';
 277    strcat(cmdline,mi->data);
 278    x = (pt_xtra) mi->extra_data;
 279    if ( (x) && (x->argsmenu)) gen_append_line(x->argsmenu,cmdline);
 280 }
 281
 282
 283 // run the given command together with additional options which may need passing
 284 void runcommand(pt_menuitem mi)
 285 {
 286    char *line;
 287    pt_xtra x;
 288    long ipappend;
 289
 290    line = (char *)malloc(sizeof(char)*MAX_CMD_LINE_LENGTH);
 291    gencommand(mi,line);
 292    x = (pt_xtra) mi->extra_data;
 293    ipappend = (x ? x->ipappend : 0);
 294
 295    runsyslinuximage(line,x->ipappend);
 296    free(line);
 297 }
 298
 299 // set the extra info for the specified menu item.
 300 void set_xtra(pt_menuitem mi, const char *argsmenu, const char *perms, unsigned int helpid, long ipappend)
 301 {
 302    pt_xtra xtra;
 303    int bad_argsmenu, bad_perms, bad_ipappend;
 304
 305    mi->extra_data = NULL; // initalize
 306    mi->helpid = helpid; // set help id
 307
 308    if (mi->action != OPT_RUN) return;
 309
 310    bad_argsmenu = bad_perms = bad_ipappend = 0;
 311    if ( (argsmenu==NULL) || (strlen(argsmenu)==0)) bad_argsmenu = 1;
 312    if ( (perms==NULL) || (strlen(perms)==0)) bad_perms = 1;
 313    if ( ipappend==0) bad_ipappend = 1;
 314
 315    if (bad_argsmenu && bad_perms && bad_ipappend) return;
 316
 317    xtra = (pt_xtra) malloc(sizeof(t_xtra));
 318    mi->extra_data = (void *) xtra;
 319    xtra->argsmenu = xtra->perms = NULL;
 320    xtra->ipappend = ipappend;
 321    if (!bad_argsmenu) {
 322       xtra->argsmenu = (char *) malloc(sizeof(char)*(strlen(argsmenu)+1));
 323       strcpy(xtra->argsmenu,argsmenu);
 324    }
 325    if (!bad_perms) {
 326       xtra->perms = (char *) malloc(sizeof(char)*(strlen(perms)+1));
 327       strcpy(xtra->perms,perms);
 328       mi->handler = &check_perms;
 329    }
 330 }
 331
 332 int main(void)
 333 {
 334   pt_menuitem curr;
 335   char quit;
 336   char exitcmd[MAX_CMD_LINE_LENGTH];
 337   char exitcmdroot[MAX_CMD_LINE_LENGTH];
 338   char onerrcmd[MAX_CMD_LINE_LENGTH];
 339   char startfile[MAX_CMD_LINE_LENGTH];
 340   char *ecmd; // effective exit command or onerrorcmd
 341   char skipbits;
 342   char skipcmd[MAX_CMD_LINE_LENGTH];
 343   int timeout; // time in multiples of 0.1 seconds
 344   int totaltimeout; // time in multiples of 0.1 seconds
 345   t_dotcmd dotrv; // to store the return value of execdotcmd
 346   int temp;
 347
 348   strcpy(username,GUEST_USER);
 349 /* ---- Initializing menu system parameters --- */
 350   vmode = 0xFF;
 351   skipbits = 0;
 352   PWD_ROW = 23;
 353   EDIT_ROW = 23;
 354   strcpy(onerrcmd,".repeat");
 355   strcpy(exitcmd,".exit");
 356   strcpy(exitcmdroot,"");
 357   // If not specified exitcmdroot = exitcmd
 358   if (exitcmdroot[0] == '\0') strcpy(exitcmdroot,exitcmd);
 359   // Timeout stuff
 360   timeout = 600;
 361   strcpy(timeoutcmd,".beep");
 362   totaltimeout = 0;
 363   strcpy(totaltimeoutcmd,".wait");
 364   strcpy(startfile,"");
 365
 366   init_help("/isolinux/help");
 367   init_passwords("");
 368   init_menusystem("A test of the test.menu file");
 369   set_window_size(1,1,23,78);
 370
 371   // Register the ontimeout handler, with a time out of 10 seconds
 372   reg_ontimeout(ontimeout,timeout*10,0);
 373   reg_ontotaltimeout(ontotaltimeout,totaltimeout*10);
 374
 375   // Register menusystem handlers
 376   reg_handler(HDLR_KEYS,&keys_handler);
 377 /* ---- End of initialization --- */
 378
 379 /* ------- MENU testing ----- */
 380   add_named_menu("testing"," Testing ",-1);
 381
 382   curr = add_item("Self Loop","Go to Testing",OPT_SUBMENU,"testing",0);
 383   set_xtra(curr,"","",65535,0); // Set associated extra info
 384   set_shortcut(-1);
 385
 386   curr = add_item("Memory Test","Perform extensive memory testing",OPT_RUN,"memtest",0);
 387   set_xtra(curr,"","",65535,0); // Set associated extra info
 388   set_shortcut(-1);
 389
 390   curr = add_item("Exit this menu","Go one level up",OPT_EXITMENU,"",0);
 391   set_xtra(curr,"","",65535,0); // Set associated extra info
 392   set_shortcut(-1);
 393
 394 /* ------- MENU rescue ----- */
 395   add_named_menu("rescue"," Rescue Options ",-1);
 396   set_menu_pos(10,10);
 397
 398   curr = add_item("Linux Rescue","linresc",OPT_RUN,"linresc",0);
 399   set_xtra(curr,"","",65535,0); // Set associated extra info
 400   set_shortcut(-1);
 401
 402   curr = add_item("Dos Rescue","dosresc",OPT_RUN,"dosresc",0);
 403   set_xtra(curr,"","",65535,0); // Set associated extra info
 404   set_shortcut(-1);
 405
 406   curr = add_item("Windows Rescue","winresc",OPT_RUN,"winresc",0);
 407   set_xtra(curr,"","",65535,0); // Set associated extra info
 408   set_shortcut(-1);
 409
 410   curr = add_item("Exit this menu","Go one level up",OPT_EXITMENU,"",0);
 411   set_xtra(curr,"","",65535,0); // Set associated extra info
 412   set_shortcut(-1);
 413
 414 /* ------- MENU main ----- */
 415   add_named_menu("main"," Main Menu ",-1);
 416
 417   curr = add_item("Prepare","prep",OPT_RUN,"prep",0);
 418   set_xtra(curr,"","",65535,0); // Set associated extra info
 419   set_shortcut(-1);
 420
 421   curr = add_item("Rescue options...","Troubleshoot a system",OPT_SUBMENU,"rescue",0);
 422   set_xtra(curr,"","",65535,0); // Set associated extra info
 423   set_shortcut(-1);
 424
 425   curr = add_item("Testing...","Options to test hardware",OPT_SUBMENU,"testing",0);
 426   set_xtra(curr,"","",65535,0); // Set associated extra info
 427   set_shortcut(-1);
 428
 429   curr = add_item("Exit this menu","Go one level up",OPT_EXITMENU,"",0);
 430   set_xtra(curr,"","",65535,0); // Set associated extra info
 431   set_shortcut(-1);
 432 /* ------- END OF MENU declarations ----- */
 433
 434 // Check if we should skip the menu altogether
 435   quit = 0; // Dont skip the menu
 436   if (getshiftflags() & skipbits) { // we must skip the menu altogther and execute skipcmd
 437     dotrv = execdotcmd(skipcmd,".beep%.exit",NULL); // Worst case we beep and exit
 438     if (dotrv == QUIT_CMD) quit = 1;
 439   }
 440
 441 // Switch vide mode if required
 442    if (vmode != 0xFF) setvideomode(vmode);
 443
 444 // Do we have a startfile to display?
 445    if (startfile[0] != '\0') runhelp(startfile);
 446
 447 // The main loop
 448   while (quit == 0) { // As long as quit is zero repeat
 449      curr = showmenus(find_menu_num("main")); // Initial menu is the one called "main"
 450
 451      if (curr) {
 452         if (curr->action == OPT_RUN) {
 453            ecmd = (char *)malloc(sizeof(char)*MAX_CMD_LINE_LENGTH);
 454            gencommand(curr,ecmd);
 455            temp = (curr->extra_data ? ((pt_xtra)curr->extra_data)->ipappend : 0);
 456            runsyslinuximage(ecmd,temp);
 457            // kernel not found so execute the appropriate dot command
 458            dotrv = execdotcmd(onerrcmd,".quit",ecmd); // pass bad cmdline as arg
 459            if (dotrv== QUIT_CMD) quit = 1;
 460            free(ecmd); ecmd = NULL;
 461         }
 462         else csprint("Error in programming!",0x07);
 463      } else {
 464         // find effective exit command
 465         ecmd = ( isallowed(username,"root") ? exitcmdroot : exitcmd);
 466         dotrv = execdotcmd(ecmd,".repeat",NULL);
 467         quit = (dotrv == QUIT_CMD ? 1 : 0); // should we exit now
 468      }
 469   }
 470
 471 // Deallocate space used and quit
 472   close_passwords();
 473   close_help();
 474   close_menusystem();
 475   return 0;
 476 }
 477