1/* 2 3/usr/src/ext2ed/main.c 4 5A part of the extended file system 2 disk editor. 6 7------------ 8Main program 9------------ 10 11This file mostly contains: 12 131. A list of global variables used through the entire program. 142. The parser, which asks the command line from the user. 153. The dispatcher, which analyzes the command line and calls the appropriate handler function. 164. A command pattern matcher which is used along with the readline completion feature. 175. A function which tells the user that an internal error has occured. 18 19First written on: March 30 1995 20 21Copyright (C) 1995 Gadi Oxman 22 23*/ 24 25#include <stdio.h> 26#include <stdlib.h> 27#include <string.h> 28#include <signal.h> 29 30#ifdef HAVE_READLINE 31#include <readline.h> 32#include <history.h> 33#endif 34 35#ifdef HAVE_GETOPT_H 36#include <getopt.h> 37#else 38extern int optind; 39extern char *optarg; 40#endif 41 42#include "ext2ed.h" 43 44/* Global variables */ 45 46/* 47 48Configuration file options 49 50The following variables will be set by init.c to the values selected in the user configuration file. 51They are initialized below to some logical defaults. 52 53*/ 54 55 56char Ext2Descriptors [200]="ext2.descriptors"; /* The location of the ext2 filesystem object definition */ 57char AlternateDescriptors [200]=""; /* We allow the user to define additional structures */ 58char LogFile [200]="ext2ed.log"; /* The location of the log file - Each write will be logged there */ 59int LogChanges=1; /* 1 enables logging, 0 diables logging */ 60int AllowChanges=0; /* When set, the enablewrite command will fail */ 61int AllowMountedRead=0; /* Behavior when trying to open a mounted filesystem read-only */ 62int ForceExt2=0; /* When set, ext2 autodetection is overridden */ 63int DefaultBlockSize=1024; 64unsigned long DefaultTotalBlocks=2097151; 65unsigned long DefaultBlocksInGroup=8192; /* The default values are used when an ext2 filesystem is not */ 66int ForceDefault=0; /* detected, or ForceDefault is set */ 67 68char last_command_line [80]; /* A simple one command cache, in addition to the readline history */ 69 70char device_name [80]; /* The location of the filesystem */ 71FILE *device_handle=NULL; /* This is passed to the fopen / fread ... commands */ 72long device_offset; /* The current position in the filesystem */ 73 /* Note that we have a 2 GB limitation */ 74 75int mounted=0; /* This is set when we find that the filesystem is mounted */ 76 77struct struct_commands general_commands,ext2_commands; /* Used to define the general and ext2 commands */ 78struct struct_descriptor *first_type,*last_type,*current_type; /* Used to access the double linked list */ 79struct struct_type_data type_data; /* The current data is sometimes stored here */ 80struct struct_file_system_info file_system_info; /* Essential information on the filesystem */ 81struct struct_file_info file_info,first_file_info; /* Used by file_com.c to access files */ 82struct struct_group_info group_info; /* Used by group_com.c */ 83struct struct_super_info super_info; /* Used by super_com.c */ 84struct struct_remember_lifo remember_lifo; /* A circular memory of objects */ 85struct struct_block_bitmap_info block_bitmap_info; /* Used by blockbitmap_com.c */ 86struct struct_inode_bitmap_info inode_bitmap_info; /* Used by inodebitmap_com.c */ 87 88int redraw_request=0; /* Is set by a signal handler to handle terminal */ 89 /* screen size change. */ 90 91 92/* 93 * We just call the parser to get commands from the user. We quit when 94 * parser returns. 95 */ 96int main (int argc, char **argv) 97{ 98 int write_priv = 0; 99 int c; 100 char *buf; 101 102 if (!init ()) 103 return (1); 104 while ((c = getopt (argc, argv, "w")) != EOF) { 105 switch (c) { 106 case 'w': 107 write_priv++; 108 break; 109 } 110 } 111 if (optind < argc) { 112 buf = malloc(strlen(argv[optind]) + 32); 113 if (!buf) { 114 fprintf(stderr, "Couldn't allocate filename buffer\n"); 115 exit(1); 116 } 117 strcpy(buf, "set_device "); 118 strcat(buf, argv[optind]); 119 set_device(buf); 120 free(buf); 121 if (write_priv) { 122 wprintw (command_win,"\n"); 123 enable_write("enable_write"); 124 } 125 } 126 parser (); /* Get and parse user commands */ 127 prepare_to_close(); /* Do some cleanup */ 128 printf("Quitting ...\n"); 129 return(0); 130} 131 132 133/* 134 * Read a character from the command window 135 */ 136int command_read_key() 137{ 138 int key = 0; 139 140 while (!key) { 141 if (redraw_request) { 142 redraw_all(); 143 redraw_request=0; 144 } 145 key = wgetch(command_win); 146 switch (key) { 147 case 0x1A: 148 key = 0; 149 kill(getpid(), SIGTSTP); 150 break; 151 152 case KEY_NPAGE: 153 pgdn(""); 154 refresh_command_win (); 155 break; 156 157 case KEY_PPAGE: 158 pgup(""); 159 refresh_command_win (); 160 break; 161 case ERR: 162 key = 0; 163 break; 164 165 case KEY_BACKSPACE: 166 key = '\b'; 167 } 168 if ((key < 32 && key != '\b' && key != '\n') || 169 (key > 127)) 170 key = 0; 171 } 172 return key; 173} 174 175#ifdef HAVE_READLINE 176int rl_getc_replacement(FILE *f) 177{ 178 int key = command_read_key(); 179 180 if (key == '\b') { 181 if (rl_point > 0) 182 wprintw(command_win, "\b \b"); 183 } else 184 wprintw(command_win, "%c", key); 185 return key; 186} 187 188/* 189 * This function asks the user for a command and calls the dispatcher 190 * function, dispatch, to analyze it. We use the readline library 191 * function readline to read the command, hence all the usual readline 192 * keys are available. The new command is saved both in the 193 * readline's history and in our tiny one-command cache, so that only 194 * the enter key is needed to retype it. 195 */ 196void parser (void) 197{ 198 char *ptr,command_line [80]; 199 int quit=0; 200 201#if 0 202 noecho(); 203 cbreak(); 204 keypad(command_win, 1); 205 wtimeout(command_win, 100); 206 207 rl_getc_function = rl_getc_replacement; 208#endif 209 210 while (!quit) { 211 /* Terminal screen size has changed */ 212 if (redraw_request) { 213 redraw_all(); 214 redraw_request=0; 215 } 216 217 wmove (command_win,0,0); 218 wclrtoeol (command_win); 219 wprintw (command_win,"ext2ed > "); 220 refresh_command_win (); 221 222 /* 223 * The ncurses library optimizes cursor movement by 224 * keeping track of the cursor position. However, by 225 * using the readline library I'm breaking its 226 * assumptions. The double -1 arguments tell ncurses 227 * to disable cursor movement optimization this 228 * time. 229 */ 230 mvcur (-1,-1,LINES-COMMAND_WIN_LINES,0); 231 232 /* echo (); */ 233 ptr=readline ("ext2ed > "); 234 /* noecho (); */ 235 236 /* 237 * Readline allocated the buffer - Copy the string 238 * and free the allocated buffer 239 * XXX WHY??? 240 */ 241 strcpy (command_line,ptr); 242 free (ptr); 243 244 if (*command_line != 0) 245 add_history (command_line); 246 247 /* If only enter was pressed, recall the last command */ 248 if (*command_line==0) 249 strcpy (command_line,last_command_line); 250 251 /* Emulate readline's actions for ncurses */ 252 mvcur (-1,-1,LINES-COMMAND_WIN_LINES,0); 253 werase (command_win); 254 wprintw (command_win,"ext2ed > "); 255 wprintw (command_win,command_line); 256 wprintw (command_win,"\n"); 257 refresh_command_win (); 258 259 /* Save this command in our tiny cache */ 260 strcpy (last_command_line,command_line); 261 262 /* And call dispatch to do the actual job */ 263 quit=dispatch (command_line); 264 } 265} 266#else 267void read_line(char * foo) { 268 char * chptr = foo; 269 int ch; 270 int done = 0; 271 272 while (!done && (ch = command_read_key())) { 273 switch (ch) { 274 case '\n': 275 done = 1; 276 break; 277 278 case '\b': 279 if (chptr > foo) { 280 wprintw(command_win, "\b \b"); 281 chptr--; 282 } 283 break; 284 285 default: 286 if (ch > 256) 287 break; 288 if (ch == '\n') break; 289 *chptr++ = ch; 290 wprintw(command_win, "%c", ch); 291 break; 292 } 293 } 294 *chptr = '\0'; 295} 296 297void parser (void) 298{ 299 char command_line [80]; 300 int quit=0; 301 302 noecho(); 303 cbreak(); 304 wtimeout(command_win, 100); 305 keypad(command_win, 1); 306 307 while (!quit) { 308 /* Terminal screen size has changed */ 309 if (redraw_request) { 310 redraw_all(); 311 redraw_request=0; 312 } 313 314 wmove (command_win,0,0);wclrtoeol (command_win); 315 316 wmove(command_win, 0, 0); 317 wprintw(command_win, "ext2ed > "); 318 read_line(command_line); 319 320 /* If only enter was pressed, recall the last command */ 321 if (*command_line==0) 322 strcpy (command_line,last_command_line); 323 324 mvcur (-1,-1,LINES-COMMAND_WIN_LINES + 1,0); 325 326 strcpy (last_command_line,command_line); /* Save this command in our tiny cache */ 327 328 /* And call dispatch to do the actual job */ 329 quit=dispatch (command_line); 330 } 331} 332#endif 333 334 335/* 336 * This is a very important function. Its task is to recieve a command 337 * name and link it to a C function. There are three types of commands: 338 * 339 * 1. General commands - Always available and accessed through 340 * general_commands. 341 * 2. Ext2 specific commands - Available when editing an ext2 342 * filesystem, accessed through ext2_commands. 343 * 3. Type specific commands - Those are changing according to the 344 * current type. The global variable current_type points to the 345 * current object definition (of type struct_descriptor). In it, the 346 * struct_commands entry contains the type specific commands links. 347 * 348 * Overriding is an important feature - Much like in C++ : The same 349 * command name can dispatch to different functions. The overriding 350 * priority is 3,2,1; That is - A type specific command will always 351 * override a general command. This is used through the program to 352 * allow fine tuned operation. 353 * 354 * When an handling function is found, it is called along with the 355 * command line that was passed to us. The handling function is then 356 * free to interpert the arguments in its own style. 357 */ 358int dispatch (char *command_line) 359{ 360 int i,found=0; 361 362 char command [80]; 363 364 parse_word (command_line,command); 365 366 if (strcasecmp (command,"quit")==0) return (1); 367 368 /* 1. Search for type specific commands FIRST - Allows 369 overriding of a general command */ 370 371 if (current_type != NULL) 372 for (i=0; 373 i<=current_type->type_commands.last_command && !found; 374 i++) { 375 if (strcasecmp (command,current_type->type_commands.names [i])==0) { 376 (*current_type->type_commands.callback [i]) (command_line); 377 found=1; 378 } 379 } 380 381 /* 2. Now search for ext2 filesystem general commands */ 382 383 if (!found) 384 for (i=0;i<=ext2_commands.last_command && !found;i++) { 385 if (strcasecmp (command,ext2_commands.names [i])==0) { 386 (*ext2_commands.callback [i]) (command_line); 387 found=1; 388 } 389 } 390 391 392 /* 3. If not found, search the general commands */ 393 394 if (!found) 395 for (i=0;i<=general_commands.last_command && !found;i++) { 396 if (strcasecmp (command,general_commands.names [i])==0) { 397 (*general_commands.callback [i]) (command_line); 398 found=1; 399 } 400 } 401 402 /* 4. If not found, issue an error message and return */ 403 404 if (!found) { 405 wprintw (command_win,"Error: Unknown command\n"); 406 refresh_command_win (); 407 } 408 409 return (0); 410} 411 412 413/* 414 * 415 * This function copies the next word in source to the variable dest, 416 * ignoring whitespaces. It returns a pointer to the next word in 417 * source. It is used to split the command line into command and arguments. 418 */ 419char *parse_word (char *source,char *dest) 420{ 421 char ch,*source_ptr,*target_ptr; 422 423 if (*source==0) { 424 *dest=0; 425 return (source); 426 }; 427 428 source_ptr=source;target_ptr=dest; 429 do { 430 ch=*source_ptr++; 431 } while (! (ch>' ' && ch<='z') && ch!=0); 432 433 while (ch>' ' && ch<='z') { 434 *target_ptr++=ch; 435 ch=*source_ptr++; 436 } 437 438 *target_ptr=0; 439 440 source_ptr--; 441 do { 442 ch=*source_ptr++; 443 } while (! (ch>' ' && ch<='z') && ch!=0); 444 445 return (--source_ptr); 446} 447 448/* 449 * text is the partial command entered by the user; We assume that it 450 * is a part of a command - I didn't write code for smarter completion. 451 * 452 * The state variable is an index which tells us how many possible 453 * completions we already returned to readline. 454 * 455 * We return only one possible completion or (char *) NULL if there 456 * are no more completions. This function will be called by readline 457 * over and over until we tell it to stop. 458 * 459 * While scanning for possible completions, we use the same priority 460 * definition which was used in dispatch. 461 */ 462#if HAVE_READLINE 463char *complete_command (char *text,int state) 464{ 465 int state_index=-1; 466 int i,len; 467 468 len=strlen (text); 469 470 /* Is the command type specific ? */ 471 472 if (current_type != NULL) 473 for (i=0;i<=current_type->type_commands.last_command;i++) { 474 if (strncmp (current_type->type_commands.names [i],text,len)==0) { 475 state_index++; 476 if (state==state_index) { 477 return (dupstr (current_type->type_commands.names [i])); 478 } 479 } 480 } 481 482 /* No, pehaps ext2 specific command then ? */ 483 484 for (i=0;i<=ext2_commands.last_command;i++) { 485 if (strncmp (ext2_commands.names [i],text,len)==0) { 486 state_index++; 487 if (state==state_index) 488 return (dupstr (ext2_commands.names [i])); 489 } 490 } 491 492 493 /* Check for a general command */ 494 495 for (i=0;i<=general_commands.last_command;i++) { 496 if (strncmp (general_commands.names [i],text,len)==0) { 497 state_index++; 498 if (state==state_index) 499 return (dupstr (general_commands.names [i])); 500 } 501 } 502 503 /* quit is handled differently */ 504 505 if (strncmp ("quit",text,len)==0) { 506 state_index++; 507 if (state==state_index) 508 return (dupstr ("quit")); 509 } 510 511 /* No more completions */ 512 513 return ((char *) NULL); 514} 515#endif 516 517 518/* 519 * Nothing special - Just allocates enough space and copy the string. 520 */ 521char *dupstr (char *src) 522{ 523 char *ptr; 524 525 ptr=(char *) malloc (strlen (src)+1); 526 strcpy (ptr,src); 527 return (ptr); 528} 529 530#ifdef DEBUG 531/* 532 * This function reports an internal error. It is almost not used. One 533 * place in which I do check for internal errors is disk.c. 534 * 535 * We just report the error, and try to continue ... 536 */ 537void internal_error (char *description,char *source_name,char *function_name) 538{ 539 wprintw (command_win,"Internal error - Found by source: %s.c , function: %s\n",source_name,function_name); 540 wprintw (command_win,"\t%s\n",description); 541 wprintw (command_win,"Press enter to (hopefully) continue\n"); 542 refresh_command_win ();getch ();werase (command_win); 543} 544 545#endif 546