gpio/gpio.c

   1 /*
   2  * gpio.c:
   3  *      Swiss-Army-Knife, Set-UID command-line interface to the Raspberry
   4  *      Pi's GPIO.
   5  *      Copyright (c) 2012-2013 Gordon Henderson
   6  ***********************************************************************
   7  * This file is part of wiringPi:
   8  *      https://projects.drogon.net/raspberry-pi/wiringpi/
   9  *
  10  *    wiringPi is free software: you can redistribute it and/or modify
  11  *    it under the terms of the GNU Lesser General Public License as published by
  12  *    the Free Software Foundation, either version 3 of the License, or
  13  *    (at your option) any later version.
  14  *
  15  *    wiringPi is distributed in the hope that it will be useful,
  16  *    but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  *    GNU Lesser General Public License for more details.
  19  *
  20  *    You should have received a copy of the GNU Lesser General Public License
  21  *    along with wiringPi.  If not, see <http://www.gnu.org/licenses/>.
  22  ***********************************************************************
  23  */
  24
  25
  26 #include <stdio.h>
  27 #include <stdlib.h>
  28 #include <stdint.h>
  29 #include <ctype.h>
  30 #include <string.h>
  31 #include <unistd.h>
  32 #include <errno.h>
  33 #include <fcntl.h>
  34 #include <sys/types.h>
  35 #include <sys/stat.h>
  36
  37 #include <wiringPi.h>
  38
  39 #include <gertboard.h>
  40 #include <piFace.h>
  41
  42 #include "extensions.h"
  43
  44 extern int wiringPiDebug ;
  45
  46 // External functions I can't be bothered creating a separate .h file for:
  47
  48 extern void doReadall    (void) ;
  49 extern void doPins       (void) ;
  50
  51 #ifndef TRUE
  52 #  define       TRUE    (1==1)
  53 #  define       FALSE   (1==2)
  54 #endif
  55
  56 #define VERSION                 "2.20"
  57 #define PI_USB_POWER_CONTROL    38
  58 #define I2CDETECT               "/usr/sbin/i2cdetect"
  59
  60 int wpMode ;
  61
  62 char *usage = "Usage: gpio -v\n"
  63               "       gpio -h\n"
  64               "       gpio [-g|-1] [-x extension:params] ...\n"
  65               "       gpio [-p] <read/write/wb> ...\n"
  66               "       gpio <read/write/aread/awritewb/pwm/clock/mode> ...\n"
  67               "       gpio readall/reset\n"
  68               "       gpio unexportall/exports\n"
  69               "       gpio export/edge/unexport ...\n"
  70               "       gpio wfi <pin> <mode>\n"
  71               "       gpio drive <group> <value>\n"
  72               "       gpio pwm-bal/pwm-ms \n"
  73               "       gpio pwmr <range> \n"
  74               "       gpio pwmc <divider> \n"
  75               "       gpio load spi/i2c\n"
  76               "       gpio i2cd/i2cdetect\n"
  77               "       gpio usbp high/low\n"
  78               "       gpio gbr <channel>\n"
  79               "       gpio gbw <channel> <value>" ;     // No trailing newline needed here.
  80
  81
  82 #ifdef  NOT_FOR_NOW
  83 /*
  84  * decodePin:
  85  *      Decode a pin "number" which can actually be a pin name to represent
  86  *      one of the Pi's on-board pins.
  87  *********************************************************************************
  88  */
  89
  90 static int decodePin (const char *str)
  91 {
  92
  93 // The first case - see if it's a number:
  94
  95   if (isdigit (str [0]))
  96     return atoi (str) ;
  97
  98   return 0 ;
  99 }
 100 #endif
 101
 102
 103 /*
 104  * changeOwner:
 105  *      Change the ownership of the file to the real userId of the calling
 106  *      program so we can access it.
 107  *********************************************************************************
 108  */
 109
 110 static void changeOwner (char *cmd, char *file)
 111 {
 112   uid_t uid = getuid () ;
 113   uid_t gid = getgid () ;
 114
 115   if (chown (file, uid, gid) != 0)
 116   {
 117     if (errno == ENOENT)        // Warn that it's not there
 118       fprintf (stderr, "%s: Warning: File not present: %s\n", cmd, file) ;
 119     else
 120     {
 121       fprintf (stderr, "%s: Unable to change ownership of %s: %s\n", cmd, file, strerror (errno)) ;
 122       exit (1) ;
 123     }
 124   }
 125 }
 126
 127
 128 /*
 129  * moduleLoaded:
 130  *      Return true/false if the supplied module is loaded
 131  *********************************************************************************
 132  */
 133
 134 static int moduleLoaded (char *modName)
 135 {
 136   int len   = strlen (modName) ;
 137   int found = FALSE ;
 138   FILE *fd = fopen ("/proc/modules", "r") ;
 139   char line [80] ;
 140
 141   if (fd == NULL)
 142   {
 143     fprintf (stderr, "gpio: Unable to check modules: %s\n", strerror (errno)) ;
 144     exit (1) ;
 145   }
 146
 147   while (fgets (line, 80, fd) != NULL)
 148   {
 149     if (strncmp (line, modName, len) != 0)
 150       continue ;
 151
 152     found = TRUE ;
 153     break ;
 154   }
 155
 156   fclose (fd) ;
 157
 158   return found ;
 159 }
 160
 161
 162 /*
 163  * doLoad:
 164  *      Load either the spi or i2c modules and change device ownerships, etc.
 165  *********************************************************************************
 166  */
 167
 168 static void _doLoadUsage (char *argv [])
 169 {
 170   fprintf (stderr, "Usage: %s load <spi/i2c> [SPI bufferSize in KB | I2C baudrate in Kb/sec]\n", argv [0]) ;
 171   exit (1) ;
 172 }
 173
 174 static void doLoad (int argc, char *argv [])
 175 {
 176   char *module1, *module2 ;
 177   char cmd [80] ;
 178   char *file1, *file2 ;
 179   char args1 [32], args2 [32] ;
 180
 181   if (argc < 3)
 182     _doLoadUsage (argv) ;
 183
 184   args1 [0] = args2 [0] = 0 ;
 185
 186   /**/ if (strcasecmp (argv [2], "spi") == 0)
 187   {
 188     module1 = "spidev" ;
 189     module2 = "spi_bcm2708" ;
 190     file1  = "/dev/spidev0.0" ;
 191     file2  = "/dev/spidev0.1" ;
 192     if (argc == 4)
 193       sprintf (args1, " bufsiz=%d", atoi (argv [3]) * 1024) ;
 194     else if (argc > 4)
 195       _doLoadUsage (argv) ;
 196   }
 197   else if (strcasecmp (argv [2], "i2c") == 0)
 198   {
 199     module1 = "i2c_dev" ;
 200     module2 = "i2c_bcm2708" ;
 201     file1  = "/dev/i2c-0" ;
 202     file2  = "/dev/i2c-1" ;
 203     if (argc == 4)
 204       sprintf (args2, " baudrate=%d", atoi (argv [3]) * 1000) ;
 205     else if (argc > 4)
 206       _doLoadUsage (argv) ;
 207   }
 208   else
 209     _doLoadUsage (argv) ;
 210
 211   if (!moduleLoaded (module1))
 212   {
 213     sprintf (cmd, "modprobe %s%s", module1, args1) ;
 214     system (cmd) ;
 215   }
 216
 217   if (!moduleLoaded (module2))
 218   {
 219     sprintf (cmd, "modprobe %s%s", module2, args2) ;
 220     system (cmd) ;
 221   }
 222
 223   if (!moduleLoaded (module2))
 224   {
 225     fprintf (stderr, "%s: Unable to load %s\n", argv [0], module2) ;
 226     exit (1) ;
 227   }
 228
 229   sleep (1) ;   // To let things get settled
 230
 231   changeOwner (argv [0], file1) ;
 232   changeOwner (argv [0], file2) ;
 233 }
 234
 235
 236 /*
 237  * doI2Cdetect:
 238  *      Run the i2cdetect command with the right runes for this Pi revision
 239  *********************************************************************************
 240  */
 241
 242 static void doI2Cdetect (int argc, char *argv [])
 243 {
 244   int port = piBoardRev () == 1 ? 0 : 1 ;
 245   char command [128] ;
 246   struct stat statBuf ;
 247
 248   if (stat (I2CDETECT, &statBuf) < 0)
 249   {
 250     fprintf (stderr, "%s: Unable to find i2cdetect command: %s\n", argv [0], strerror (errno)) ;
 251     return ;
 252   }
 253
 254   if (!moduleLoaded ("i2c_dev"))
 255   {
 256     fprintf (stderr, "%s: The I2C kernel module(s) are not loaded.\n", argv [0]) ;
 257     return ;
 258   }
 259
 260   sprintf (command, "%s -y %d", I2CDETECT, port) ;
 261   if (system (command) < 0)
 262     fprintf (stderr, "%s: Unable to run i2cdetect: %s\n", argv [0], strerror (errno)) ;
 263
 264 }
 265
 266
 267 /*
 268  * doExports:
 269  *      List all GPIO exports
 270  *********************************************************************************
 271  */
 272
 273 static void doExports (int argc, char *argv [])
 274 {
 275   int fd ;
 276   int i, l, first ;
 277   char fName [128] ;
 278   char buf [16] ;
 279
 280   for (first = 0, i = 0 ; i < 64 ; ++i) // Crude, but effective
 281   {
 282
 283 // Try to read the direction
 284
 285     sprintf (fName, "/sys/class/gpio/gpio%d/direction", i) ;
 286     if ((fd = open (fName, O_RDONLY)) == -1)
 287       continue ;
 288
 289     if (first == 0)
 290     {
 291       ++first ;
 292       printf ("GPIO Pins exported:\n") ;
 293     }
 294
 295     printf ("%4d: ", i) ;
 296
 297     if ((l = read (fd, buf, 16)) == 0)
 298       sprintf (buf, "%s", "?") ;
 299
 300     buf [l] = 0 ;
 301     if ((buf [strlen (buf) - 1]) == '\n')
 302       buf [strlen (buf) - 1] = 0 ;
 303
 304     printf ("%-3s", buf) ;
 305
 306     close (fd) ;
 307
 308 // Try to Read the value
 309
 310     sprintf (fName, "/sys/class/gpio/gpio%d/value", i) ;
 311     if ((fd = open (fName, O_RDONLY)) == -1)
 312     {
 313       printf ("No Value file (huh?)\n") ;
 314       continue ;
 315     }
 316
 317     if ((l = read (fd, buf, 16)) == 0)
 318       sprintf (buf, "%s", "?") ;
 319
 320     buf [l] = 0 ;
 321     if ((buf [strlen (buf) - 1]) == '\n')
 322       buf [strlen (buf) - 1] = 0 ;
 323
 324     printf ("  %s", buf) ;
 325
 326 // Read any edge trigger file
 327
 328     sprintf (fName, "/sys/class/gpio/gpio%d/edge", i) ;
 329     if ((fd = open (fName, O_RDONLY)) == -1)
 330     {
 331       printf ("\n") ;
 332       continue ;
 333     }
 334
 335     if ((l = read (fd, buf, 16)) == 0)
 336       sprintf (buf, "%s", "?") ;
 337
 338     buf [l] = 0 ;
 339     if ((buf [strlen (buf) - 1]) == '\n')
 340       buf [strlen (buf) - 1] = 0 ;
 341
 342     printf ("  %-8s\n", buf) ;
 343
 344     close (fd) ;
 345   }
 346 }
 347
 348
 349 /*
 350  * doExport:
 351  *      gpio export pin mode
 352  *      This uses the /sys/class/gpio device interface.
 353  *********************************************************************************
 354  */
 355
 356 void doExport (int argc, char *argv [])
 357 {
 358   FILE *fd ;
 359   int pin ;
 360   char *mode ;
 361   char fName [128] ;
 362
 363   if (argc != 4)
 364   {
 365     fprintf (stderr, "Usage: %s export pin mode\n", argv [0]) ;
 366     exit (1) ;
 367   }
 368
 369   pin = atoi (argv [2]) ;
 370
 371   mode = argv [3] ;
 372
 373   if ((fd = fopen ("/sys/class/gpio/export", "w")) == NULL)
 374   {
 375     fprintf (stderr, "%s: Unable to open GPIO export interface: %s\n", argv [0], strerror (errno)) ;
 376     exit (1) ;
 377   }
 378
 379   fprintf (fd, "%d\n", pin) ;
 380   fclose (fd) ;
 381
 382   sprintf (fName, "/sys/class/gpio/gpio%d/direction", pin) ;
 383   if ((fd = fopen (fName, "w")) == NULL)
 384   {
 385     fprintf (stderr, "%s: Unable to open GPIO direction interface for pin %d: %s\n", argv [0], pin, strerror (errno)) ;
 386     exit (1) ;
 387   }
 388
 389   /**/ if ((strcasecmp (mode, "in")  == 0) || (strcasecmp (mode, "input")  == 0))
 390     fprintf (fd, "in\n") ;
 391   else if ((strcasecmp (mode, "out") == 0) || (strcasecmp (mode, "output") == 0))
 392     fprintf (fd, "out\n") ;
 393   else
 394   {
 395     fprintf (stderr, "%s: Invalid mode: %s. Should be in or out\n", argv [1], mode) ;
 396     exit (1) ;
 397   }
 398
 399   fclose (fd) ;
 400
 401 // Change ownership so the current user can actually use it!
 402
 403   sprintf (fName, "/sys/class/gpio/gpio%d/value", pin) ;
 404   changeOwner (argv [0], fName) ;
 405
 406   sprintf (fName, "/sys/class/gpio/gpio%d/edge", pin) ;
 407   changeOwner (argv [0], fName) ;
 408
 409 }
 410
 411
 412 /*
 413  * doWfi:
 414  *      gpio wfi pin mode
 415  *      Wait for Interrupt on a given pin.
 416  *      Slight cheat here - it's easier to actually use ISR now (which calls
 417  *      gpio to set the pin modes!) then we simply sleep, and expect the thread
 418  *      to exit the program. Crude but effective.
 419  *********************************************************************************
 420  */
 421
 422 static void wfi (void)
 423   { exit (0) ; }
 424
 425 void doWfi (int argc, char *argv [])
 426 {
 427   int pin, mode ;
 428
 429   if (argc != 4)
 430   {
 431     fprintf (stderr, "Usage: %s wfi pin mode\n", argv [0]) ;
 432     exit (1) ;
 433   }
 434
 435   pin  = atoi (argv [2]) ;
 436
 437   /**/ if (strcasecmp (argv [3], "rising")  == 0) mode = INT_EDGE_RISING ;
 438   else if (strcasecmp (argv [3], "falling") == 0) mode = INT_EDGE_FALLING ;
 439   else if (strcasecmp (argv [3], "both")    == 0) mode = INT_EDGE_BOTH ;
 440   else
 441   {
 442     fprintf (stderr, "%s: wfi: Invalid mode: %s. Should be rising, falling or both\n", argv [1], argv [3]) ;
 443     exit (1) ;
 444   }
 445
 446   if (wiringPiISR (pin, mode, &wfi) < 0)
 447   {
 448     fprintf (stderr, "%s: wfi: Unable to setup ISR: %s\n", argv [1], strerror (errno)) ;
 449     exit (1) ;
 450   }
 451
 452   for (;;)
 453     delay (9999) ;
 454 }
 455
 456
 457
 458 /*
 459  * doEdge:
 460  *      gpio edge pin mode
 461  *      Easy access to changing the edge trigger on a GPIO pin
 462  *      This uses the /sys/class/gpio device interface.
 463  *********************************************************************************
 464  */
 465
 466 void doEdge (int argc, char *argv [])
 467 {
 468   FILE *fd ;
 469   int pin ;
 470   char *mode ;
 471   char fName [128] ;
 472
 473   if (argc != 4)
 474   {
 475     fprintf (stderr, "Usage: %s edge pin mode\n", argv [0]) ;
 476     exit (1) ;
 477   }
 478
 479   pin  = atoi (argv [2]) ;
 480   mode = argv [3] ;
 481
 482 // Export the pin and set direction to input
 483
 484   if ((fd = fopen ("/sys/class/gpio/export", "w")) == NULL)
 485   {
 486     fprintf (stderr, "%s: Unable to open GPIO export interface: %s\n", argv [0], strerror (errno)) ;
 487     exit (1) ;
 488   }
 489
 490   fprintf (fd, "%d\n", pin) ;
 491   fclose (fd) ;
 492
 493   sprintf (fName, "/sys/class/gpio/gpio%d/direction", pin) ;
 494   if ((fd = fopen (fName, "w")) == NULL)
 495   {
 496     fprintf (stderr, "%s: Unable to open GPIO direction interface for pin %d: %s\n", argv [0], pin, strerror (errno)) ;
 497     exit (1) ;
 498   }
 499
 500   fprintf (fd, "in\n") ;
 501   fclose (fd) ;
 502
 503   sprintf (fName, "/sys/class/gpio/gpio%d/edge", pin) ;
 504   if ((fd = fopen (fName, "w")) == NULL)
 505   {
 506     fprintf (stderr, "%s: Unable to open GPIO edge interface for pin %d: %s\n", argv [0], pin, strerror (errno)) ;
 507     exit (1) ;
 508   }
 509
 510   /**/ if (strcasecmp (mode, "none")    == 0) fprintf (fd, "none\n") ;
 511   else if (strcasecmp (mode, "rising")  == 0) fprintf (fd, "rising\n") ;
 512   else if (strcasecmp (mode, "falling") == 0) fprintf (fd, "falling\n") ;
 513   else if (strcasecmp (mode, "both")    == 0) fprintf (fd, "both\n") ;
 514   else
 515   {
 516     fprintf (stderr, "%s: Invalid mode: %s. Should be none, rising, falling or both\n", argv [1], mode) ;
 517     exit (1) ;
 518   }
 519
 520 // Change ownership of the value and edge files, so the current user can actually use it!
 521
 522   sprintf (fName, "/sys/class/gpio/gpio%d/value", pin) ;
 523   changeOwner (argv [0], fName) ;
 524
 525   sprintf (fName, "/sys/class/gpio/gpio%d/edge", pin) ;
 526   changeOwner (argv [0], fName) ;
 527
 528   fclose (fd) ;
 529 }
 530
 531
 532 /*
 533  * doUnexport:
 534  *      gpio unexport pin
 535  *      This uses the /sys/class/gpio device interface.
 536  *********************************************************************************
 537  */
 538
 539 void doUnexport (int argc, char *argv [])
 540 {
 541   FILE *fd ;
 542   int pin ;
 543
 544   if (argc != 3)
 545   {
 546     fprintf (stderr, "Usage: %s unexport pin\n", argv [0]) ;
 547     exit (1) ;
 548   }
 549
 550   pin = atoi (argv [2]) ;
 551
 552   if ((fd = fopen ("/sys/class/gpio/unexport", "w")) == NULL)
 553   {
 554     fprintf (stderr, "%s: Unable to open GPIO export interface\n", argv [0]) ;
 555     exit (1) ;
 556   }
 557
 558   fprintf (fd, "%d\n", pin) ;
 559   fclose (fd) ;
 560 }
 561
 562
 563 /*
 564  * doUnexportAll:
 565  *      gpio unexportall
 566  *      Un-Export all the GPIO pins.
 567  *      This uses the /sys/class/gpio device interface.
 568  *********************************************************************************
 569  */
 570
 571 void doUnexportall (char *progName)
 572 {
 573   FILE *fd ;
 574   int pin ;
 575
 576   for (pin = 0 ; pin < 63 ; ++pin)
 577   {
 578     if ((fd = fopen ("/sys/class/gpio/unexport", "w")) == NULL)
 579     {
 580       fprintf (stderr, "%s: Unable to open GPIO export interface\n", progName) ;
 581       exit (1) ;
 582     }
 583     fprintf (fd, "%d\n", pin) ;
 584     fclose (fd) ;
 585   }
 586 }
 587
 588
 589 /*
 590  * doResetExternal:
 591  *      Load readallExternal, we try to do this with an external device.
 592  *********************************************************************************
 593  */
 594
 595 static void doResetExternal (void)
 596 {
 597   int pin ;
 598
 599   for (pin = wiringPiNodes->pinBase ; pin <= wiringPiNodes->pinMax ; ++pin)
 600   {
 601     pinMode         (pin, INPUT) ;
 602     pullUpDnControl (pin, PUD_OFF) ;
 603   }
 604 }
 605
 606
 607 /*
 608  * doReset:
 609  *      Reset the GPIO pins - as much as we can do
 610  *********************************************************************************
 611  */
 612
 613 static void doReset (char *progName)
 614 {
 615   int pin ;
 616
 617   if (wiringPiNodes != NULL)    // External reset
 618     doResetExternal () ;
 619   else
 620   {
 621     doUnexportall (progName) ;
 622
 623     for (pin = 0 ; pin < 64 ; ++pin)
 624     {
 625       if (wpiPinToGpio (pin) == -1)
 626         continue ;
 627
 628       digitalWrite    (pin, LOW) ;
 629       pinMode         (pin, INPUT) ;
 630       pullUpDnControl (pin, PUD_OFF) ;
 631     }
 632   }
 633 }
 634
 635
 636 /*
 637  * doMode:
 638  *      gpio mode pin mode ...
 639  *********************************************************************************
 640  */
 641
 642 void doMode (int argc, char *argv [])
 643 {
 644   int pin ;
 645   char *mode ;
 646
 647   if (argc != 4)
 648   {
 649     fprintf (stderr, "Usage: %s mode pin mode\n", argv [0]) ;
 650     exit (1) ;
 651   }
 652
 653   pin = atoi (argv [2]) ;
 654
 655   mode = argv [3] ;
 656
 657   /**/ if (strcasecmp (mode, "in")      == 0) pinMode         (pin, INPUT) ;
 658   else if (strcasecmp (mode, "input")   == 0) pinMode         (pin, INPUT) ;
 659   else if (strcasecmp (mode, "out")     == 0) pinMode         (pin, OUTPUT) ;
 660   else if (strcasecmp (mode, "output")  == 0) pinMode         (pin, OUTPUT) ;
 661   else if (strcasecmp (mode, "pwm")     == 0) pinMode         (pin, PWM_OUTPUT) ;
 662   else if (strcasecmp (mode, "pwmTone") == 0) pinMode         (pin, PWM_TONE_OUTPUT) ;
 663   else if (strcasecmp (mode, "clock")   == 0) pinMode         (pin, GPIO_CLOCK) ;
 664   else if (strcasecmp (mode, "up")      == 0) pullUpDnControl (pin, PUD_UP) ;
 665   else if (strcasecmp (mode, "down")    == 0) pullUpDnControl (pin, PUD_DOWN) ;
 666   else if (strcasecmp (mode, "tri")     == 0) pullUpDnControl (pin, PUD_OFF) ;
 667   else if (strcasecmp (mode, "off")     == 0) pullUpDnControl (pin, PUD_OFF) ;
 668
 669 // Undocumented
 670
 671   else if (strcasecmp (mode, "alt0")    == 0) pinModeAlt (pin, 0b100) ;
 672   else if (strcasecmp (mode, "alt1")    == 0) pinModeAlt (pin, 0b101) ;
 673   else if (strcasecmp (mode, "alt2")    == 0) pinModeAlt (pin, 0b110) ;
 674   else if (strcasecmp (mode, "alt3")    == 0) pinModeAlt (pin, 0b111) ;
 675   else if (strcasecmp (mode, "alt4")    == 0) pinModeAlt (pin, 0b011) ;
 676   else if (strcasecmp (mode, "alt5")    == 0) pinModeAlt (pin, 0b010) ;
 677   else
 678   {
 679     fprintf (stderr, "%s: Invalid mode: %s. Should be in/out/pwm/clock/up/down/tri\n", argv [1], mode) ;
 680     exit (1) ;
 681   }
 682 }
 683
 684
 685 /*
 686  * doPadDrive:
 687  *      gpio drive group value
 688  *********************************************************************************
 689  */
 690
 691 static void doPadDrive (int argc, char *argv [])
 692 {
 693   int group, val ;
 694
 695   if (argc != 4)
 696   {
 697     fprintf (stderr, "Usage: %s drive group value\n", argv [0]) ;
 698     exit (1) ;
 699   }
 700
 701   group = atoi (argv [2]) ;
 702   val   = atoi (argv [3]) ;
 703
 704   if ((group < 0) || (group > 2))
 705   {
 706     fprintf (stderr, "%s: drive group not 0, 1 or 2: %d\n", argv [0], group) ;
 707     exit (1) ;
 708   }
 709
 710   if ((val < 0) || (val > 7))
 711   {
 712     fprintf (stderr, "%s: drive value not 0-7: %d\n", argv [0], val) ;
 713     exit (1) ;
 714   }
 715
 716   setPadDrive (group, val) ;
 717 }
 718
 719
 720 /*
 721  * doUsbP:
 722  *      Control USB Power - High (1.2A) or Low (600mA)
 723  *      gpio usbp high/low
 724  *********************************************************************************
 725  */
 726
 727 static void doUsbP (int argc, char *argv [])
 728 {
 729   int model, rev, mem, maker, overVolted ;
 730
 731   if (argc != 3)
 732   {
 733     fprintf (stderr, "Usage: %s usbp high|low\n", argv [0]) ;
 734     exit (1) ;
 735   }
 736
 737 // Make sure we're on a B+
 738
 739   piBoardId (&model, &rev, &mem, &maker, &overVolted) ;
 740
 741   if (model != PI_MODEL_BP)
 742   {
 743     fprintf (stderr, "USB power contol is applicable to B+ boards only.\n") ;
 744     exit (1) ;
 745   }
 746
 747 // Need to force BCM_GPIO mode:
 748
 749   wiringPiSetupGpio () ;
 750
 751   if ((strcasecmp (argv [2], "high") == 0) || (strcasecmp (argv [2], "hi") == 0))
 752   {
 753     digitalWrite (PI_USB_POWER_CONTROL, 1) ;
 754     pinMode (PI_USB_POWER_CONTROL, OUTPUT) ;
 755     printf ("Switched to HIGH current USB (1.2A)\n") ;
 756     return ;
 757   }
 758
 759   if ((strcasecmp (argv [2], "low") == 0) || (strcasecmp (argv [2], "lo") == 0))
 760   {
 761     digitalWrite (PI_USB_POWER_CONTROL, 0) ;
 762     pinMode (PI_USB_POWER_CONTROL, OUTPUT) ;
 763     printf ("Switched to LOW current USB (600mA)\n") ;
 764     return ;
 765   }
 766
 767   fprintf (stderr, "Usage: %s usbp high|low\n", argv [0]) ;
 768   exit (1) ;
 769 }
 770
 771
 772 /*
 773  * doGbw:
 774  *      gpio gbw channel value
 775  *      Gertboard Write - To the Analog output
 776  *********************************************************************************
 777  */
 778
 779 static void doGbw (int argc, char *argv [])
 780 {
 781   int channel, value ;
 782
 783   if (argc != 4)
 784   {
 785     fprintf (stderr, "Usage: %s gbw <channel> <value>\n", argv [0]) ;
 786     exit (1) ;
 787   }
 788
 789   channel = atoi (argv [2]) ;
 790   value   = atoi (argv [3]) ;
 791
 792   if ((channel < 0) || (channel > 1))
 793   {
 794     fprintf (stderr, "%s: gbw: Channel number must be 0 or 1\n", argv [0]) ;
 795     exit (1) ;
 796   }
 797
 798   if ((value < 0) || (value > 255))
 799   {
 800     fprintf (stderr, "%s: gbw: Value must be from 0 to 255\n", argv [0]) ;
 801     exit (1) ;
 802   }
 803
 804   if (gertboardAnalogSetup (64) < 0)
 805   {
 806     fprintf (stderr, "Unable to initialise the Gertboard SPI interface: %s\n", strerror (errno)) ;
 807     exit (1) ;
 808   }
 809
 810   analogWrite (64 + channel, value) ;
 811 }
 812
 813
 814 /*
 815  * doGbr:
 816  *      gpio gbr channel
 817  *      From the analog input
 818  *********************************************************************************
 819  */
 820
 821 static void doGbr (int argc, char *argv [])
 822 {
 823   int channel ;
 824
 825   if (argc != 3)
 826   {
 827     fprintf (stderr, "Usage: %s gbr <channel>\n", argv [0]) ;
 828     exit (1) ;
 829   }
 830
 831   channel = atoi (argv [2]) ;
 832
 833   if ((channel < 0) || (channel > 1))
 834   {
 835     fprintf (stderr, "%s: gbr: Channel number must be 0 or 1\n", argv [0]) ;
 836     exit (1) ;
 837   }
 838
 839   if (gertboardAnalogSetup (64) < 0)
 840   {
 841     fprintf (stderr, "Unable to initialise the Gertboard SPI interface: %s\n", strerror (errno)) ;
 842     exit (1) ;
 843   }
 844
 845   printf ("%d\n", analogRead (64 + channel)) ;
 846 }
 847
 848
 849 /*
 850  * doWrite:
 851  *      gpio write pin value
 852  *********************************************************************************
 853  */
 854
 855 static void doWrite (int argc, char *argv [])
 856 {
 857   int pin, val ;
 858
 859   if (argc != 4)
 860   {
 861     fprintf (stderr, "Usage: %s write pin value\n", argv [0]) ;
 862     exit (1) ;
 863   }
 864
 865   pin = atoi (argv [2]) ;
 866
 867   /**/ if ((strcasecmp (argv [3], "up") == 0) || (strcasecmp (argv [3], "on") == 0))
 868     val = 1 ;
 869   else if ((strcasecmp (argv [3], "down") == 0) || (strcasecmp (argv [3], "off") == 0))
 870     val = 0 ;
 871   else
 872     val = atoi (argv [3]) ;
 873
 874   /**/ if (val == 0)
 875     digitalWrite (pin, LOW) ;
 876   else
 877     digitalWrite (pin, HIGH) ;
 878 }
 879
 880
 881 /*
 882  * doAwriterite:
 883  *      gpio awrite pin value
 884  *********************************************************************************
 885  */
 886
 887 static void doAwrite (int argc, char *argv [])
 888 {
 889   int pin, val ;
 890
 891   if (argc != 4)
 892   {
 893     fprintf (stderr, "Usage: %s awrite pin value\n", argv [0]) ;
 894     exit (1) ;
 895   }
 896
 897   pin = atoi (argv [2]) ;
 898
 899   val = atoi (argv [3]) ;
 900
 901   analogWrite (pin, val) ;
 902 }
 903
 904
 905 /*
 906  * doWriteByte:
 907  *      gpio write value
 908  *********************************************************************************
 909  */
 910
 911 static void doWriteByte (int argc, char *argv [])
 912 {
 913   int val ;
 914
 915   if (argc != 3)
 916   {
 917     fprintf (stderr, "Usage: %s wb value\n", argv [0]) ;
 918     exit (1) ;
 919   }
 920
 921   val = (int)strtol (argv [2], NULL, 0) ;
 922
 923   digitalWriteByte (val) ;
 924 }
 925
 926
 927 /*
 928  * doRead:
 929  *      Read a pin and return the value
 930  *********************************************************************************
 931  */
 932
 933 void doRead (int argc, char *argv [])
 934 {
 935   int pin, val ;
 936
 937   if (argc != 3)
 938   {
 939     fprintf (stderr, "Usage: %s read pin\n", argv [0]) ;
 940     exit (1) ;
 941   }
 942
 943   pin = atoi (argv [2]) ;
 944   val = digitalRead (pin) ;
 945
 946   printf ("%s\n", val == 0 ? "0" : "1") ;
 947 }
 948
 949
 950 /*
 951  * doAread:
 952  *      Read an analog pin and return the value
 953  *********************************************************************************
 954  */
 955
 956 void doAread (int argc, char *argv [])
 957 {
 958   if (argc != 3)
 959   {
 960     fprintf (stderr, "Usage: %s aread pin\n", argv [0]) ;
 961     exit (1) ;
 962   }
 963
 964   printf ("%d\n", analogRead (atoi (argv [2]))) ;
 965 }
 966
 967
 968 /*
 969  * doToggle:
 970  *      Toggle an IO pin
 971  *********************************************************************************
 972  */
 973
 974 void doToggle (int argc, char *argv [])
 975 {
 976   int pin ;
 977
 978   if (argc != 3)
 979   {
 980     fprintf (stderr, "Usage: %s toggle pin\n", argv [0]) ;
 981     exit (1) ;
 982   }
 983
 984   pin = atoi (argv [2]) ;
 985
 986   digitalWrite (pin, !digitalRead (pin)) ;
 987 }
 988
 989
 990 /*
 991  * doPwmTone:
 992  *      Output a tone in a PWM pin
 993  *********************************************************************************
 994  */
 995
 996 void doPwmTone (int argc, char *argv [])
 997 {
 998   int pin, freq ;
 999
1000   if (argc != 4)
1001   {
1002     fprintf (stderr, "Usage: %s pwmTone <pin> <freq>\n", argv [0]) ;
1003     exit (1) ;
1004   }
1005
1006   pin = atoi (argv [2]) ;
1007   freq = atoi (argv [3]) ;
1008
1009   pwmToneWrite (pin, freq) ;
1010 }
1011
1012
1013 /*
1014  * doClock:
1015  *      Output a clock on a pin
1016  *********************************************************************************
1017  */
1018
1019 void doClock (int argc, char *argv [])
1020 {
1021   int pin, freq ;
1022
1023   if (argc != 4)
1024   {
1025     fprintf (stderr, "Usage: %s clock <pin> <freq>\n", argv [0]) ;
1026     exit (1) ;
1027   }
1028
1029   pin = atoi (argv [2]) ;
1030
1031   freq = atoi (argv [3]) ;
1032
1033   gpioClockSet (pin, freq) ;
1034 }
1035
1036
1037 /*
1038  * doPwm:
1039  *      Output a PWM value on a pin
1040  *********************************************************************************
1041  */
1042
1043 void doPwm (int argc, char *argv [])
1044 {
1045   int pin, val ;
1046
1047   if (argc != 4)
1048   {
1049     fprintf (stderr, "Usage: %s pwm <pin> <value>\n", argv [0]) ;
1050     exit (1) ;
1051   }
1052
1053   pin = atoi (argv [2]) ;
1054
1055   val = atoi (argv [3]) ;
1056
1057   pwmWrite (pin, val) ;
1058 }
1059
1060
1061 /*
1062  * doPwmMode: doPwmRange: doPwmClock:
1063  *      Change the PWM mode, range and clock divider values
1064  *********************************************************************************
1065  */
1066
1067 static void doPwmMode (int mode)
1068 {
1069   pwmSetMode (mode) ;
1070 }
1071
1072 static void doPwmRange (int argc, char *argv [])
1073 {
1074   unsigned int range ;
1075
1076   if (argc != 3)
1077   {
1078     fprintf (stderr, "Usage: %s pwmr <range>\n", argv [0]) ;
1079     exit (1) ;
1080   }
1081
1082   range = (unsigned int)strtoul (argv [2], NULL, 10) ;
1083
1084   if (range == 0)
1085   {
1086     fprintf (stderr, "%s: range must be > 0\n", argv [0]) ;
1087     exit (1) ;
1088   }
1089
1090   pwmSetRange (range) ;
1091 }
1092
1093 static void doPwmClock (int argc, char *argv [])
1094 {
1095   unsigned int clock ;
1096
1097   if (argc != 3)
1098   {
1099     fprintf (stderr, "Usage: %s pwmc <clock>\n", argv [0]) ;
1100     exit (1) ;
1101   }
1102
1103   clock = (unsigned int)strtoul (argv [2], NULL, 10) ;
1104
1105   if ((clock < 1) || (clock > 4095))
1106   {
1107     fprintf (stderr, "%s: clock must be between 0 and 4096\n", argv [0]) ;
1108     exit (1) ;
1109   }
1110
1111   pwmSetClock (clock) ;
1112 }
1113
1114
1115 /*
1116  * main:
1117  *      Start here
1118  *********************************************************************************
1119  */
1120
1121 int main (int argc, char *argv [])
1122 {
1123   int i ;
1124   int model, rev, mem, maker, overVolted ;
1125
1126   if (getenv ("WIRINGPI_DEBUG") != NULL)
1127   {
1128     printf ("gpio: wiringPi debug mode enabled\n") ;
1129     wiringPiDebug = TRUE ;
1130   }
1131
1132   if (argc == 1)
1133   {
1134     fprintf (stderr, "%s\n", usage) ;
1135     return 1 ;
1136   }
1137
1138 // Help
1139
1140   if (strcasecmp (argv [1], "-h") == 0)
1141   {
1142     printf ("%s: %s\n", argv [0], usage) ;
1143     return 0 ;
1144   }
1145
1146 // Sort of a special:
1147
1148   if (strcmp (argv [1], "-R") == 0)
1149   {
1150     printf ("%d\n", piBoardRev ()) ;
1151     return 0 ;
1152   }
1153
1154 // Version & Warranty
1155
1156   if (strcmp (argv [1], "-V") == 0)
1157   {
1158     printf ("%d\n", piBoardRev ()) ;
1159     return 0 ;
1160   }
1161
1162   if (strcmp (argv [1], "-v") == 0)
1163   {
1164     printf ("gpio version: %s\n", VERSION) ;
1165     printf ("Copyright (c) 2012-2014 Gordon Henderson\n") ;
1166     printf ("This is free software with ABSOLUTELY NO WARRANTY.\n") ;
1167     printf ("For details type: %s -warranty\n", argv [0]) ;
1168     printf ("\n") ;
1169     piBoardId (&model, &rev, &mem, &maker, &overVolted) ;
1170     if (model == PI_MODEL_UNKNOWN)
1171     {
1172       printf ("Your Raspberry Pi has an unknown model type. Please report this to\n") ;
1173       printf ("    projects@drogon.net\n") ;
1174       printf ("with a copy of your /proc/cpuinfo if possible\n") ;
1175     }
1176     else
1177     {
1178       printf ("Raspberry Pi Details:\n") ;
1179       printf ("  Type: %s, Revision: %s, Memory: %dMB, Maker: %s %s\n",
1180           piModelNames [model], piRevisionNames [rev], mem, piMakerNames [maker], overVolted ? "[OV]" : "") ;
1181     }
1182     return 0 ;
1183   }
1184
1185   if (strcasecmp (argv [1], "-warranty") == 0)
1186   {
1187     printf ("gpio version: %s\n", VERSION) ;
1188     printf ("Copyright (c) 2012-2014 Gordon Henderson\n") ;
1189     printf ("\n") ;
1190     printf ("    This program is free software; you can redistribute it and/or modify\n") ;
1191     printf ("    it under the terms of the GNU Leser General Public License as published\n") ;
1192     printf ("    by the Free Software Foundation, either version 3 of the License, or\n") ;
1193     printf ("    (at your option) any later version.\n") ;
1194     printf ("\n") ;
1195     printf ("    This program is distributed in the hope that it will be useful,\n") ;
1196     printf ("    but WITHOUT ANY WARRANTY; without even the implied warranty of\n") ;
1197     printf ("    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\n") ;
1198     printf ("    GNU Lesser General Public License for more details.\n") ;
1199     printf ("\n") ;
1200     printf ("    You should have received a copy of the GNU Lesser General Public License\n") ;
1201     printf ("    along with this program. If not, see <http://www.gnu.org/licenses/>.\n") ;
1202     printf ("\n") ;
1203     return 0 ;
1204   }
1205
1206   if (geteuid () != 0)
1207   {
1208     fprintf (stderr, "%s: Must be root to run. Program should be suid root. This is an error.\n", argv [0]) ;
1209     return 1 ;
1210   }
1211
1212 // Initial test for /sys/class/gpio operations:
1213
1214   /**/ if (strcasecmp (argv [1], "exports"    ) == 0)   { doExports     (argc, argv) ;  return 0 ; }
1215   else if (strcasecmp (argv [1], "export"     ) == 0)   { doExport      (argc, argv) ;  return 0 ; }
1216   else if (strcasecmp (argv [1], "edge"       ) == 0)   { doEdge        (argc, argv) ;  return 0 ; }
1217   else if (strcasecmp (argv [1], "unexport"   ) == 0)   { doUnexport    (argc, argv) ;  return 0 ; }
1218   else if (strcasecmp (argv [1], "unexportall") == 0)   { doUnexportall (argv [0]) ;    return 0 ; }
1219
1220 // Check for load command:
1221
1222   if (strcasecmp (argv [1], "load" ) == 0)      { doLoad     (argc, argv) ; return 0 ; }
1223
1224 // Gertboard commands
1225
1226   if (strcasecmp (argv [1], "gbr" ) == 0)       { doGbr (argc, argv) ; return 0 ; }
1227   if (strcasecmp (argv [1], "gbw" ) == 0)       { doGbw (argc, argv) ; return 0 ; }
1228
1229 // Check for -g argument
1230
1231   /**/ if (strcasecmp (argv [1], "-g") == 0)
1232   {
1233     wiringPiSetupGpio () ;
1234
1235     for (i = 2 ; i < argc ; ++i)
1236       argv [i - 1] = argv [i] ;
1237     --argc ;
1238     wpMode = WPI_MODE_GPIO ;
1239   }
1240
1241 // Check for -1 argument
1242
1243   else if (strcasecmp (argv [1], "-1") == 0)
1244   {
1245     wiringPiSetupPhys () ;
1246
1247     for (i = 2 ; i < argc ; ++i)
1248       argv [i - 1] = argv [i] ;
1249     --argc ;
1250     wpMode = WPI_MODE_PHYS ;
1251   }
1252
1253 // Check for -p argument for PiFace
1254
1255   else if (strcasecmp (argv [1], "-p") == 0)
1256   {
1257     piFaceSetup (200) ;
1258
1259     for (i = 2 ; i < argc ; ++i)
1260       argv [i - 1] = argv [i] ;
1261     --argc ;
1262     wpMode = WPI_MODE_PIFACE ;
1263   }
1264
1265 // Default to wiringPi mode
1266
1267   else
1268   {
1269     wiringPiSetup () ;
1270     wpMode = WPI_MODE_PINS ;
1271   }
1272
1273 // Check for -x argument to load in a new extension
1274
1275   if (strcasecmp (argv [1], "-x") == 0)
1276   {
1277     if (argc < 3)
1278     {
1279       fprintf (stderr, "%s: -x missing extension specification.\n", argv [0]) ;
1280       exit (EXIT_FAILURE) ;
1281     }
1282
1283     if (!doExtension (argv [0], argv [2]))      // Prints its own error messages
1284       exit (EXIT_FAILURE) ;
1285
1286     for (i = 3 ; i < argc ; ++i)
1287       argv [i - 2] = argv [i] ;
1288     argc -= 2 ;
1289   }
1290
1291   if (argc <= 1)
1292   {
1293     fprintf (stderr, "%s: no command given\n", argv [0]) ;
1294     exit (EXIT_FAILURE) ;
1295   }
1296
1297 // Core wiringPi functions
1298
1299   /**/ if (strcasecmp (argv [1], "mode"   ) == 0) doMode      (argc, argv) ;
1300   else if (strcasecmp (argv [1], "read"   ) == 0) doRead      (argc, argv) ;
1301   else if (strcasecmp (argv [1], "write"  ) == 0) doWrite     (argc, argv) ;
1302   else if (strcasecmp (argv [1], "pwm"    ) == 0) doPwm       (argc, argv) ;
1303   else if (strcasecmp (argv [1], "awrite" ) == 0) doAwrite    (argc, argv) ;
1304   else if (strcasecmp (argv [1], "aread"  ) == 0) doAread     (argc, argv) ;
1305
1306 // GPIO Nicies
1307
1308   else if (strcasecmp (argv [1], "toggle" ) == 0) doToggle    (argc, argv) ;
1309
1310 // Pi Specifics
1311
1312   else if (strcasecmp (argv [1], "pwm-bal"  ) == 0) doPwmMode    (PWM_MODE_BAL) ;
1313   else if (strcasecmp (argv [1], "pwm-ms"   ) == 0) doPwmMode    (PWM_MODE_MS) ;
1314   else if (strcasecmp (argv [1], "pwmr"     ) == 0) doPwmRange   (argc, argv) ;
1315   else if (strcasecmp (argv [1], "pwmc"     ) == 0) doPwmClock   (argc, argv) ;
1316   else if (strcasecmp (argv [1], "pwmTone"  ) == 0) doPwmTone    (argc, argv) ;
1317   else if (strcasecmp (argv [1], "drive"    ) == 0) doPadDrive   (argc, argv) ;
1318   else if (strcasecmp (argv [1], "usbp"     ) == 0) doUsbP       (argc, argv) ;
1319   else if (strcasecmp (argv [1], "readall"  ) == 0) doReadall    () ;
1320   else if (strcasecmp (argv [1], "nreadall" ) == 0) doReadall    () ;
1321   else if (strcasecmp (argv [1], "pins"     ) == 0) doPins       () ;
1322   else if (strcasecmp (argv [1], "i2cdetect") == 0) doI2Cdetect  (argc, argv) ;
1323   else if (strcasecmp (argv [1], "i2cd"     ) == 0) doI2Cdetect  (argc, argv) ;
1324   else if (strcasecmp (argv [1], "reset"    ) == 0) doReset      (argv [0]) ;
1325   else if (strcasecmp (argv [1], "wb"       ) == 0) doWriteByte  (argc, argv) ;
1326   else if (strcasecmp (argv [1], "clock"    ) == 0) doClock      (argc, argv) ;
1327   else if (strcasecmp (argv [1], "wfi"      ) == 0) doWfi        (argc, argv) ;
1328   else
1329   {
1330     fprintf (stderr, "%s: Unknown command: %s.\n", argv [0], argv [1]) ;
1331     exit (EXIT_FAILURE) ;
1332   }
1333   return 0 ;
1334 }