import v0.1
[project/mountd.git] / mount.c
1 #include <stdio.h>
2 #include <string.h>
3 #include <stdlib.h>
4 #include <unistd.h>
5 #include <sys/stat.h>
6 #include <sys/types.h>
7 #include <fcntl.h>
8 #include <sys/ioctl.h>
9 #include <linux/hdreg.h>
10 #include <scsi/sg.h>
11 #include <dirent.h>
12 #include <sys/wait.h>
13 #include <sys/inotify.h>
14 #include <sys/stat.h>
15 #include <sys/types.h>
16 #include <glob.h>
17 #include <libgen.h>
18 #include <poll.h>
19
20 #include "include/log.h"
21 #include "include/list.h"
22 #include "include/sys.h"
23 #include "include/signal.h"
24 #include "include/timer.h"
25 #include "include/autofs.h"
26 #include "include/ucix.h"
27 #include "include/fs.h"
28
29 int mount_new(char *path, char *dev);
30
31 struct list_head mounts;
32
33 struct mount {
34 struct list_head list;
35 char name[64];
36 char dev[64];
37 char serial[64];
38 char vendor[64];
39 char model[64];
40 char rev[64];
41 int mounted;
42 int ignore;
43 char size[64];
44 char sector_size[64];
45 int fs;
46 };
47
48 char *fs_names[] = {
49 "",
50 "",
51 "MBR",
52 "EXT2",
53 "EXT3",
54 "FAT",
55 "HFSPLUS",
56 "NTFS"
57 };
58
59 #define MAX_MOUNTED 32
60 #define MAX_MOUNT_NAME 32
61
62 char mounted[MAX_MOUNTED][3][MAX_MOUNT_NAME];
63 int mounted_count = 0;
64 extern char uci_path[32];
65
66 void mount_dump_uci_state(void)
67 {
68 struct uci_context *ctx;
69 struct list_head *p;
70 char mountd[] = {"mountd"};
71 char type[] = {"mountd_disc"};
72 int mounted = 0;
73 unsigned long long int size = 0;
74 unlink("/var/state/mountd");
75 ctx = ucix_init("mountd");
76 uci_set_savedir(ctx, "/var/state/");
77 ucix_add_option_int(ctx, mountd, mountd, "count", list_count(&mounts));
78 list_for_each(p, &mounts)
79 {
80 struct mount *q = container_of(p, struct mount, list);
81 char t[64];
82 if(q->fs == EXTENDED)
83 continue;
84 ucix_add_section(ctx, mountd, q->serial, type);
85 strcpy(t, q->dev);
86 t[3] = '\0';
87 ucix_add_option(ctx, mountd, q->serial, "disc", t);
88 ucix_add_option(ctx, mountd, q->serial, "sector_size", q->sector_size);
89 snprintf(t, 64, "part%dmounted", atoi(&q->dev[3]));
90 ucix_add_option(ctx, mountd, q->serial, t, (q->mounted)?("1"):("0"));
91 ucix_add_option(ctx, mountd, q->serial, "vendor", q->vendor);
92 ucix_add_option(ctx, mountd, q->serial, "model", q->model);
93 ucix_add_option(ctx, mountd, q->serial, "rev", q->rev);
94 snprintf(t, 64, "size%d", atoi(&q->dev[3]));
95 ucix_add_option(ctx, mountd, q->serial, t, q->size);
96 if(q->fs > MBR && q->fs <= NTFS)
97 {
98 snprintf(t, 64, "fs%d", atoi(&q->dev[3]));
99 ucix_add_option(ctx, mountd, q->serial, t, fs_names[q->fs]);
100 }
101 if(q->mounted)
102 mounted++;
103 if((!q->ignore) && q->size && q->sector_size)
104 size = size + (((unsigned long long int)atoi(q->size)) * ((unsigned long long int)atoi(q->sector_size)));
105 }
106 ucix_add_option_int(ctx, mountd, mountd, "mounted", mounted);
107 ucix_add_option_int(ctx, mountd, mountd, "total", size);
108 system_printf("echo -n %llu > /tmp/run/mountd_size", size);
109 ucix_save_state(ctx, "mountd");
110 ucix_cleanup(ctx);
111 }
112
113 struct mount* mount_find(char *name, char *dev)
114 {
115 struct list_head *p;
116 list_for_each(p, &mounts)
117 {
118 struct mount *q = container_of(p, struct mount, list);
119 if(name)
120 if(!strcmp(q->name, name))
121 return q;
122 if(dev)
123 if(!strcmp(q->dev, dev))
124 return q;
125 }
126 return 0;
127 }
128
129 void mount_add_list(char *name, char *dev, char *serial,
130 char *vendor, char *model, char *rev, int ignore, char *size, char *sector_size, int fs)
131 {
132 struct mount *mount;
133 char tmp[64], tmp2[64];
134 if(fs <= MBR || fs > NTFS)
135 return;
136 mount = malloc(sizeof(struct mount));
137 INIT_LIST_HEAD(&mount->list);
138 strncpy(mount->vendor, vendor, 64);
139 strncpy(mount->model, model, 64);
140 strncpy(mount->rev, rev, 64);
141 strncpy(mount->name, name, 64);
142 strncpy(mount->dev, dev, 64);
143 strncpy(mount->serial, serial, 64);
144 strncpy(mount->size, size, 64);
145 strncpy(mount->sector_size, sector_size, 64);
146 mount->ignore = ignore;
147 mount->mounted = 0;
148 mount->fs = fs;
149 list_add(&mount->list, &mounts);
150 if((!mount->ignore) && (mount->fs > MBR) && (mount->fs <= NTFS))
151 {
152 log_printf("new mount : %s -> %s (%s)\n", name, dev, fs_names[mount->fs]);
153 snprintf(tmp, 64, "%s%s", uci_path, name);
154 snprintf(tmp2, 64, "/tmp/run/mountd/%s", dev);
155 symlink(tmp2, tmp);
156 mount_new("/tmp/run/mountd/", dev);
157 }
158 }
159
160 int mount_check_disc(char *disc)
161 {
162 FILE *fp = fopen("/proc/mounts", "r");
163 char tmp[256];
164 int avail = -1;
165 if(!fp)
166 {
167 log_printf("error reading /proc/mounts");
168 fclose(fp);
169 return avail;
170 }
171 while((fgets(tmp, 256, fp) > 0) && (avail == -1))
172 {
173 char *t;
174 char tmp2[32];
175 t = strstr(tmp, " ");
176 if(t)
177 {
178 int l;
179 *t = '\0';
180 l = snprintf(tmp2, 31, "/dev/%s", disc);
181
182 if(!strncmp(tmp, tmp2, l))
183 avail = 0;
184 }
185 }
186 fclose(fp);
187 return avail;
188 }
189
190 int mount_wait_for_disc(char *disc)
191 {
192 int i = 10;
193 while(i--)
194 {
195 int ret = mount_check_disc(disc);
196 if(!ret)
197 return ret;
198 poll(0, 0, 100);
199 }
200 return -1;
201 }
202
203 int mount_new(char *path, char *dev)
204 {
205 struct mount *mount;
206 char tmp[256];
207 int ret = 1;
208 pid_t pid;
209 mount = mount_find(0, dev);
210 if(!mount)
211 {
212 log_printf("request for invalid path %s%s\n", path, dev);
213 return -1;
214 }
215 if(mount->ignore || mount->mounted || mount->fs == EXTENDED)
216 return -1;
217 snprintf(tmp, 256, "%s%s", path, mount->dev);
218 log_printf("mounting %s\n", tmp);
219 mkdir(tmp, 777);
220
221 pid = autofs_safe_fork();
222 if(!pid)
223 {
224 if(mount->fs == FAT)
225 {
226 log_printf("mount -t vfat -o rw,uid=1000,gid=1000 /dev/%s %s", mount->dev, tmp);
227 ret = system_printf("mount -t vfat -o rw,uid=1000,gid=1000 /dev/%s %s", mount->dev, tmp);
228 }
229 if(mount->fs == EXT3)
230 {
231 log_printf("mount -t ext3 -o rw,defaults /dev/%s %s", mount->dev, tmp);
232 ret = system_printf("mount -t ext3 -o rw,defaults /dev/%s %s", mount->dev, tmp);
233 }
234 if(mount->fs == EXT2)
235 {
236 log_printf("mount -t ext2 -o rw,defaults /dev/%s %s", mount->dev, tmp);
237 ret = system_printf("mount -t ext2 -o rw,defaults /dev/%s %s", mount->dev, tmp);
238 }
239 if(mount->fs == HFSPLUS)
240 {
241 log_printf("mount -t hfsplus -o rw,defaults,uid=1000,gid=1000 /dev/%s %s", mount->dev, tmp);
242 ret = system_printf("mount -t hfsplus -o rw,defaults,uid=1000,gid=1000 /dev/%s %s", mount->dev, tmp);
243 }
244 if(mount->fs == NTFS)
245 {
246 log_printf("ntfs-3g /dev/%s %s -o force", mount->dev, tmp);
247 ret = system_printf("ntfs-3g /dev/%s %s -o force", mount->dev, tmp);
248 }
249 exit(WEXITSTATUS(ret));
250 }
251 pid = waitpid(pid, &ret, 0);
252 ret = WEXITSTATUS(ret);
253 log_printf("----------> mount ret = %d\n", ret);
254 if(ret && (ret != 0xff))
255 return -1;
256 if(mount_wait_for_disc(mount->dev) == 0)
257 {
258 mount->mounted = 1;
259 mount_dump_uci_state();
260 } else return -1;
261 return 0;
262 }
263
264 int mount_remove(char *path, char *dev)
265 {
266 struct mount *mount;
267 char tmp[256];
268 int ret;
269 snprintf(tmp, 256, "%s%s", path, dev);
270 log_printf("%s has expired... unmounting\n", tmp);
271 ret = system_printf("/bin/umount %s", tmp);
272 if(ret != 0)
273 return 0;
274 rmdir(tmp);
275 mount = mount_find(0, dev);
276 if(mount)
277 mount->mounted = 0;
278 log_printf("finished unmounting\n");
279 mount_dump_uci_state();
280 return 0;
281 }
282
283 int dir_sort(const void *a, const void *b)
284 {
285 return 0;
286 }
287
288 int dir_filter(const struct dirent *a)
289 {
290 if(strstr(a->d_name, ":"))
291 return 1;
292 return 0;
293 }
294
295 char* mount_get_serial(char *dev)
296 {
297 static char tmp[64];
298 static char tmp2[64];
299 int disc;
300 static struct hd_driveid hd;
301 int i;
302 static char *serial;
303 snprintf(tmp, 64, "/dev/%s", dev);
304 disc = open(tmp, O_RDONLY);
305 if(!disc)
306 {
307 log_printf("Trying to open unknown disc\n");
308 return 0;
309 }
310 i = ioctl(disc, HDIO_GET_IDENTITY, &hd);
311 close(disc);
312 if(!i)
313 serial = (char*)hd.serial_no;
314 /* if we failed, it probably a usb storage device */
315 /* there must be a better way for this */
316 if(i)
317 {
318 struct dirent **namelist;
319 int n = scandir("/sys/bus/scsi/devices/", &namelist, dir_filter, dir_sort);
320 if(n > 0)
321 {
322 while(n--)
323 {
324 char *t = strstr(namelist[n]->d_name, ":");
325 if(t)
326 {
327 int id;
328 struct stat buf;
329 char tmp3[64];
330 int ret;
331 *t = 0;
332 id = atoi(namelist[n]->d_name);
333 *t = ':';
334 sprintf(tmp3, "/sys/bus/scsi/devices/%s/block:%s/", namelist[n]->d_name, dev);
335 ret = stat(tmp3, &buf);
336 if(ret)
337 {
338 sprintf(tmp3, "/sys/bus/scsi/devices/%s/block/%s/", namelist[n]->d_name, dev);
339 ret = stat(tmp3, &buf);
340 }
341 if(!ret)
342 {
343 FILE *fp;
344 snprintf(tmp2, 64, "/proc/scsi/usb-storage/%d", id);
345 fp = fopen(tmp2, "r");
346 if(fp)
347 {
348 while(fgets(tmp2, 64, fp) > 0)
349 {
350 serial = strstr(tmp2, "Serial Number:");
351 if(serial)
352 {
353 serial += strlen("Serial Number: ");
354 serial[strlen(serial) - 1] = '\0';
355 i = 0;
356 break;
357 }
358 }
359 fclose(fp);
360 }
361 }
362 }
363 free(namelist[n]);
364 }
365 free(namelist);
366 }
367 }
368 if(i)
369 {
370 log_printf("could not find a serial number for the device %s\n", dev);
371 } else {
372 /* serial string id is cheap, but makes the discs anonymous */
373 unsigned char uniq[6];
374 int l = strlen(serial);
375 int i;
376 static char disc_id[13];
377 memset(disc_id, 0, 13);
378 memset(uniq, 0, 6);
379 for(i = 0; i < l; i++)
380 {
381 uniq[i%6] += serial[i];
382 }
383 sprintf(disc_id, "%08X%02X%02X", *((unsigned int*)&uniq[0]), uniq[4], uniq[5]);
384 //log_printf("Serial number - %s %s\n", serial, disc_id);
385 return disc_id;
386 }
387 return 0;
388 }
389
390 void mount_dev_add(char *dev)
391 {
392 struct mount *mount = mount_find(0, dev);
393 if(!mount)
394 {
395 char node[64];
396 char name[64];
397 int ignore = 0;
398 char *s;
399 char tmp[64];
400 char tmp2[64];
401 char *p;
402 struct uci_context *ctx;
403 char vendor[64];
404 char model[64];
405 char rev[64];
406 char size[64];
407 char sector_size[64];
408 FILE *fp;
409 strcpy(name, dev);
410 name[3] = '\0';
411 s = mount_get_serial(name);
412 if(!s)
413 return;
414 snprintf(tmp, 64, "part%s", &dev[3]);
415 snprintf(node, 64, "Disc-%s", &dev[2]);
416 if(node[5] >= 'a' && node[5] <= 'z')
417 {
418 node[5] -= 'a';
419 node[5] += 'A';
420 }
421 ctx = ucix_init("mountd");
422 p = ucix_get_option(ctx, "mountd", s, tmp);
423 ucix_cleanup(ctx);
424 if(p)
425 {
426 if(strlen(p) == 1)
427 {
428 if(*p == '0')
429 ignore = 1;
430 } else {
431 snprintf(node, 64, "%s", p);
432 }
433 }
434 strcpy(name, dev);
435 name[3] = '\0';
436 snprintf(tmp, 64, "/sys/class/block/%s/device/model", name);
437 fp = fopen(tmp, "r");
438 if(!fp)
439 {
440 snprintf(tmp, 64, "/sys/block/%s/device/model", name);
441 fp = fopen(tmp, "r");
442 }
443 if(!fp)
444 snprintf(model, 64, "unknown");
445 else {
446 fgets(model, 64, fp);
447 model[strlen(model) - 1] = '\0';;
448 fclose(fp);
449 }
450 snprintf(tmp, 64, "/sys/class/block/%s/device/vendor", name);
451 fp = fopen(tmp, "r");
452 if(!fp)
453 {
454 snprintf(tmp, 64, "/sys/block/%s/device/vendor", name);
455 fp = fopen(tmp, "r");
456 }
457 if(!fp)
458 snprintf(vendor, 64, "unknown");
459 else {
460 fgets(vendor, 64, fp);
461 vendor[strlen(vendor) - 1] = '\0';
462 fclose(fp);
463 }
464 snprintf(tmp, 64, "/sys/class/block/%s/device/rev", name);
465 fp = fopen(tmp, "r");
466 if(!fp)
467 {
468 snprintf(tmp, 64, "/sys/block/%s/device/rev", name);
469 fp = fopen(tmp, "r");
470 }
471 if(!fp)
472 snprintf(rev, 64, "unknown");
473 else {
474 fgets(rev, 64, fp);
475 rev[strlen(rev) - 1] = '\0';
476 fclose(fp);
477 }
478 snprintf(tmp, 64, "/sys/class/block/%s/size", dev);
479 fp = fopen(tmp, "r");
480 if(!fp)
481 {
482 snprintf(tmp, 64, "/sys/block/%s/%s/size", name, dev);
483 fp = fopen(tmp, "r");
484 }
485 if(!fp)
486 snprintf(size, 64, "unknown");
487 else {
488 fgets(size, 64, fp);
489 size[strlen(size) - 1] = '\0';
490 fclose(fp);
491 }
492 strcpy(tmp2, dev);
493 tmp2[3] = '\0';
494 snprintf(tmp, 64, "/sys/block/%s/queue/hw_sector_size", tmp2);
495 fp = fopen(tmp, "r");
496 if(!fp)
497 snprintf(sector_size, 64, "unknown");
498 else {
499 fgets(sector_size, 64, fp);
500 sector_size[strlen(sector_size) - 1] = '\0';
501 fclose(fp);
502 }
503 snprintf(tmp, 64, "/dev/%s", dev);
504 mount_add_list(node, dev, s, vendor, model, rev, ignore, size, sector_size, detect_fs(tmp));
505 mount_dump_uci_state();
506 }
507 }
508
509 void mount_dev_del(char *dev)
510 {
511 struct mount *mount = mount_find(0, dev);
512 char tmp[256];
513 if(mount)
514 {
515 if(mount->mounted)
516 {
517 snprintf(tmp, 256, "%s%s", "/tmp/run/mountd/", mount->name);
518 log_printf("%s has dissappeared ... unmounting\n", tmp);
519 snprintf(tmp, 256, "%s%s", "/tmp/run/mountd/", mount->dev);
520 system_printf("/bin/umount %s", tmp);
521 rmdir(tmp);
522 snprintf(tmp, 64, "%s%s", uci_path, mount->name);
523 unlink(tmp);
524 mount_dump_uci_state();
525 }
526 }
527 }
528
529 void mount_dump_list(void)
530 {
531 struct list_head *p;
532 list_for_each(p, &mounts)
533 {
534 struct mount *q = container_of(p, struct mount, list);
535 log_printf("* %s %s %d\n", q->name, q->dev, q->mounted);
536 }
537 }
538
539 char* is_mounted(char *block, char *path)
540 {
541 int i;
542 for(i = 0; i < mounted_count; i++)
543 {
544 if(block)
545 if(!strncmp(&mounted[i][0][0], block, strlen(&mounted[i][0][0])))
546 return &mounted[i][0][0];
547 if(path)
548 if(!strncmp(&mounted[i][1][1], &path[1], strlen(&mounted[i][1][0])))
549 return &mounted[i][0][0];
550 }
551 return 0;
552 }
553
554 void mount_check_mount_list(void)
555 {
556 FILE *fp = fopen("/proc/mounts", "r");
557 char tmp[256];
558
559 if(!fp)
560 {
561 log_printf("error reading /proc/mounts");
562 fclose(fp);
563 return;
564 }
565 mounted_count = 0;
566 while(fgets(tmp, 256, fp) > 0)
567 {
568 char *t, *t2;
569 t = strstr(tmp, " ");
570 if(t)
571 {
572 *t = '\0';
573 t++;
574 } else t = tmp;
575 strncpy(&mounted[mounted_count][0][0], tmp, MAX_MOUNT_NAME);
576 t2 = strstr(t, " ");
577 if(t2)
578 {
579 *t2 = '\0';
580 t2++;
581 } else t2 = t;
582 strncpy(&mounted[mounted_count][1][0], t, MAX_MOUNT_NAME);
583 t = strstr(t2, " ");
584 if(t)
585 {
586 *t = '\0';
587 t++;
588 } else t = tmp;
589 strncpy(&mounted[mounted_count][2][0], t2, MAX_MOUNT_NAME);
590 /* printf("%s %s %s\n",
591 mounted[mounted_count][0],
592 mounted[mounted_count][1],
593 mounted[mounted_count][2]);*/
594 if(mounted_count < MAX_MOUNTED - 1)
595 mounted_count++;
596 else
597 log_printf("found more than %d mounts \n", MAX_MOUNTED);
598 }
599 fclose(fp);
600 }
601
602 /* FIXME: we need ore intelligence here */
603 int dir_filter2(const struct dirent *a)
604 {
605 if(/*strcmp(a->d_name, "sda") &&*/(!strncmp(a->d_name, "sd", 2)))
606 return 1;
607 return 0;
608 }
609 #define MAX_BLOCK 64
610 char block[MAX_BLOCK][MAX_BLOCK];
611 int blk_cnt = 0;
612
613 int check_block(char *b)
614 {
615 int i;
616 for(i = 0; i < blk_cnt; i++)
617 {
618 if(!strcmp(block[i], b))
619 return 1;
620 }
621 return 0;
622 }
623
624 void mount_enum_drives(void)
625 {
626 struct dirent **namelist, **namelist2;
627 int i, n = scandir("/sys/block/", &namelist, dir_filter2, dir_sort);
628 struct list_head *p;
629 blk_cnt = 0;
630 if(n > 0)
631 {
632 while(n--)
633 {
634 if(blk_cnt < MAX_BLOCK)
635 {
636 int m;
637 char tmp[64];
638 snprintf(tmp, 64, "/sys/block/%s/", namelist[n]->d_name);
639 m = scandir(tmp, &namelist2, dir_filter2, dir_sort);
640 while(m--)
641 {
642 strncpy(&block[blk_cnt][0], namelist2[m]->d_name, MAX_BLOCK);
643 blk_cnt++;
644 free(namelist2[m]);
645 }
646 free(namelist2);
647 }
648 free(namelist[n]);
649 }
650 free(namelist);
651 }
652 p = mounts.next;
653 while(p != &mounts)
654 {
655 struct mount *q = container_of(p, struct mount, list);
656 char tmp[64];
657 struct uci_context *ctx;
658 int del = 0;
659 char *t;
660 snprintf(tmp, 64, "part%s", &q->dev[3]);
661 ctx = ucix_init("mountd");
662 t = ucix_get_option(ctx, "mountd", q->serial, tmp);
663 ucix_cleanup(ctx);
664 if(t && !q->mounted)
665 {
666 if(!strcmp(t, "0"))
667 {
668 if(!q->ignore)
669 del = 1;
670 } else if(!strcmp(t, "1"))
671 {
672 if(strncmp(q->name, "Disc-", 5))
673 del = 1;
674 } else if(strcmp(q->name, t))
675 {
676 del = 1;
677 }
678 }
679 if(!check_block(q->dev)||del)
680 {
681 mount_dev_del(q->dev);
682 p->prev->next = p->next;
683 p->next->prev = p->prev;
684 p = p->next;
685 log_printf("removing %s\n", q->dev);
686 snprintf(tmp, 64, "%s%s", uci_path, q->name);
687 unlink(tmp);
688 system_printf("/etc/mountd/event remove %s %s", q->dev, q->name);
689 free(q);
690 mount_dump_uci_state();
691 system_printf("/etc/fonstated/ReloadSamba");
692 } else p = p->next;
693 }
694
695 for(i = 0; i < blk_cnt; i++)
696 mount_dev_add(block[i]);
697 }
698
699 void mount_check_enum(void)
700 {
701 waitpid(-1, 0, WNOHANG);
702 mount_enum_drives();
703 }
704
705 void mount_init(void)
706 {
707 INIT_LIST_HEAD(&mounts);
708 timer_add(mount_check_mount_list, 2);
709 timer_add(mount_check_enum, 1);
710 mount_check_mount_list();
711 }