BSD 4_3_Net_2 release

[unix-history] / usr / src / usr.bin / ktrace / ktrace / ktrace.c

diff --git a/usr/src/usr.bin/ktrace/ktrace/ktrace.c b/usr/src/usr.bin/ktrace/ktrace/ktrace.c
index 2af9e19..8f25878 100644 (file)
--- a/usr/src/usr.bin/ktrace/ktrace/ktrace.c
+++ b/usr/src/usr.bin/ktrace/ktrace/ktrace.c
@@ -2,19 +2,33 @@
  * Copyright (c) 1988 The Regents of the University of California.
  * All rights reserved.
  *
  * Copyright (c) 1988 The Regents of the University of California.
  * All rights reserved.
  *

- * Redistribution and use in source and binary forms are permitted provided
- * that: (1) source distributions retain this entire copyright notice and
- * comment, and (2) distributions including binaries display the following
- * acknowledgement:  ``This product includes software developed by the
- * University of California, Berkeley and its contributors'' in the
- * documentation or other materials provided with the distribution and in
- * all advertising materials mentioning features or use of this software.
- * Neither the name of the University nor the names of its contributors may
- * be used to endorse or promote products derived from this software without
- * specific prior written permission.
- * THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR IMPLIED
- * WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF
- * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ * 3. All advertising materials mentioning features or use of this software
+ *    must display the following acknowledgement:
+ *     This product includes software developed by the University of
+ *     California, Berkeley and its contributors.
+ * 4. Neither the name of the University nor the names of its contributors
+ *    may be used to endorse or promote products derived from this software
+ *    without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
+ * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.

  */
 
 #ifndef lint
  */
 
 #ifndef lint


@@ -24,108 +38,138 @@ char copyright[] =
 #endif /* not lint */
 
 #ifndef lint
 #endif /* not lint */
 
 #ifndef lint

-static char sccsid[] = "@(#)ktrace.c   1.5 (Berkeley) 6/29/90";
+static char sccsid[] = "@(#)ktrace.c   5.2 (Berkeley) 3/5/91";

 #endif /* not lint */
 
 #endif /* not lint */
 

+#include <sys/param.h>
+#include <sys/stat.h>
+#include <sys/file.h>
+#include <sys/time.h>
+#include <sys/errno.h>
+#include <sys/uio.h>
+#include <sys/ktrace.h>
+#include <stdio.h>

 #include "ktrace.h"
 
 #include "ktrace.h"
 

-#define USAGE \
- "usage: ktrace [-acid] [-f trfile] [-t trpoints] [-p pid] [-g pgid]\n\
-       trops: c = syscalls, n = namei, g = generic-i/o, a = everything\n\
-       ktrace -C (clear everthing)\n"
-       
-
-char   *tracefile = DEF_TRACEFILE;
-int    append, clear, descend, inherit;
-

 main(argc, argv)
 main(argc, argv)

-       char *argv[];
+       int argc;
+       char **argv;

 {
        extern int optind;
        extern char *optarg;
 {
        extern int optind;
        extern char *optarg;

-       int trpoints = ALL_POINTS;
-       int ops = 0;
-       int pid = 0;
-       int ch;
+       enum { NOTSET, CLEAR, CLEARALL } clear;
+       int append, ch, fd, inherit, ops, pid, pidset, trpoints;
+       char *tracefile;

 
 

-       while ((ch = getopt(argc,argv,"Cacdp:g:if:t:")) != EOF)
+       clear = NOTSET;
+       append = ops = pidset = 0;
+       trpoints = ALL_POINTS;
+       tracefile = DEF_TRACEFILE;
+       while ((ch = getopt(argc,argv,"aCcdf:g:ip:t:")) != EOF)

                switch((char)ch) {
                switch((char)ch) {

+               case 'a':
+                       append = 1;
+                       break;

                case 'C':
                case 'C':

-                       clear = 2;
+                       clear = CLEARALL;
+                       pidset = 1;

                        break;
                case 'c':
                        break;
                case 'c':

-                       clear = 1;
+                       clear = CLEAR;

                        break;
                case 'd':
                        ops |= KTRFLAG_DESCEND;
                        break;
                        break;
                case 'd':
                        ops |= KTRFLAG_DESCEND;
                        break;

-               case 't':
-                       trpoints = getpoints(optarg);
-                       if (trpoints < 0) {
-                               fprintf(stderr, 
-                                   "ktrace: unknown facility in %s\n",
-                                    optarg);
-                               exit(1);
-                       }
-                       break;
-               case 'p':
-                       pid = atoi(optarg);
+               case 'f':
+                       tracefile = optarg;

                        break;
                case 'g':
                        break;
                case 'g':

-                       pid = -atoi(optarg);
+                       pid = -rpid(optarg);
+                       pidset = 1;

                        break;
                case 'i':
                        break;
                case 'i':

-                       inherit++;
+                       inherit = 1;

                        break;
                        break;

-               case 'f':
-                       tracefile = optarg;
+               case 'p':
+                       pid = rpid(optarg);
+                       pidset = 1;

                        break;
                        break;

-               case 'a':
-                       append++;
+               case 't':
+                       trpoints = getpoints(optarg);
+                       if (trpoints < 0) {
+                               (void)fprintf(stderr, 
+                                   "ktrace: unknown facility in %s\n", optarg);
+                               usage();
+                       }

                        break;
                default:
                        break;
                default:

-                       fprintf(stderr,"usage: \n",*argv);
-                       exit(-1);
+                       usage();

                }
                }

-       argv += optind, argc -= optind;
+       argv += optind;
+       argc -= optind;

        
        

+       if (pidset && *argv || !pidset && !*argv)
+               usage();
+                       

        if (inherit)
                trpoints |= KTRFAC_INHERIT;
        if (inherit)
                trpoints |= KTRFAC_INHERIT;

-       if (clear) {                    /* untrace something */
-               if (clear == 2) {       /* -C */
+
+       if (clear != NOTSET) {
+               if (clear == CLEARALL) {

                        ops = KTROP_CLEAR | KTRFLAG_DESCEND;
                        pid = 1;
                        ops = KTROP_CLEAR | KTRFLAG_DESCEND;
                        pid = 1;

-               } else {
+               } else

                        ops |= pid ? KTROP_CLEAR : KTROP_CLEARFILE;
                        ops |= pid ? KTROP_CLEAR : KTROP_CLEARFILE;

-               }
-               if (ktrace(tracefile, ops, trpoints, pid) < 0) {
-                       perror("ktrace");
-                       exit(1);
-               }
+
+               if (ktrace(tracefile, ops, trpoints, pid) < 0)
+                       error(tracefile);

                exit(0);
        }
 
                exit(0);
        }
 

-       if (pid == 0 && !*argv) {       /* nothing to trace */
-               fprintf(stderr, USAGE);
-               exit(1);
-       }
-                       
-       close(open(tracefile, O_WRONLY | O_CREAT, 0666));
-       if (!append)
-               close(open(tracefile, O_WRONLY | O_TRUNC));
-       if (!*argv) {
-               if (ktrace(tracefile, ops, trpoints, pid) < 0) {
-                       perror("ktrace");
-                       exit(1);
-               }
-       } else {
-               pid = getpid();
-               if (ktrace(tracefile, ops, trpoints, pid) < 0) {
-                       perror("ktrace");
-                       exit(1);
-               }
+       if ((fd = open(tracefile, O_CREAT | O_WRONLY | (append ? 0 : O_TRUNC),
+           DEFFILEMODE)) < 0)
+               error(tracefile);
+       (void)close(fd);
+
+       if (*argv) { 
+               if (ktrace(tracefile, ops, trpoints, getpid()) < 0)
+                       error();

                execvp(argv[0], &argv[0]);
                execvp(argv[0], &argv[0]);

-               perror("ktrace: exec failed");
+               error(argv[0]);
+               exit(1);

        }
        }

+       else if (ktrace(tracefile, ops, trpoints, pid) < 0)
+               error(tracefile);

        exit(0);
 }
        exit(0);
 }

+
+rpid(p)
+       char *p;
+{
+       static int first;
+
+       if (first++) {
+               (void)fprintf(stderr,
+                   "ktrace: only one -g or -p flag is permitted.\n");
+               usage();
+       }
+       if (!*p) {
+               (void)fprintf(stderr, "ktrace: illegal process id.\n");
+               usage();
+       }
+       return(atoi(p));
+}
+
+error(name)
+       char *name;
+{
+       (void)fprintf(stderr, "ktrace: %s: %s.\n", name, strerror(errno));
+       exit(1);
+}
+
+usage()
+{
+       (void)fprintf(stderr,
+"usage:\tktrace [-aCcid] [-f trfile] [-g pgid] [-p pid] [-t [acgn]\n\tktrace [-aCcid] [-f trfile] [-t [acgn] command\n");
+       exit(1);
+}