ioctls/ioctls.c

   1 /* trinity ioctl() support routines */
   2
   3 #include <string.h>
   4 #include <stdio.h>
   5
   6 #include "trinity.h"    // ARRAY_SIZE
   7 #include "files.h"
   8 #include "shm.h"
   9 #include "ioctls.h"
  10
  11 #define IOCTL_GROUPS_MAX 48
  12
  13 static const struct ioctl_group *grps[IOCTL_GROUPS_MAX];
  14 static int grps_cnt;
  15
  16 void register_ioctl_group(const struct ioctl_group *grp)
  17 {
  18         /* group could be empty e.g. if everything is ifdeffed out */
  19         if (grp->ioctls_cnt == 0)
  20                 return;
  21
  22         if (grps_cnt == ARRAY_SIZE(grps)) {
  23                 fprintf(stderr, "WARNING: please grow IOCTL_GROUPS_MAX.\n");
  24                 return;
  25         }
  26
  27         grps[grps_cnt] = grp;
  28
  29         ++grps_cnt;
  30 }
  31
  32 const struct ioctl_group *find_ioctl_group(int fd)
  33 {
  34         const char *devname;
  35         struct stat stbuf;
  36         int i;
  37         size_t j;
  38
  39         if (fstat(fd, &stbuf) < 0)
  40                 return NULL;
  41
  42         if (stbuf.st_rdev == 0)
  43                 return NULL;
  44
  45         devname = map_dev(stbuf.st_rdev, stbuf.st_mode);
  46         if (!devname)
  47                 return NULL;
  48
  49         for (i=0; i < grps_cnt; ++i) {
  50                 if (grps[i]->fd_test) {
  51                         if (grps[i]->fd_test(fd, &stbuf) == 0)
  52                                 return grps[i];
  53                         else
  54                                 continue;
  55                 }
  56
  57                 switch (grps[i]->devtype) {
  58                 case DEV_MISC:
  59                         /* fall through. misc devices are char devices. */
  60                 case DEV_CHAR:
  61                         if (!S_ISCHR(stbuf.st_mode))
  62                                 continue;
  63                         break;
  64                 case DEV_BLOCK:
  65                         if (!S_ISBLK(stbuf.st_mode))
  66                                 continue;
  67                         break;
  68                 default: break;
  69                 }
  70
  71                 for (j=0; j < grps[i]->devs_cnt; ++j)
  72                         if (strcmp(devname, grps[i]->devs[j]) == 0)
  73                                 return grps[i];
  74         }
  75
  76         return NULL;
  77 }
  78
  79 const struct ioctl_group *get_random_ioctl_group(void)
  80 {
  81         if (grps_cnt == 0)
  82                 return NULL;
  83
  84         return grps[rand() % grps_cnt];
  85 }
  86
  87 void pick_random_ioctl(const struct ioctl_group *grp, int childno)
  88 {
  89         int ioctlnr;
  90
  91         ioctlnr = rand() % grp->ioctls_cnt;
  92
  93         shm->a2[childno] = grp->ioctls[ioctlnr].request;
  94 }
  95
  96 void dump_ioctls(void)
  97 {
  98         int i;
  99         size_t j;
 100
 101         for (i=0; i < grps_cnt; ++i) {
 102                 if (grps[i]->name)
 103                         printf("- %s:\n", grps[i]->name);
 104                 else if (grps[i]->devtype) {
 105                         if (grps[i]->devtype == DEV_MISC)
 106                                 printf("- misc devices");
 107                         else if (grps[i]->devtype == DEV_CHAR)
 108                                 printf("- char devices");
 109                         else if (grps[i]->devtype == DEV_BLOCK)
 110                                 printf("- block devices");
 111                         for (j=0; j < grps[i]->devs_cnt; ++j)
 112                                 printf("%s '%s'",
 113                                         j == 0 ? "" : ",",
 114                                         grps[i]->devs[j]);
 115                         printf(":\n");
 116                 } else
 117                         printf("- <unknown>:\n");
 118
 119                 for (j=0; j < grps[i]->ioctls_cnt; ++j) {
 120                         printf("  - 0x%08x : %s\n",
 121                                         grps[i]->ioctls[j].request,
 122                                         grps[i]->ioctls[j].name ? : "");
 123                 }
 124         }
 125 }