16c35299cb8806586ffd9b85a54102f25d58bcea
[oweals/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 }