wmhdplop/devnames.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <errno.h>
   4 #include <string.h>
   5 #include <sys/types.h>
   6 #include <sys/stat.h>
   7 #include <sys/sysmacros.h>
   8 #include <unistd.h>
   9 #include <linux/major.h>
  10 #include <ctype.h>
  11 #include "global.h"
  12 #include "devnames.h"
  13
  14 static DiskList *dlist = NULL;
  15
  16 const char *stripdev(const char *s) {
  17   if (strncmp(s,"/dev",4) == 0) s+= 4;
  18   if (s[0] == '/') s++;
  19   return s;
  20 }
  21
  22 DiskList *find_dev(unsigned major, unsigned minor) {
  23   DiskList *dl;
  24   for (dl = dlist; dl; dl = dl->next) {
  25     if (dl->major == major && dl->minor == minor) return dl;
  26   }
  27   return NULL;
  28 }
  29
  30 DiskList *find_id(int hd_id, int part_id) {
  31   DiskList *dl;
  32   for (dl = dlist; dl; dl = dl->next) {
  33     if (((dl->hd_id == hd_id) || hd_id == -1) &&
  34         ((dl->part_id == part_id) || part_id == -1)) return dl;
  35   }
  36   return NULL;
  37 }
  38 /*
  39 DiskList *find_dev_by_name(const char *name) {
  40   DiskList *dl;
  41   const char *sname = stripdev(name);
  42   for (dl = dlist; dl; dl = dl->next) {
  43     if (strcmp(sname, dl->name)==0) return dl;
  44   }
  45   return NULL;
  46   }*/
  47
  48 DiskList *next_hd_in_list(DiskList *prev) {
  49   DiskList *dl = (prev ? prev->next : dlist);
  50   while (dl && IS_PARTITION(dl)) dl = dl->next;
  51   return dl;
  52 }
  53
  54 DiskList *first_hd_in_list() {
  55   return next_hd_in_list(NULL);
  56 }
  57
  58 DiskList *first_dev_in_list() {
  59   return dlist;
  60 }
  61
  62 int nb_hd_in_list() {
  63   int i=0;
  64   DiskList *dl;
  65   for (dl = first_hd_in_list(); dl; dl = next_hd_in_list(dl)) ++i;
  66   return i;
  67 }
  68
  69 int nb_dev_in_list() {
  70   int i=0;
  71   DiskList *dl;
  72   for (dl = dlist; dl; dl = dl->next) ++i;
  73   return i;
  74 }
  75
  76 /*int is_partition_name(const char *name) {
  77   int l = strlen(name);
  78   if (l && name[l-1] >= '0' && name[l-1] <= '9') return 1;
  79   return 0;
  80 }
  81 */
  82
  83 static const char *to_num_(unsigned minor) {
  84   static char id[16];
  85   if (minor == 0) return "";
  86   else snprintf(id,16,"%d",minor);
  87   return id;
  88 }
  89
  90 int device_info(unsigned major, unsigned minor, char *name, int *hd_id, int *part_id) {
  91   switch(major) {
  92 #ifdef SCSI_DISK0_MAJOR
  93     case (SCSI_DISK0_MAJOR):
  94 #else
  95     case (SCSI_DISK_MAJOR):
  96 #endif
  97       if (name) sprintf(name, "sd%c%s", "abcdefghijklmnop"[(minor)/16], to_num_(minor%16));
  98       if (hd_id) *hd_id = (minor)/16;
  99       if (part_id) *part_id = minor%16;
 100       return 1;
 101     case IDE0_MAJOR:
 102       if (name) sprintf(name, "hd%c%s", "ab"[(minor)/64], to_num_(minor%64));
 103       if (hd_id) *hd_id = 100 + (minor)/64;
 104       if (part_id) *part_id = minor%64;
 105       return 2;
 106     case IDE1_MAJOR:
 107       if (name) sprintf(name, "hd%c%s", "cd"[(minor)/64], to_num_(minor%64));
 108       if (hd_id) *hd_id = 200 + (minor)/64;
 109       if (part_id) *part_id = minor%64;
 110       return 3;
 111 #ifdef IDE2_MAJOR
 112     case IDE2_MAJOR:
 113       if (name) sprintf(name, "hd%c%s", "ef"[(minor)/64], to_num_(minor%64));
 114       if (hd_id) *hd_id = 300 + (minor)/64;
 115       if (part_id) *part_id = minor%64;
 116       return 4;
 117 #endif
 118 #ifdef IDE3_MAJOR
 119     case IDE3_MAJOR:
 120       if (name) sprintf(name, "hd%c%s", "gh"[(minor)/64], to_num_(minor%64));
 121       if (hd_id) *hd_id = 400 + (minor)/64;
 122       if (part_id) *part_id = minor%64;
 123       return 5;
 124 #endif
 125 #ifdef IDE4_MAJOR
 126     case IDE4_MAJOR:
 127       if (name) sprintf(name, "hd%c%s", "ij"[(minor)/64], to_num_(minor%64));
 128       if (hd_id) *hd_id = 400 + (minor)/64;
 129       if (part_id) *part_id = minor%64;
 130       return 6;
 131 #endif
 132
 133 #ifdef IDE5_MAJOR
 134     case IDE5_MAJOR:
 135       if (name) sprintf(name, "hd%c%s", "kl"[(minor)/64], to_num_(minor%64));
 136       if (hd_id) *hd_id = 400 + (minor)/64;
 137       if (part_id) *part_id = minor%64;
 138       return 7;
 139 #endif
 140
 141 #ifdef IDE6_MAJOR
 142     case IDE6_MAJOR:
 143       if (name) sprintf(name, "hd%c%s", "mn"[(minor)/64], to_num_(minor%64));
 144       if (hd_id) *hd_id = 400 + (minor)/64;
 145       if (part_id) *part_id = minor%64;
 146       return 8;
 147 #endif
 148
 149
 150 #ifdef MD_MAJOR
 151     case (MD_MAJOR):
 152       if (name) sprintf(name, "md%s", to_num_(minor));
 153       if (hd_id) *hd_id = 400 + minor;
 154       if (part_id) *part_id = 0;
 155       return 9;
 156 #endif
 157
 158 #ifdef BLOCK_EXT_MAJOR
 159     case (BLOCK_EXT_MAJOR):
 160       if (name) sprintf(name, "nvme0n%cp%s", "0123456789"[(minor/16)], to_num_(minor));
 161       if (hd_id) *hd_id = 400 + minor;
 162       if (part_id) *part_id = 0;
 163       return 10;
 164 #endif
 165   }
 166   if (name) *name = 0;
 167   if (hd_id) *hd_id = -1;
 168   if (part_id) *part_id = -1;
 169   return 0;
 170 }
 171
 172 int short_name_for_device(unsigned major, unsigned minor, char *name) {
 173   return device_info(major,minor,name,NULL,NULL);
 174 }
 175
 176 int is_partition(unsigned major, unsigned minor) {
 177   int part_id;
 178   if (!device_info(major,minor,NULL,NULL,&part_id)) return 0;
 179   return (part_id != 0);
 180 }
 181
 182 int device_id_from_name(const char *devname__, unsigned *pmajor, unsigned *pminor) {
 183   if (strlen(devname__)>=512) return -1;
 184   char *devname_, devname[512];
 185
 186   devname_ = str_substitute(devname__, "/dev/mapper", "/dev");
 187
 188   struct stat stat_buf;
 189   BLAHBLAH(1,printf("looking for %s in /dev..\n", devname_));
 190
 191   if (devname_[0] != '/') snprintf(devname,512,"/dev/%s", devname_);
 192   else snprintf(devname,512,"%s",devname_);
 193
 194   free(devname_); devname_ = 0;
 195
 196   if (lstat(devname,&stat_buf)) { BLAHBLAH(1,perror(devname)); return -1; }
 197   if (S_ISLNK(stat_buf.st_mode)) {
 198     devname_ = realpath(devname, NULL);
 199     if(!devname_) { BLAHBLAH(1,perror(devname)); return -1; }
 200     strncpy(devname, devname_, 512); devname[511] = 0;
 201     free(devname_);
 202     if (stat(devname,&stat_buf)) { BLAHBLAH(1,perror(devname)); return -1; }
 203   }
 204   if (!S_ISBLK(stat_buf.st_mode)) {
 205     fprintf(stderr, "%s is not a block device..\n", devname);
 206     return -2;
 207   }
 208   *pmajor = major(stat_buf.st_rdev);
 209   *pminor = minor(stat_buf.st_rdev);
 210   return 0;
 211 }
 212
 213
 214
 215 /* add a hard-drive if it is a regular ide/scsi/md disk/partition, or return NULL */
 216 static DiskList *create_device(unsigned major, unsigned minor, const char *mtab_name) {
 217   DiskList *dl;
 218   char dev_path[512];
 219   ALLOC_OBJ(dl);
 220   if (mtab_name && strlen(mtab_name)) dl->name = strdup(mtab_name);
 221   else {
 222     char s[512]; short_name_for_device(major,minor,s); dl->name = strdup(s);
 223   }
 224   BLAHBLAH(1, printf("create_device(major=%d, minor=%d, mtab_name=%s) : name=%s\n", major,minor,mtab_name,dl->name));
 225   dl->major = major; dl->minor = minor;
 226   if (device_info(major,minor,dev_path,&dl->hd_id,&dl->part_id) == 0) {
 227     BLAHBLAH(1, printf("(%d,%d) is not a known disc type..\n", major,minor));
 228     free(dl); return NULL;
 229   }
 230   dl->dev_path = malloc(strlen(dev_path)+6); assert(dl->dev_path);
 231   sprintf(dl->dev_path, "/dev/%s", dev_path);
 232   dl->next = NULL;
 233   if (dl->part_id == 0)
 234     dl->enable_hddtemp = 1;
 235   return dl;
 236 }
 237
 238 /* add a device (after checking that it is a known device type) */
 239 int add_device_by_name(const char *devname, const char *mtab_name) {
 240   unsigned major, minor;
 241
 242   BLAHBLAH(1,printf("add_device_by_name(%s,%s)\n", devname, mtab_name));
 243   if (device_id_from_name(devname, &major, &minor) != 0) return -1;
 244
 245   return add_device_by_id(major,minor,mtab_name);
 246 }
 247
 248 /* add a device (after checking that it is a known device type) */
 249 int add_device_by_id(unsigned major, unsigned minor, const char *mtab_name) {
 250   DiskList *dl, *next, *prev;
 251   BLAHBLAH(1,printf("add_device_by_id(%d,%d,%s)\n", major, minor,mtab_name));
 252   /* already known ? */
 253   if (find_dev(major,minor)) return -1;
 254     /* check for known ide/scsi/md disks and then create */
 255   dl = create_device(major, minor,  mtab_name); if (!dl) return -1;
 256
 257   for (next = dlist, prev = NULL; next; next=next->next) {
 258     if (dl->hd_id > next->hd_id || (dl->hd_id == next->hd_id && dl->part_id > next->part_id)) break;
 259     prev = next;
 260   }
 261   if (prev == NULL) {
 262     dl->next = dlist; dlist = dl;
 263   } else {
 264     dl->next = prev->next; prev->next = dl;
 265   }
 266   return 0;
 267 }
 268
 269 strlist *swaplst = NULL;
 270 void add_swap(const char *swapdev) {
 271   swaplst = strlist_ins(swaplst, swapdev);
 272 }
 273 strlist *swap_list() {
 274   return swaplst;
 275 }