16c35299cb8806586ffd9b85a54102f25d58bcea
9 #include <linux/hdreg.h>
13 #include <sys/inotify.h>
15 #include <sys/types.h>
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"
29 int mount_new(char *path
, char *dev
);
31 struct list_head mounts
;
34 struct list_head list
;
59 #define MAX_MOUNTED 32
60 #define MAX_MOUNT_NAME 32
62 char mounted
[MAX_MOUNTED
][3][MAX_MOUNT_NAME
];
63 int mounted_count
= 0;
64 extern char uci_path
[32];
66 void mount_dump_uci_state(void)
68 struct uci_context
*ctx
;
70 char mountd
[] = {"mountd"};
71 char type
[] = {"mountd_disc"};
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
)
80 struct mount
*q
= container_of(p
, struct mount
, list
);
84 ucix_add_section(ctx
, mountd
, q
->serial
, type
);
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
)
98 snprintf(t
, 64, "fs%d", atoi(&q
->dev
[3]));
99 ucix_add_option(ctx
, mountd
, q
->serial
, t
, fs_names
[q
->fs
]);
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
)));
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");
113 struct mount
* mount_find(char *name
, char *dev
)
116 list_for_each(p
, &mounts
)
118 struct mount
*q
= container_of(p
, struct mount
, list
);
120 if(!strcmp(q
->name
, name
))
123 if(!strcmp(q
->dev
, dev
))
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
)
133 char tmp
[64], tmp2
[64];
134 if(fs
<= MBR
|| fs
> NTFS
)
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
;
149 list_add(&mount
->list
, &mounts
);
150 if((!mount
->ignore
) && (mount
->fs
> MBR
) && (mount
->fs
<= NTFS
))
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
);
156 mount_new("/tmp/run/mountd/", dev
);
160 int mount_check_disc(char *disc
)
162 FILE *fp
= fopen("/proc/mounts", "r");
167 log_printf("error reading /proc/mounts");
171 while((fgets(tmp
, 256, fp
) > 0) && (avail
== -1))
175 t
= strstr(tmp
, " ");
180 l
= snprintf(tmp2
, 31, "/dev/%s", disc
);
182 if(!strncmp(tmp
, tmp2
, l
))
190 int mount_wait_for_disc(char *disc
)
195 int ret
= mount_check_disc(disc
);
203 int mount_new(char *path
, char *dev
)
209 mount
= mount_find(0, dev
);
212 log_printf("request for invalid path %s%s\n", path
, dev
);
215 if(mount
->ignore
|| mount
->mounted
|| mount
->fs
== EXTENDED
)
217 snprintf(tmp
, 256, "%s%s", path
, mount
->dev
);
218 log_printf("mounting %s\n", tmp
);
221 pid
= autofs_safe_fork();
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
);
229 if(mount
->fs
== EXT3
)
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
);
234 if(mount
->fs
== EXT2
)
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
);
239 if(mount
->fs
== HFSPLUS
)
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
);
244 if(mount
->fs
== NTFS
)
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
);
249 exit(WEXITSTATUS(ret
));
251 pid
= waitpid(pid
, &ret
, 0);
252 ret
= WEXITSTATUS(ret
);
253 log_printf("----------> mount ret = %d\n", ret
);
254 if(ret
&& (ret
!= 0xff))
256 if(mount_wait_for_disc(mount
->dev
) == 0)
259 mount_dump_uci_state();
264 int mount_remove(char *path
, char *dev
)
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
);
275 mount
= mount_find(0, dev
);
278 log_printf("finished unmounting\n");
279 mount_dump_uci_state();
283 int dir_sort(const void *a
, const void *b
)
288 int dir_filter(const struct dirent
*a
)
290 if(strstr(a
->d_name
, ":"))
295 char* mount_get_serial(char *dev
)
298 static char tmp2
[64];
300 static struct hd_driveid hd
;
303 snprintf(tmp
, 64, "/dev/%s", dev
);
304 disc
= open(tmp
, O_RDONLY
);
307 log_printf("Trying to open unknown disc\n");
310 i
= ioctl(disc
, HDIO_GET_IDENTITY
, &hd
);
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 */
318 struct dirent
**namelist
;
319 int n
= scandir("/sys/bus/scsi/devices/", &namelist
, dir_filter
, dir_sort
);
324 char *t
= strstr(namelist
[n
]->d_name
, ":");
332 id
= atoi(namelist
[n
]->d_name
);
334 sprintf(tmp3
, "/sys/bus/scsi/devices/%s/block:%s/", namelist
[n
]->d_name
, dev
);
335 ret
= stat(tmp3
, &buf
);
338 sprintf(tmp3
, "/sys/bus/scsi/devices/%s/block/%s/", namelist
[n
]->d_name
, dev
);
339 ret
= stat(tmp3
, &buf
);
344 snprintf(tmp2
, 64, "/proc/scsi/usb-storage/%d", id
);
345 fp
= fopen(tmp2
, "r");
348 while(fgets(tmp2
, 64, fp
) > 0)
350 serial
= strstr(tmp2
, "Serial Number:");
353 serial
+= strlen("Serial Number: ");
354 serial
[strlen(serial
) - 1] = '\0';
370 log_printf("could not find a serial number for the device %s\n", dev
);
372 /* serial string id is cheap, but makes the discs anonymous */
373 unsigned char uniq
[6];
374 int l
= strlen(serial
);
376 static char disc_id
[13];
377 memset(disc_id
, 0, 13);
379 for(i
= 0; i
< l
; i
++)
381 uniq
[i
%6] += serial
[i
];
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);
390 void mount_dev_add(char *dev
)
392 struct mount
*mount
= mount_find(0, dev
);
402 struct uci_context
*ctx
;
407 char sector_size
[64];
411 s
= mount_get_serial(name
);
414 snprintf(tmp
, 64, "part%s", &dev
[3]);
415 snprintf(node
, 64, "Disc-%s", &dev
[2]);
416 if(node
[5] >= 'a' && node
[5] <= 'z')
421 ctx
= ucix_init("mountd");
422 p
= ucix_get_option(ctx
, "mountd", s
, tmp
);
431 snprintf(node
, 64, "%s", p
);
436 snprintf(tmp
, 64, "/sys/class/block/%s/device/model", name
);
437 fp
= fopen(tmp
, "r");
440 snprintf(tmp
, 64, "/sys/block/%s/device/model", name
);
441 fp
= fopen(tmp
, "r");
444 snprintf(model
, 64, "unknown");
446 fgets(model
, 64, fp
);
447 model
[strlen(model
) - 1] = '\0';;
450 snprintf(tmp
, 64, "/sys/class/block/%s/device/vendor", name
);
451 fp
= fopen(tmp
, "r");
454 snprintf(tmp
, 64, "/sys/block/%s/device/vendor", name
);
455 fp
= fopen(tmp
, "r");
458 snprintf(vendor
, 64, "unknown");
460 fgets(vendor
, 64, fp
);
461 vendor
[strlen(vendor
) - 1] = '\0';
464 snprintf(tmp
, 64, "/sys/class/block/%s/device/rev", name
);
465 fp
= fopen(tmp
, "r");
468 snprintf(tmp
, 64, "/sys/block/%s/device/rev", name
);
469 fp
= fopen(tmp
, "r");
472 snprintf(rev
, 64, "unknown");
475 rev
[strlen(rev
) - 1] = '\0';
478 snprintf(tmp
, 64, "/sys/class/block/%s/size", dev
);
479 fp
= fopen(tmp
, "r");
482 snprintf(tmp
, 64, "/sys/block/%s/%s/size", name
, dev
);
483 fp
= fopen(tmp
, "r");
486 snprintf(size
, 64, "unknown");
489 size
[strlen(size
) - 1] = '\0';
494 snprintf(tmp
, 64, "/sys/block/%s/queue/hw_sector_size", tmp2
);
495 fp
= fopen(tmp
, "r");
497 snprintf(sector_size
, 64, "unknown");
499 fgets(sector_size
, 64, fp
);
500 sector_size
[strlen(sector_size
) - 1] = '\0';
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();
509 void mount_dev_del(char *dev
)
511 struct mount
*mount
= mount_find(0, dev
);
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
);
522 snprintf(tmp
, 64, "%s%s", uci_path
, mount
->name
);
524 mount_dump_uci_state();
529 void mount_dump_list(void)
532 list_for_each(p
, &mounts
)
534 struct mount
*q
= container_of(p
, struct mount
, list
);
535 log_printf("* %s %s %d\n", q
->name
, q
->dev
, q
->mounted
);
539 char* is_mounted(char *block
, char *path
)
542 for(i
= 0; i
< mounted_count
; i
++)
545 if(!strncmp(&mounted
[i
][0][0], block
, strlen(&mounted
[i
][0][0])))
546 return &mounted
[i
][0][0];
548 if(!strncmp(&mounted
[i
][1][1], &path
[1], strlen(&mounted
[i
][1][0])))
549 return &mounted
[i
][0][0];
554 void mount_check_mount_list(void)
556 FILE *fp
= fopen("/proc/mounts", "r");
561 log_printf("error reading /proc/mounts");
566 while(fgets(tmp
, 256, fp
) > 0)
569 t
= strstr(tmp
, " ");
575 strncpy(&mounted
[mounted_count
][0][0], tmp
, MAX_MOUNT_NAME
);
582 strncpy(&mounted
[mounted_count
][1][0], t
, MAX_MOUNT_NAME
);
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)
597 log_printf("found more than %d mounts \n", MAX_MOUNTED
);
602 /* FIXME: we need ore intelligence here */
603 int dir_filter2(const struct dirent
*a
)
605 if(/*strcmp(a->d_name, "sda") &&*/(!strncmp(a
->d_name
, "sd", 2)))
610 char block
[MAX_BLOCK
][MAX_BLOCK
];
613 int check_block(char *b
)
616 for(i
= 0; i
< blk_cnt
; i
++)
618 if(!strcmp(block
[i
], b
))
624 void mount_enum_drives(void)
626 struct dirent
**namelist
, **namelist2
;
627 int i
, n
= scandir("/sys/block/", &namelist
, dir_filter2
, dir_sort
);
634 if(blk_cnt
< MAX_BLOCK
)
638 snprintf(tmp
, 64, "/sys/block/%s/", namelist
[n
]->d_name
);
639 m
= scandir(tmp
, &namelist2
, dir_filter2
, dir_sort
);
642 strncpy(&block
[blk_cnt
][0], namelist2
[m
]->d_name
, MAX_BLOCK
);
655 struct mount
*q
= container_of(p
, struct mount
, list
);
657 struct uci_context
*ctx
;
660 snprintf(tmp
, 64, "part%s", &q
->dev
[3]);
661 ctx
= ucix_init("mountd");
662 t
= ucix_get_option(ctx
, "mountd", q
->serial
, tmp
);
670 } else if(!strcmp(t
, "1"))
672 if(strncmp(q
->name
, "Disc-", 5))
674 } else if(strcmp(q
->name
, t
))
679 if(!check_block(q
->dev
)||del
)
681 mount_dev_del(q
->dev
);
682 p
->prev
->next
= p
->next
;
683 p
->next
->prev
= p
->prev
;
685 log_printf("removing %s\n", q
->dev
);
686 snprintf(tmp
, 64, "%s%s", uci_path
, q
->name
);
688 system_printf("/etc/mountd/event remove %s %s", q
->dev
, q
->name
);
690 mount_dump_uci_state();
691 system_printf("/etc/fonstated/ReloadSamba");
695 for(i
= 0; i
< blk_cnt
; i
++)
696 mount_dev_add(block
[i
]);
699 void mount_check_enum(void)
701 waitpid(-1, 0, WNOHANG
);
705 void mount_init(void)
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();