usr.sbin/getpmac/getpmac.c

   1 /*-
   2  * Copyright (c) 2002 Networks Associates Technology, Inc.
   3  * All rights reserved.
   4  *
   5  * This software was developed for the FreeBSD Project by Network
   6  * Associates Laboratories, the Security Research Division of Network
   7  * Associates, Inc. under DARPA/SPAWAR contract N66001-01-C-8035
   8  * ("CBOSS"), as part of the DARPA CHATS research program.
   9  *
  10  * Redistribution and use in source and binary forms, with or without
  11  * modification, are permitted provided that the following conditions
  12  * are met:
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  * 3. The names of the authors may not be used to endorse or promote
  19  *    products derived from this software without specific prior written
  20  *    permission.
  21  *
  22  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  23  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  24  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  25  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  26  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  27  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  28  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  29  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  31  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  32  * SUCH DAMAGE.
  33  *
  34  * $FreeBSD$
  35  */
  36 #include <sys/types.h>
  37 #include <sys/mac.h>
  38
  39 #include <err.h>
  40 #include <paths.h>
  41 #include <stdio.h>
  42 #include <stdlib.h>
  43 #include <string.h>
  44 #include <sysexits.h>
  45 #include <unistd.h>
  46
  47 #define MAXELEMENTS     32
  48
  49 static void
  50 usage(void)
  51 {
  52
  53         fprintf(stderr, "getpmac [-l list,of,labels] [-p pid]\n");
  54         exit (EX_USAGE);
  55 }
  56
  57 int
  58 main(int argc, char *argv[])
  59 {
  60         char *labellist, *string;
  61         mac_t label;
  62         pid_t pid;
  63         int ch, error, pid_set;
  64
  65         pid_set = 0;
  66         pid = 0;
  67         labellist = NULL;
  68         while ((ch = getopt(argc, argv, "l:p:")) != -1) {
  69                 switch (ch) {
  70                 case 'l':
  71                         if (labellist != NULL)
  72                                 usage();
  73                         labellist = argv[optind - 1];
  74                         break;
  75                 case 'p':
  76                         if (pid_set)
  77                                 usage();
  78                         pid = atoi(argv[optind - 1]);
  79                         pid_set = 1;
  80                         break;
  81                 default:
  82                         usage();
  83                 }
  84
  85         }
  86
  87         argc -= optind;
  88         argv += optind;
  89
  90         if (argc != 0)
  91                 usage();
  92
  93         if (labellist != NULL)
  94                 error = mac_prepare(&label, labellist);
  95         else
  96                 error = mac_prepare_process_label(&label);
  97         if (error != 0) {
  98                 perror("mac_prepare");
  99                 return (-1);
 100         }
 101
 102         if (pid_set) {
 103                 error = mac_get_pid(pid, label);
 104                 if (error)
 105                         perror("mac_get_pid");
 106         } else {
 107                 error = mac_get_proc(label);
 108                 if (error)
 109                         perror("mac_get_proc");
 110         }
 111         if (error) {
 112                 mac_free(label);
 113                 exit (-1);
 114         }
 115         error = mac_to_text(label, &string);
 116         if (error != 0) {
 117                 perror("mac_to_text");
 118                 exit(EX_DATAERR);
 119         }
 120
 121         if (strlen(string) > 0)
 122                 printf("%s\n", string);
 123
 124         mac_free(label);
 125         free(string);
 126         exit(EX_OK);
 127 }