diff options
| author | Steve French <sfrench@us.ibm.com> | 2006-01-12 17:47:08 -0500 |
|---|---|---|
| committer | Steve French <sfrench@us.ibm.com> | 2006-01-12 17:47:08 -0500 |
| commit | 94bc2be31a01a3055ec94176e595dfe208e92d3b (patch) | |
| tree | ebfbe81c6718a6390bfa1b99c6d228237d818576 /arch/um/kernel/umid.c | |
| parent | c32a0b689cb9cc160cfcd19735bbf50bb70c6ef4 (diff) | |
| parent | 58cba4650a7a414eabd2b40cc9d8e45fcdf192d9 (diff) | |
Merge with /pub/scm/linux/kernel/git/torvalds/linux-2.6.git
Signed-off-by: Steve French <sfrench@us.ibm.com>
Diffstat (limited to 'arch/um/kernel/umid.c')
| -rw-r--r-- | arch/um/kernel/umid.c | 323 |
1 files changed, 15 insertions, 308 deletions
diff --git a/arch/um/kernel/umid.c b/arch/um/kernel/umid.c index 0b21d59ba0cd..4eaee823bfd2 100644 --- a/arch/um/kernel/umid.c +++ b/arch/um/kernel/umid.c | |||
| @@ -3,61 +3,30 @@ | |||
| 3 | * Licensed under the GPL | 3 | * Licensed under the GPL |
| 4 | */ | 4 | */ |
| 5 | 5 | ||
| 6 | #include <stdio.h> | 6 | #include "asm/errno.h" |
| 7 | #include <unistd.h> | ||
| 8 | #include <errno.h> | ||
| 9 | #include <string.h> | ||
| 10 | #include <stdlib.h> | ||
| 11 | #include <dirent.h> | ||
| 12 | #include <signal.h> | ||
| 13 | #include <sys/stat.h> | ||
| 14 | #include <sys/param.h> | ||
| 15 | #include "user.h" | ||
| 16 | #include "umid.h" | ||
| 17 | #include "init.h" | 7 | #include "init.h" |
| 18 | #include "os.h" | 8 | #include "os.h" |
| 19 | #include "user_util.h" | 9 | #include "kern.h" |
| 20 | #include "choose-mode.h" | 10 | #include "linux/kernel.h" |
| 21 | 11 | ||
| 22 | #define UMID_LEN 64 | 12 | /* Changed by set_umid_arg */ |
| 23 | #define UML_DIR "~/.uml/" | ||
| 24 | |||
| 25 | /* Changed by set_umid and make_umid, which are run early in boot */ | ||
| 26 | static char umid[UMID_LEN] = { 0 }; | ||
| 27 | |||
| 28 | /* Changed by set_uml_dir and make_uml_dir, which are run early in boot */ | ||
| 29 | static char *uml_dir = UML_DIR; | ||
| 30 | |||
| 31 | /* Changed by set_umid */ | ||
| 32 | static int umid_is_random = 1; | ||
| 33 | static int umid_inited = 0; | 13 | static int umid_inited = 0; |
| 34 | /* Have we created the files? Should we remove them? */ | ||
| 35 | static int umid_owned = 0; | ||
| 36 | 14 | ||
| 37 | static int make_umid(int (*printer)(const char *fmt, ...)); | 15 | static int __init set_umid_arg(char *name, int *add) |
| 38 | |||
| 39 | static int __init set_umid(char *name, int is_random, | ||
| 40 | int (*printer)(const char *fmt, ...)) | ||
| 41 | { | 16 | { |
| 42 | if(umid_inited){ | 17 | int err; |
| 43 | (*printer)("Unique machine name can't be set twice\n"); | ||
| 44 | return(-1); | ||
| 45 | } | ||
| 46 | 18 | ||
| 47 | if(strlen(name) > UMID_LEN - 1) | 19 | if(umid_inited) |
| 48 | (*printer)("Unique machine name is being truncated to %d " | 20 | return 0; |
| 49 | "characters\n", UMID_LEN); | ||
| 50 | strlcpy(umid, name, sizeof(umid)); | ||
| 51 | 21 | ||
| 52 | umid_is_random = is_random; | ||
| 53 | umid_inited = 1; | ||
| 54 | return 0; | ||
| 55 | } | ||
| 56 | |||
| 57 | static int __init set_umid_arg(char *name, int *add) | ||
| 58 | { | ||
| 59 | *add = 0; | 22 | *add = 0; |
| 60 | return(set_umid(name, 0, printf)); | 23 | err = set_umid(name); |
| 24 | if(err == -EEXIST) | ||
| 25 | printf("umid '%s' already in use\n", name); | ||
| 26 | else if(!err) | ||
| 27 | umid_inited = 1; | ||
| 28 | |||
| 29 | return 0; | ||
| 61 | } | 30 | } |
| 62 | 31 | ||
| 63 | __uml_setup("umid=", set_umid_arg, | 32 | __uml_setup("umid=", set_umid_arg, |
| @@ -66,265 +35,3 @@ __uml_setup("umid=", set_umid_arg, | |||
| 66 | " is used for naming the pid file and management console socket.\n\n" | 35 | " is used for naming the pid file and management console socket.\n\n" |
| 67 | ); | 36 | ); |
| 68 | 37 | ||
| 69 | int __init umid_file_name(char *name, char *buf, int len) | ||
| 70 | { | ||
| 71 | int n; | ||
| 72 | |||
| 73 | if(!umid_inited && make_umid(printk)) return(-1); | ||
| 74 | |||
| 75 | n = strlen(uml_dir) + strlen(umid) + strlen(name) + 1; | ||
| 76 | if(n > len){ | ||
| 77 | printk("umid_file_name : buffer too short\n"); | ||
| 78 | return(-1); | ||
| 79 | } | ||
| 80 | |||
| 81 | sprintf(buf, "%s%s/%s", uml_dir, umid, name); | ||
| 82 | return(0); | ||
| 83 | } | ||
| 84 | |||
| 85 | extern int tracing_pid; | ||
| 86 | |||
| 87 | static void __init create_pid_file(void) | ||
| 88 | { | ||
| 89 | char file[strlen(uml_dir) + UMID_LEN + sizeof("/pid\0")]; | ||
| 90 | char pid[sizeof("nnnnn\0")]; | ||
| 91 | int fd, n; | ||
| 92 | |||
| 93 | if(umid_file_name("pid", file, sizeof(file))) | ||
| 94 | return; | ||
| 95 | |||
| 96 | fd = os_open_file(file, of_create(of_excl(of_rdwr(OPENFLAGS()))), | ||
| 97 | 0644); | ||
| 98 | if(fd < 0){ | ||
| 99 | printf("Open of machine pid file \"%s\" failed: %s\n", | ||
| 100 | file, strerror(-fd)); | ||
| 101 | return; | ||
| 102 | } | ||
| 103 | |||
| 104 | sprintf(pid, "%d\n", os_getpid()); | ||
| 105 | n = os_write_file(fd, pid, strlen(pid)); | ||
| 106 | if(n != strlen(pid)) | ||
| 107 | printf("Write of pid file failed - err = %d\n", -n); | ||
| 108 | os_close_file(fd); | ||
| 109 | } | ||
| 110 | |||
| 111 | static int actually_do_remove(char *dir) | ||
| 112 | { | ||
| 113 | DIR *directory; | ||
| 114 | struct dirent *ent; | ||
| 115 | int len; | ||
| 116 | char file[256]; | ||
| 117 | |||
| 118 | directory = opendir(dir); | ||
| 119 | if(directory == NULL){ | ||
| 120 | printk("actually_do_remove : couldn't open directory '%s', " | ||
| 121 | "errno = %d\n", dir, errno); | ||
| 122 | return(1); | ||
| 123 | } | ||
| 124 | while((ent = readdir(directory)) != NULL){ | ||
| 125 | if(!strcmp(ent->d_name, ".") || !strcmp(ent->d_name, "..")) | ||
| 126 | continue; | ||
| 127 | len = strlen(dir) + sizeof("/") + strlen(ent->d_name) + 1; | ||
| 128 | if(len > sizeof(file)){ | ||
| 129 | printk("Not deleting '%s' from '%s' - name too long\n", | ||
| 130 | ent->d_name, dir); | ||
| 131 | continue; | ||
| 132 | } | ||
| 133 | sprintf(file, "%s/%s", dir, ent->d_name); | ||
| 134 | if(unlink(file) < 0){ | ||
| 135 | printk("actually_do_remove : couldn't remove '%s' " | ||
| 136 | "from '%s', errno = %d\n", ent->d_name, dir, | ||
| 137 | errno); | ||
| 138 | return(1); | ||
| 139 | } | ||
| 140 | } | ||
| 141 | if(rmdir(dir) < 0){ | ||
| 142 | printk("actually_do_remove : couldn't rmdir '%s', " | ||
| 143 | "errno = %d\n", dir, errno); | ||
| 144 | return(1); | ||
| 145 | } | ||
| 146 | return(0); | ||
| 147 | } | ||
| 148 | |||
| 149 | void remove_umid_dir(void) | ||
| 150 | { | ||
| 151 | char dir[strlen(uml_dir) + UMID_LEN + 1]; | ||
| 152 | if (!umid_owned) | ||
| 153 | return; | ||
| 154 | |||
| 155 | sprintf(dir, "%s%s", uml_dir, umid); | ||
| 156 | actually_do_remove(dir); | ||
| 157 | } | ||
| 158 | |||
| 159 | char *get_umid(int only_if_set) | ||
| 160 | { | ||
| 161 | if(only_if_set && umid_is_random) | ||
| 162 | return NULL; | ||
| 163 | return umid; | ||
| 164 | } | ||
| 165 | |||
| 166 | static int not_dead_yet(char *dir) | ||
| 167 | { | ||
| 168 | char file[strlen(uml_dir) + UMID_LEN + sizeof("/pid\0")]; | ||
| 169 | char pid[sizeof("nnnnn\0")], *end; | ||
| 170 | int dead, fd, p, n; | ||
| 171 | |||
| 172 | sprintf(file, "%s/pid", dir); | ||
| 173 | dead = 0; | ||
| 174 | fd = os_open_file(file, of_read(OPENFLAGS()), 0); | ||
| 175 | if(fd < 0){ | ||
| 176 | if(fd != -ENOENT){ | ||
| 177 | printk("not_dead_yet : couldn't open pid file '%s', " | ||
| 178 | "err = %d\n", file, -fd); | ||
| 179 | return(1); | ||
| 180 | } | ||
| 181 | dead = 1; | ||
| 182 | } | ||
| 183 | if(fd > 0){ | ||
| 184 | n = os_read_file(fd, pid, sizeof(pid)); | ||
| 185 | if(n < 0){ | ||
| 186 | printk("not_dead_yet : couldn't read pid file '%s', " | ||
| 187 | "err = %d\n", file, -n); | ||
| 188 | return(1); | ||
| 189 | } | ||
| 190 | p = strtoul(pid, &end, 0); | ||
| 191 | if(end == pid){ | ||
| 192 | printk("not_dead_yet : couldn't parse pid file '%s', " | ||
| 193 | "errno = %d\n", file, errno); | ||
| 194 | dead = 1; | ||
| 195 | } | ||
| 196 | if(((kill(p, 0) < 0) && (errno == ESRCH)) || | ||
| 197 | (p == CHOOSE_MODE(tracing_pid, os_getpid()))) | ||
| 198 | dead = 1; | ||
| 199 | } | ||
| 200 | if(!dead) | ||
| 201 | return(1); | ||
| 202 | return(actually_do_remove(dir)); | ||
| 203 | } | ||
| 204 | |||
| 205 | static int __init set_uml_dir(char *name, int *add) | ||
| 206 | { | ||
| 207 | if((strlen(name) > 0) && (name[strlen(name) - 1] != '/')){ | ||
| 208 | uml_dir = malloc(strlen(name) + 2); | ||
| 209 | if(uml_dir == NULL){ | ||
| 210 | printf("Failed to malloc uml_dir - error = %d\n", | ||
| 211 | errno); | ||
| 212 | uml_dir = name; | ||
| 213 | /* Return 0 here because do_initcalls doesn't look at | ||
| 214 | * the return value. | ||
| 215 | */ | ||
| 216 | return(0); | ||
| 217 | } | ||
| 218 | sprintf(uml_dir, "%s/", name); | ||
| 219 | } | ||
| 220 | else uml_dir = name; | ||
| 221 | return(0); | ||
| 222 | } | ||
| 223 | |||
| 224 | static int __init make_uml_dir(void) | ||
| 225 | { | ||
| 226 | char dir[MAXPATHLEN + 1] = { '\0' }; | ||
| 227 | int len; | ||
| 228 | |||
| 229 | if(*uml_dir == '~'){ | ||
| 230 | char *home = getenv("HOME"); | ||
| 231 | |||
| 232 | if(home == NULL){ | ||
| 233 | printf("make_uml_dir : no value in environment for " | ||
| 234 | "$HOME\n"); | ||
| 235 | exit(1); | ||
| 236 | } | ||
| 237 | strlcpy(dir, home, sizeof(dir)); | ||
| 238 | uml_dir++; | ||
| 239 | } | ||
| 240 | strlcat(dir, uml_dir, sizeof(dir)); | ||
| 241 | len = strlen(dir); | ||
| 242 | if (len > 0 && dir[len - 1] != '/') | ||
| 243 | strlcat(dir, "/", sizeof(dir)); | ||
| 244 | |||
| 245 | uml_dir = malloc(strlen(dir) + 1); | ||
| 246 | if (uml_dir == NULL) { | ||
| 247 | printf("make_uml_dir : malloc failed, errno = %d\n", errno); | ||
| 248 | exit(1); | ||
| 249 | } | ||
| 250 | strcpy(uml_dir, dir); | ||
| 251 | |||
| 252 | if((mkdir(uml_dir, 0777) < 0) && (errno != EEXIST)){ | ||
| 253 | printf("Failed to mkdir %s: %s\n", uml_dir, strerror(errno)); | ||
| 254 | return(-1); | ||
| 255 | } | ||
| 256 | return 0; | ||
| 257 | } | ||
| 258 | |||
| 259 | static int __init make_umid(int (*printer)(const char *fmt, ...)) | ||
| 260 | { | ||
| 261 | int fd, err; | ||
| 262 | char tmp[strlen(uml_dir) + UMID_LEN + 1]; | ||
| 263 | |||
| 264 | strlcpy(tmp, uml_dir, sizeof(tmp)); | ||
| 265 | |||
| 266 | if(!umid_inited){ | ||
| 267 | strcat(tmp, "XXXXXX"); | ||
| 268 | fd = mkstemp(tmp); | ||
| 269 | if(fd < 0){ | ||
| 270 | (*printer)("make_umid - mkstemp(%s) failed: %s\n", | ||
| 271 | tmp,strerror(errno)); | ||
| 272 | return(1); | ||
| 273 | } | ||
| 274 | |||
| 275 | os_close_file(fd); | ||
| 276 | /* There's a nice tiny little race between this unlink and | ||
| 277 | * the mkdir below. It'd be nice if there were a mkstemp | ||
| 278 | * for directories. | ||
| 279 | */ | ||
| 280 | unlink(tmp); | ||
| 281 | set_umid(&tmp[strlen(uml_dir)], 1, printer); | ||
| 282 | } | ||
| 283 | |||
| 284 | sprintf(tmp, "%s%s", uml_dir, umid); | ||
| 285 | |||
| 286 | err = mkdir(tmp, 0777); | ||
| 287 | if(err < 0){ | ||
| 288 | if(errno == EEXIST){ | ||
| 289 | if(not_dead_yet(tmp)){ | ||
| 290 | (*printer)("umid '%s' is in use\n", umid); | ||
| 291 | umid_owned = 0; | ||
| 292 | return(-1); | ||
| 293 | } | ||
| 294 | err = mkdir(tmp, 0777); | ||
| 295 | } | ||
| 296 | } | ||
| 297 | if(err < 0){ | ||
| 298 | (*printer)("Failed to create %s - errno = %d\n", umid, errno); | ||
| 299 | return(-1); | ||
| 300 | } | ||
| 301 | |||
| 302 | umid_owned = 1; | ||
| 303 | return 0; | ||
| 304 | } | ||
| 305 | |||
| 306 | __uml_setup("uml_dir=", set_uml_dir, | ||
| 307 | "uml_dir=<directory>\n" | ||
| 308 | " The location to place the pid and umid files.\n\n" | ||
| 309 | ); | ||
| 310 | |||
| 311 | static int __init make_umid_setup(void) | ||
| 312 | { | ||
| 313 | /* one function with the ordering we need ... */ | ||
| 314 | make_uml_dir(); | ||
| 315 | make_umid(printf); | ||
| 316 | create_pid_file(); | ||
| 317 | return 0; | ||
| 318 | } | ||
| 319 | __uml_postsetup(make_umid_setup); | ||
| 320 | |||
| 321 | /* | ||
| 322 | * Overrides for Emacs so that we follow Linus's tabbing style. | ||
| 323 | * Emacs will notice this stuff at the end of the file and automatically | ||
| 324 | * adjust the settings for this buffer only. This must remain at the end | ||
| 325 | * of the file. | ||
| 326 | * --------------------------------------------------------------------------- | ||
| 327 | * Local variables: | ||
| 328 | * c-file-style: "linux" | ||
| 329 | * End: | ||
| 330 | */ | ||
