Return to test-privs.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / quagga / tests

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

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: }
        !            77: 
        !            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: }