add the 'y' and 'Y' options to ps, and add the 'iac' keyword. The 'y'
[dragonfly.git] / sys / checkpt / support.c
1 /*-
2  * Copyright (c) 2003 Kip Macy All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
14  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
16  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
17  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
18  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
19  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
20  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
21  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
22  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
23  * SUCH DAMAGE.
24  *
25  * $DragonFly: src/sys/checkpt/Attic/support.c,v 1.4 2004/06/05 16:34:04 dillon Exp $
26  */
27
28 /* prune the includes XXX */
29 #include <sys/types.h>
30 #include <sys/param.h>
31 #include <sys/proc.h>
32 #include <sys/module.h>
33 #include <sys/sysent.h>
34 #include <sys/kernel.h>
35 #include <sys/systm.h>
36
37 #include <sys/file.h>
38 /* only on dragonfly */
39 #include <sys/file2.h>
40 #include <sys/fcntl.h>
41 #include <sys/signal.h>
42 #include <vm/vm.h>
43 #include <sys/imgact_elf.h>
44 #include <sys/procfs.h>
45
46 #include <sys/lock.h>
47 #include <vm/pmap.h>
48 #include <vm/vm_map.h>
49
50 #include <sys/mman.h>
51 #include <sys/sysproto.h>
52 #include <sys/malloc.h>
53 #include <sys/stat.h>
54 #include <sys/uio.h>
55 #include <sys/namei.h>
56 #include <sys/vnode.h>
57 #include <machine/limits.h>
58 #include <i386/include/frame.h>
59 #include <sys/signalvar.h>
60 #include <sys/syslog.h>
61 #include <sys/sysctl.h>
62 #include <i386/include/sigframe.h>
63 #include <sys/exec.h>
64 #include <sys/unistd.h>
65 #include <sys/time.h>
66 #include <sys/kern_syscall.h>
67 #include "checkpt.h"
68
69
70 static char ckptfilename[MAXPATHLEN] = {"%N.ckpt"};
71 SYSCTL_STRING(_kern, OID_AUTO, ckptfile, CTLFLAG_RW, ckptfilename,
72               sizeof(ckptfilename), "process checkpoint name format string");
73
74 /*
75  * expand_name(name, uid, pid)
76  * Expand the name described in corefilename, using name, uid, and pid.
77  * corefilename is a printf-like string, with three format specifiers:
78  *      %N      name of process ("name")
79  *      %P      process id (pid)
80  *      %U      user id (uid)
81  * For example, "%N.core" is the default; they can be disabled completely
82  * by using "/dev/null", or all core files can be stored in "/cores/%U/%N-%P".
83  * This is controlled by the sysctl variable kern.corefile (see above).
84  *
85  * -- taken from the coredump code
86  */
87
88 char *
89 ckpt_expand_name(const char *name, uid_t uid, pid_t pid)
90 {
91         char *temp;
92         char *bp;
93         char buf[11];           /* Buffer for pid/uid -- max 4B */
94         int error;
95         int i;
96         int n;
97         char *format = ckptfilename;
98         size_t namelen;
99
100         temp = malloc(MAXPATHLEN + 1, M_TEMP, M_NOWAIT);
101         if (temp == NULL)
102                 return NULL;
103         namelen = strlen(name);
104         n = 0;
105         if (ckptfilename[0] != '/') {
106                 if ((bp = kern_getcwd(temp, MAXPATHLEN - 1, &error)) == NULL) {
107                         free(temp, M_TEMP);
108                         return NULL;
109                 }
110                 n = strlen(bp);
111                 bcopy(bp, temp, n + 1); /* normalize location of the path */
112                 temp[n++] = '/';
113                 temp[n] = '\0';
114         }
115         for (i= 0; n < MAXPATHLEN && format[i]; i++) {
116                 int l;
117                 switch (format[i]) {
118                 case '%':       /* Format character */
119                         i++;
120                         switch (format[i]) {
121                         case '%':
122                                 temp[n++] = '%';
123                                 break;
124                         case 'N':       /* process name */
125                                 if ((n + namelen) > MAXPATHLEN) {
126                                         log(LOG_ERR, "pid %d (%s), uid (%u):  Path `%s%s' is too long\n",
127                                             pid, name, uid, temp, name);
128                                         free(temp, M_TEMP);
129                                         return NULL;
130                                 }
131                                 memcpy(temp+n, name, namelen);
132                                 n += namelen;
133                                 break;
134                         case 'P':       /* process id */
135                                 l = sprintf(buf, "%u", pid);
136                                 if ((n + l) > MAXPATHLEN) {
137                                         log(LOG_ERR, "pid %d (%s), uid (%u):  Path `%s%s' is too long\n",
138                                             pid, name, uid, temp, name);
139                                         free(temp, M_TEMP);
140                                         return NULL;
141                                 }
142                                 memcpy(temp+n, buf, l);
143                                 n += l;
144                                 break;
145                         case 'U':       /* user id */
146                                 l = sprintf(buf, "%u", uid);
147                                 if ((n + l) > MAXPATHLEN) {
148                                         log(LOG_ERR, "pid %d (%s), uid (%u):  Path `%s%s' is too long\n",
149                                             pid, name, uid, temp, name);
150                                         free(temp, M_TEMP);
151                                         return NULL;
152                                 }
153                                 memcpy(temp+n, buf, l);
154                                 n += l;
155                                 break;
156                         default:
157                                 log(LOG_ERR, "Unknown format character %c in `%s'\n", format[i], format);
158                         }
159                         break;
160                 default:
161                         temp[n++] = format[i];
162                 }
163         }
164         temp[n] = '\0';
165         return temp;
166 }
167