Annotation of embedaddon/quagga/tests/test-privs.c, revision 1.1.1.2

1.1       misho       1: /*
                      2:  * $Id: test-privs.c,v 1.1 2005/10/11 03:48:28 paul Exp $
                      3:  *
                      4:  * This file is part of Quagga.
                      5:  *
                      6:  * Quagga is free software; you can redistribute it and/or modify it
                      7:  * under the terms of the GNU General Public License as published by the
                      8:  * Free Software Foundation; either version 2, or (at your option) any
                      9:  * later version.
                     10:  *
                     11:  * Quagga is distributed in the hope that it will be useful, but
                     12:  * WITHOUT ANY WARRANTY; without even the implied warranty of
                     13:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
                     14:  * General Public License for more details.
                     15:  *
                     16:  * You should have received a copy of the GNU General Public License
                     17:  * along with Quagga; see the file COPYING.  If not, write to the Free
                     18:  * Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
                     19:  * 02111-1307, USA.
                     20:  */
                     21: 
                     22: #include <zebra.h>
                     23: 
                     24: #include <lib/version.h>
                     25: #include "getopt.h"
                     26: #include "privs.h"
                     27: #include "memory.h"
                     28: 
                     29: zebra_capabilities_t _caps_p [] = 
                     30: {
                     31:   ZCAP_NET_RAW,
                     32:   ZCAP_BIND,
                     33:   ZCAP_NET_ADMIN,
                     34:   ZCAP_DAC_OVERRIDE,
                     35: };
                     36: 
                     37: struct zebra_privs_t test_privs =
                     38: {
                     39: #if defined(QUAGGA_USER) && defined(QUAGGA_GROUP)
                     40:   .user = QUAGGA_USER,
                     41:   .group = QUAGGA_GROUP,
                     42: #endif
                     43: #if defined(VTY_GROUP)
                     44:   .vty_group = VTY_GROUP,
                     45: #endif
                     46:   .caps_p = _caps_p,
                     47:   .cap_num_p = sizeof(_caps_p)/sizeof(_caps_p[0]),
                     48:   .cap_num_i = 0
                     49: };
                     50: 
                     51: struct option longopts[] = 
                     52: {
                     53:   { "help",        no_argument,       NULL, 'h'},
                     54:   { "user",        required_argument, NULL, 'u'},
                     55:   { "group",       required_argument, NULL, 'g'},
                     56:   { 0 }
                     57: };
                     58: 
                     59: /* Help information display. */
                     60: static void
                     61: usage (char *progname, int status)
                     62: {
                     63:   if (status != 0)
                     64:     fprintf (stderr, "Try `%s --help' for more information.\n", progname);
                     65:   else
                     66:     {    
                     67:       printf ("Usage : %s [OPTION...]\n\
                     68: Daemon which does 'slow' things.\n\n\
                     69: -u, --user         User to run as\n\
                     70: -g, --group        Group to run as\n\
                     71: -h, --help         Display this help and exit\n\
                     72: \n\
                     73: Report bugs to %s\n", progname, ZEBRA_BUG_ADDRESS);
                     74:     }
                     75:   exit (status);
                     76: }
1.1.1.2 ! misho      77: 
1.1       misho      78: struct thread_master *master;
                     79: /* main routine. */
                     80: int
                     81: main (int argc, char **argv)
                     82: {
                     83:   char *p;
                     84:   char *progname;
                     85:   struct zprivs_ids_t ids;
                     86:   
                     87:   /* Set umask before anything for security */
                     88:   umask (0027);
                     89: 
                     90:   /* get program name */
                     91:   progname = ((p = strrchr (argv[0], '/')) ? ++p : argv[0]);
                     92: 
                     93:   while (1) 
                     94:     {
                     95:       int opt;
                     96: 
                     97:       opt = getopt_long (argc, argv, "hu:g:", longopts, 0);
                     98:     
                     99:       if (opt == EOF)
                    100:        break;
                    101: 
                    102:       switch (opt) 
                    103:        {
                    104:        case 0:
                    105:          break;
                    106:         case 'u':
                    107:           test_privs.user = optarg;
                    108:           break;
                    109:         case 'g':
                    110:           test_privs.group = optarg;
                    111:           break;
                    112:        case 'h':
                    113:          usage (progname, 0);
                    114:          break;
                    115:        default:
                    116:          usage (progname, 1);
                    117:          break;
                    118:        }
                    119:     }
                    120: 
                    121:   /* Library inits. */
                    122:   memory_init ();
                    123:   zprivs_init (&test_privs);
                    124: 
                    125: #define PRIV_STATE() \
                    126:   ((test_privs.current_state() == ZPRIVS_RAISED) ? "Raised" : "Lowered")
                    127:   
                    128:   printf ("%s\n", PRIV_STATE());
                    129:   test_privs.change(ZPRIVS_RAISE);
                    130:   
                    131:   printf ("%s\n", PRIV_STATE());
                    132:   test_privs.change(ZPRIVS_LOWER);
                    133:   
                    134:   printf ("%s\n", PRIV_STATE());
                    135:   zprivs_get_ids (&ids);  
                    136:   
                    137:   /* terminate privileges */
                    138:   zprivs_terminate(&test_privs);
                    139:   
                    140:   /* but these should continue to work... */
                    141:   printf ("%s\n", PRIV_STATE());
                    142:   test_privs.change(ZPRIVS_RAISE);
                    143:   
                    144:   printf ("%s\n", PRIV_STATE());
                    145:   test_privs.change(ZPRIVS_LOWER);
                    146:   
                    147:   printf ("%s\n", PRIV_STATE());
                    148:   zprivs_get_ids (&ids);  
                    149:   
                    150:   printf ("terminating\n");
                    151:   return 0;
                    152: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>