gpio.c

   1 #include <unistd.h>
   2 #include <stdlib.h>
   3 #include <stdio.h>
   4 #include <string.h>
   5 #include <fcntl.h>
   6 #include <errno.h>
   7 #include <err.h>
   8
   9 static const int bmap[] = {
  10     -1,
  11     -1, -1,  2, -1,  3, -1,  4, 14, -1, 15, 17, 18, 27, -1, 22, 23, -1, 24, 10, -1,
  12      9, 25, 11,  8, -1,  7, -1, -1,  5, -1,  6, 12, 13, -1, 19, 16, 21, 20, -1, 21,
  13 };
  14 static const int imap[] = {
  15     -1, -1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13,
  16     14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, -1, 27,
  17 };
  18 static const int *umap = bmap;
  19 static int mapn = sizeof(bmap) / sizeof(*bmap);
  20
  21 static int mappin(int num)
  22 {
  23     if((num < 0) || (num >= mapn))
  24         return(-1);
  25     return(umap[num]);
  26 }
  27
  28 static void setport(int p, int v)
  29 {
  30     char path[256], line[256];
  31     FILE *fp;
  32     int rv, rp;
  33
  34     if((rp = mappin(p)) < 0) {
  35         fprintf(stderr, "gpio: %i: no such port\n", p);
  36         exit(1);
  37     }
  38     sprintf(path, "/sys/class/gpio/gpio%i/direction", rp);
  39     if((fp = fopen(path, "r")) == NULL)
  40         err(1, "%s", path);
  41     rv = !!fgets(line, sizeof(line), fp);
  42     fclose(fp);
  43     if(!rv || strcmp(line, "out\n")) {
  44         if((fp = fopen(path, "w")) == NULL)
  45             err(1, "%s", path);
  46         fprintf(fp, "out\n"); fflush(fp);
  47         if(ferror(fp))
  48             errx(1, "gpio%i: could not set to output", rp);
  49         fclose(fp);
  50     }
  51     sprintf(path, "/sys/class/gpio/gpio%i/value", rp);
  52     if((fp = fopen(path, "w")) == NULL)
  53         err(1, "%s", path);
  54     fprintf(fp, "%i\n", v); fflush(fp);
  55     if(ferror(fp))
  56         errx(1, "gpio%i: could not set value", rp);
  57     fclose(fp);
  58 }
  59
  60 static void usage(FILE *out)
  61 {
  62     fprintf(out, "usage: gpio [-hi] PORT=VAL...\n");
  63 }
  64
  65 int main(int argc, char **argv)
  66 {
  67     int c, i, port, val;
  68     char *p, *e;
  69
  70     while((c = getopt(argc, argv, "hi")) >= 0) {
  71         switch(c) {
  72         case 'h':
  73             usage(stdout);
  74             return(0);
  75         case 'i':
  76             umap = imap;
  77             mapn = sizeof(imap) / sizeof(*imap);
  78             break;
  79         default:
  80             usage(stderr);
  81             return(1);
  82         }
  83     }
  84     if(optind >= argc) {
  85         usage(stderr);
  86         return(1);
  87     }
  88     for(i = optind; i < argc; i++) {
  89         if((p = strchr(argv[i], '=')) == NULL) {
  90             fprintf(stderr, "gpio: %s: not of the form PORT=VAL\n", argv[i]);
  91             exit(1);
  92         }
  93         port = strtol(argv[i], &e, 10);
  94         if(e != p) {
  95             fprintf(stderr, "gpio: %s: not of the form PORT=VAL\n", argv[i]);
  96             exit(1);
  97         }
  98         val = strtol(p + 1, &e, 10);
  99         if(*e) {
 100             fprintf(stderr, "gpio: %s: not of the form PORT=VAL\n", argv[i]);
 101             exit(1);
 102         }
 103         setport(port, val);
 104     }
 105     return(0);
 106 }
 107
 108 /*
 109  * Local Variables:
 110  * compile-command: "gcc -Wall -g -O2 -march=native -o gpio gpio.c"
 111  * End:
 112  */