home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / perl560.zip / ext / SDBM_File / sdbm / dbu.c < prev    next >
C/C++ Source or Header  |  1999-07-20  |  4KB  |  244 lines
  1. #include <stdio.h>
  2. #include <sys/file.h>
  3. #ifdef SDBM
  4. #include "EXTERN.h"
  5. #include "sdbm.h"
  6. #else
  7. #include <ndbm.h>
  8. #endif
  9. #include <string.h>
  10.  
  11. #ifdef BSD42
  12. #define strchr    index
  13. #endif
  14.  
  15. extern int    getopt();
  16. extern char    *strchr();
  17. extern void    oops();
  18.  
  19. char *progname;
  20.  
  21. static int rflag;
  22. static char *usage = "%s [-R] cat | look |... dbmname";
  23.  
  24. #define DERROR    0
  25. #define DLOOK    1
  26. #define DINSERT    2
  27. #define DDELETE 3
  28. #define    DCAT    4
  29. #define DBUILD    5
  30. #define DPRESS    6
  31. #define DCREAT    7
  32.  
  33. #define LINEMAX    8192
  34.  
  35. typedef struct {
  36.     char *sname;
  37.     int scode;
  38.     int flags;
  39. } cmd;
  40.  
  41. static cmd cmds[] = {
  42.  
  43.     "fetch", DLOOK,     O_RDONLY,
  44.     "get", DLOOK,        O_RDONLY,
  45.     "look", DLOOK,        O_RDONLY,
  46.     "add", DINSERT,        O_RDWR,
  47.     "insert", DINSERT,    O_RDWR,
  48.     "store", DINSERT,    O_RDWR,
  49.     "delete", DDELETE,    O_RDWR,
  50.     "remove", DDELETE,    O_RDWR,
  51.     "dump", DCAT,        O_RDONLY,
  52.     "list", DCAT,         O_RDONLY,
  53.     "cat", DCAT,        O_RDONLY,
  54.     "creat", DCREAT,    O_RDWR | O_CREAT | O_TRUNC,
  55.     "new", DCREAT,        O_RDWR | O_CREAT | O_TRUNC,
  56.     "build", DBUILD,    O_RDWR | O_CREAT,
  57.     "squash", DPRESS,    O_RDWR,
  58.     "compact", DPRESS,    O_RDWR,
  59.     "compress", DPRESS,    O_RDWR
  60. };
  61.  
  62. #define CTABSIZ (sizeof (cmds)/sizeof (cmd))
  63.  
  64. static cmd *parse();
  65. static void badk(), doit(), prdatum();
  66.  
  67. int
  68. main(int argc, char **argv)
  69. {
  70.     int c;
  71.     register cmd *act;
  72.     extern int optind;
  73.     extern char *optarg;
  74.  
  75.     progname = argv[0];
  76.  
  77.     while ((c = getopt(argc, argv, "R")) != EOF)
  78.         switch (c) {
  79.         case 'R':           /* raw processing  */
  80.             rflag++;
  81.             break;
  82.  
  83.         default:
  84.             oops("usage: %s", usage);
  85.             break;
  86.         }
  87.  
  88.     if ((argc -= optind) < 2)
  89.         oops("usage: %s", usage);
  90.  
  91.     if ((act = parse(argv[optind])) == NULL)
  92.         badk(argv[optind]);
  93.     optind++;
  94.     doit(act, argv[optind]);
  95.     return 0;
  96. }
  97.  
  98. static void
  99. doit(register cmd *act, char *file)
  100. {
  101.     datum key;
  102.     datum val;
  103.     register DBM *db;
  104.     register char *op;
  105.     register int n;
  106.     char *line;
  107. #ifdef TIME
  108.     long start;
  109.     extern long time();
  110. #endif
  111.  
  112.     if ((db = dbm_open(file, act->flags, 0644)) == NULL)
  113.         oops("cannot open: %s", file);
  114.  
  115.     if ((line = (char *) malloc(LINEMAX)) == NULL)
  116.         oops("%s: cannot get memory", "line alloc");
  117.  
  118.     switch (act->scode) {
  119.  
  120.     case DLOOK:
  121.         while (fgets(line, LINEMAX, stdin) != NULL) {
  122.             n = strlen(line) - 1;
  123.             line[n] = 0;
  124.             key.dptr = line;
  125.             key.dsize = n;
  126.             val = dbm_fetch(db, key);
  127.             if (val.dptr != NULL) {
  128.                 prdatum(stdout, val);
  129.                 putchar('\n');
  130.                 continue;
  131.             }
  132.             prdatum(stderr, key);
  133.             fprintf(stderr, ": not found.\n");
  134.         }
  135.         break;
  136.     case DINSERT:
  137.         break;
  138.     case DDELETE:
  139.         while (fgets(line, LINEMAX, stdin) != NULL) {
  140.             n = strlen(line) - 1;
  141.             line[n] = 0;
  142.             key.dptr = line;
  143.             key.dsize = n;
  144.             if (dbm_delete(db, key) == -1) {
  145.                 prdatum(stderr, key);
  146.                 fprintf(stderr, ": not found.\n");
  147.             }
  148.         }
  149.         break;
  150.     case DCAT:
  151.         for (key = dbm_firstkey(db); key.dptr != 0; 
  152.              key = dbm_nextkey(db)) {
  153.             prdatum(stdout, key);
  154.             putchar('\t');
  155.             prdatum(stdout, dbm_fetch(db, key));
  156.             putchar('\n');
  157.         }
  158.         break;
  159.     case DBUILD:
  160. #ifdef TIME
  161.         start = time(0);
  162. #endif
  163.         while (fgets(line, LINEMAX, stdin) != NULL) {
  164.             n = strlen(line) - 1;
  165.             line[n] = 0;
  166.             key.dptr = line;
  167.             if ((op = strchr(line, '\t')) != 0) {
  168.                 key.dsize = op - line;
  169.                 *op++ = 0;
  170.                 val.dptr = op;
  171.                 val.dsize = line + n - op;
  172.             }
  173.             else
  174.                 oops("bad input; %s", line);
  175.     
  176.             if (dbm_store(db, key, val, DBM_REPLACE) < 0) {
  177.                 prdatum(stderr, key);
  178.                 fprintf(stderr, ": ");
  179.                 oops("store: %s", "failed");
  180.             }
  181.         }
  182. #ifdef TIME
  183.         printf("done: %d seconds.\n", time(0) - start);
  184. #endif
  185.         break;
  186.     case DPRESS:
  187.         break;
  188.     case DCREAT:
  189.         break;
  190.     }
  191.  
  192.     dbm_close(db);
  193. }
  194.  
  195. static void
  196. badk(char *word)
  197. {
  198.     register int i;
  199.  
  200.     if (progname)
  201.         fprintf(stderr, "%s: ", progname);
  202.     fprintf(stderr, "bad keywd %s. use one of\n", word);
  203.     for (i = 0; i < (int)CTABSIZ; i++)
  204.         fprintf(stderr, "%-8s%c", cmds[i].sname,
  205.             ((i + 1) % 6 == 0) ? '\n' : ' ');
  206.     fprintf(stderr, "\n");
  207.     exit(1);
  208.     /*NOTREACHED*/
  209. }
  210.  
  211. static cmd *
  212. parse(register char *str)
  213. {
  214.     register int i = CTABSIZ;
  215.     register cmd *p;
  216.     
  217.     for (p = cmds; i--; p++)
  218.         if (strcmp(p->sname, str) == 0)
  219.             return p;
  220.     return NULL;
  221. }
  222.  
  223. static void
  224. prdatum(FILE *stream, datum d)
  225. {
  226.     register int c;
  227.     register char *p = d.dptr;
  228.     register int n = d.dsize;
  229.  
  230.     while (n--) {
  231.         c = *p++ & 0377;
  232.         if (c & 0200) {
  233.             fprintf(stream, "M-");
  234.             c &= 0177;
  235.         }
  236.         if (c == 0177 || c < ' ') 
  237.             fprintf(stream, "^%c", (c == 0177) ? '?' : c + '@');
  238.         else
  239.             putc(c, stream);
  240.     }
  241. }
  242.  
  243.  
  244.