ppmanager.c

   1 /*
   2 ** Copyright (C) 2006 Thadeu Lima de Souza Cascardo <cascardo@minaslivre.org>
   3 ** Copyright (C) 2009 Thadeu Lima de Souza Cascardo <cascardo@holoscopio.com>
   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; either version 2 of the License, or
   8 ** (at your option) any later version.
   9 **
  10 ** This program is distributed in the hope that it will be useful,
  11 ** but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 ** GNU General Public License for more details.
  14 **
  15 ** You should have received a copy of the GNU General Public License
  16 ** along with this program; if not, write to the Free Software
  17 ** Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  18 **
  19 */
  20
  21 #include <stdio.h>
  22 #include <stdlib.h>
  23 #include <string.h>
  24 #include <depot.h>
  25
  26 enum
  27 {
  28   ACTION_ADD,
  29   ACTION_DEL,
  30   ACTION_LIST
  31 };
  32
  33 void
  34 pp_dplistkeys (DEPOT *dp)
  35 {
  36   char *key;
  37   dpiterinit (dp);
  38   int ksz;
  39   while ((key = dpiternext (dp, &ksz)) != NULL && ksz > 0)
  40     {
  41       printf ("%s\n", key);
  42       free (key);
  43     }
  44 }
  45
  46 void
  47 usage (char *programname)
  48 {
  49   fprintf (stderr, "%s {allow | deny} {add | del | list} [user]\n",
  50            programname);
  51   exit (1);
  52 }
  53
  54 int
  55 main (int argc, char **argv)
  56 {
  57   DEPOT *dp;
  58   int action;
  59   char *db;
  60   char *username;
  61   int r = 0;
  62   if (argc < 3)
  63     usage (argv[0]);
  64   if (!strcmp (argv[1], "deny"))
  65     db = "/var/lib/popproxy/deny.db";
  66   else if (!strcmp (argv[1], "allow"))
  67     db = "/var/lib/popproxy/allow.db";
  68   else
  69     usage (argv[0]);
  70   if (!strcmp (argv[2], "add"))
  71     action = ACTION_ADD;
  72   else if (!strcmp (argv[2], "del"))
  73     action = ACTION_DEL;
  74   else if (!strcmp (argv[2], "list"))
  75     action = ACTION_LIST;
  76   else
  77     usage (argv[0]);
  78   if ((action == ACTION_ADD || action == ACTION_DEL) && argc < 4)
  79     usage (argv[0]);
  80   username = argv[3];
  81   dp = dpopen (db, DP_OWRITER | DP_OCREAT, 0);
  82   if (dp == NULL)
  83     {
  84       fprintf (stderr, "Could not create database.\n");
  85       exit (1);
  86     }
  87   switch (action)
  88     {
  89     case ACTION_ADD:
  90       dpput (dp, username, -1, "", -1, DP_DOVER);
  91       break;
  92     case ACTION_DEL:
  93       dpout (dp, username, -1);
  94       break;
  95     case ACTION_LIST:
  96       pp_dplistkeys (dp);
  97       r = 1;
  98       break;
  99     default:
 100       fprintf (stderr, "Action unrecognized.\n");
 101       r = 1;
 102     }
 103   dpclose (dp);
 104   return r;
 105 }