Gordons Projects

--> Projects Top-Level GIT

Updated versions for new maker - mbest
[wiringPi] / 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-2014 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.22"
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 model, rev, mem, maker, overVolted ;
616   int pin, endPin ;
617
618   printf ("GPIO Reset is dangerous!\n") ;
619   printf (" - Do Not rely on this to do anything sensible!\n") ;
620
621   if (wiringPiNodes != NULL)    // External
622   {
623     doResetExternal () ;
624     return ;
625   }
626
627   piBoardId (&model, &rev, &mem, &maker, &overVolted) ;
628
629   /**/ if ((model == PI_MODEL_A) || (model == PI_MODEL_B))
630     endPin = 16 ;
631   else if (model == PI_MODEL_BP)
632     endPin = 39 ;
633   else if (model == PI_MODEL_CM)
634   {
635     printf (" - Don't know how to reset a comput module:\n") ;
636     printf ("   Write a shell-script to reset the pins to the state you need.\n") ;
637     return ;
638   }
639   else
640   {
641     printf ("Oops - unable to determine board type... model: %d\n", model) ;
642     return ;
643   }
644
645   for (pin = 0 ; pin <= endPin ; ++pin)
646   {
647     if (wpiPinToGpio (pin) == -1)
648       continue ;
649
650     digitalWrite    (pin, LOW) ;
651     pinMode         (pin, INPUT) ;
652     pullUpDnControl (pin, PUD_OFF) ;
653   }
654 }
655
656
657 /*
658  * doMode:
659  *      gpio mode pin mode ...
660  *********************************************************************************
661  */
662
663 void doMode (int argc, char *argv [])
664 {
665   int pin ;
666   char *mode ;
667
668   if (argc != 4)
669   {
670     fprintf (stderr, "Usage: %s mode pin mode\n", argv [0]) ;
671     exit (1) ;
672   }
673
674   pin = atoi (argv [2]) ;
675
676   mode = argv [3] ;
677
678   /**/ if (strcasecmp (mode, "in")      == 0) pinMode         (pin, INPUT) ;
679   else if (strcasecmp (mode, "input")   == 0) pinMode         (pin, INPUT) ;
680   else if (strcasecmp (mode, "out")     == 0) pinMode         (pin, OUTPUT) ;
681   else if (strcasecmp (mode, "output")  == 0) pinMode         (pin, OUTPUT) ;
682   else if (strcasecmp (mode, "pwm")     == 0) pinMode         (pin, PWM_OUTPUT) ;
683   else if (strcasecmp (mode, "pwmTone") == 0) pinMode         (pin, PWM_TONE_OUTPUT) ;
684   else if (strcasecmp (mode, "clock")   == 0) pinMode         (pin, GPIO_CLOCK) ;
685   else if (strcasecmp (mode, "up")      == 0) pullUpDnControl (pin, PUD_UP) ;
686   else if (strcasecmp (mode, "down")    == 0) pullUpDnControl (pin, PUD_DOWN) ;
687   else if (strcasecmp (mode, "tri")     == 0) pullUpDnControl (pin, PUD_OFF) ;
688   else if (strcasecmp (mode, "off")     == 0) pullUpDnControl (pin, PUD_OFF) ;
689
690 // Undocumented
691
692   else if (strcasecmp (mode, "alt0")    == 0) pinModeAlt (pin, 0b100) ;
693   else if (strcasecmp (mode, "alt1")    == 0) pinModeAlt (pin, 0b101) ;
694   else if (strcasecmp (mode, "alt2")    == 0) pinModeAlt (pin, 0b110) ;
695   else if (strcasecmp (mode, "alt3")    == 0) pinModeAlt (pin, 0b111) ;
696   else if (strcasecmp (mode, "alt4")    == 0) pinModeAlt (pin, 0b011) ;
697   else if (strcasecmp (mode, "alt5")    == 0) pinModeAlt (pin, 0b010) ;
698   else
699   {
700     fprintf (stderr, "%s: Invalid mode: %s. Should be in/out/pwm/clock/up/down/tri\n", argv [1], mode) ;
701     exit (1) ;
702   }
703 }
704
705
706 /*
707  * doPadDrive:
708  *      gpio drive group value
709  *********************************************************************************
710  */
711
712 static void doPadDrive (int argc, char *argv [])
713 {
714   int group, val ;
715
716   if (argc != 4)
717   {
718     fprintf (stderr, "Usage: %s drive group value\n", argv [0]) ;
719     exit (1) ;
720   }
721
722   group = atoi (argv [2]) ;
723   val   = atoi (argv [3]) ;
724
725   if ((group < 0) || (group > 2))
726   {
727     fprintf (stderr, "%s: drive group not 0, 1 or 2: %d\n", argv [0], group) ;
728     exit (1) ;
729   }
730
731   if ((val < 0) || (val > 7))
732   {
733     fprintf (stderr, "%s: drive value not 0-7: %d\n", argv [0], val) ;
734     exit (1) ;
735   }
736
737   setPadDrive (group, val) ;
738 }
739
740
741 /*
742  * doUsbP:
743  *      Control USB Power - High (1.2A) or Low (600mA)
744  *      gpio usbp high/low
745  *********************************************************************************
746  */
747
748 static void doUsbP (int argc, char *argv [])
749 {
750   int model, rev, mem, maker, overVolted ;
751
752   if (argc != 3)
753   {
754     fprintf (stderr, "Usage: %s usbp high|low\n", argv [0]) ;
755     exit (1) ;
756   }
757
758 // Make sure we're on a B+
759
760   piBoardId (&model, &rev, &mem, &maker, &overVolted) ;
761
762   if (model != PI_MODEL_BP)
763   {
764     fprintf (stderr, "USB power contol is applicable to B+ boards only.\n") ;
765     exit (1) ;
766   }
767     
768 // Need to force BCM_GPIO mode:
769
770   wiringPiSetupGpio () ;
771
772   if ((strcasecmp (argv [2], "high") == 0) || (strcasecmp (argv [2], "hi") == 0))
773   {
774     digitalWrite (PI_USB_POWER_CONTROL, 1) ;
775     pinMode (PI_USB_POWER_CONTROL, OUTPUT) ;
776     printf ("Switched to HIGH current USB (1.2A)\n") ;
777     return ;
778   }
779
780   if ((strcasecmp (argv [2], "low") == 0) || (strcasecmp (argv [2], "lo") == 0))
781   {
782     digitalWrite (PI_USB_POWER_CONTROL, 0) ;
783     pinMode (PI_USB_POWER_CONTROL, OUTPUT) ;
784     printf ("Switched to LOW current USB (600mA)\n") ;
785     return ;
786   }
787
788   fprintf (stderr, "Usage: %s usbp high|low\n", argv [0]) ;
789   exit (1) ;
790 }
791
792
793 /*
794  * doGbw:
795  *      gpio gbw channel value
796  *      Gertboard Write - To the Analog output
797  *********************************************************************************
798  */
799
800 static void doGbw (int argc, char *argv [])
801 {
802   int channel, value ;
803
804   if (argc != 4)
805   {
806     fprintf (stderr, "Usage: %s gbw <channel> <value>\n", argv [0]) ;
807     exit (1) ;
808   }
809
810   channel = atoi (argv [2]) ;
811   value   = atoi (argv [3]) ;
812
813   if ((channel < 0) || (channel > 1))
814   {
815     fprintf (stderr, "%s: gbw: Channel number must be 0 or 1\n", argv [0]) ;
816     exit (1) ;
817   }
818
819   if ((value < 0) || (value > 255))
820   {
821     fprintf (stderr, "%s: gbw: Value must be from 0 to 255\n", argv [0]) ;
822     exit (1) ;
823   }
824
825   if (gertboardAnalogSetup (64) < 0)
826   {
827     fprintf (stderr, "Unable to initialise the Gertboard SPI interface: %s\n", strerror (errno)) ;
828     exit (1) ;
829   }
830
831   analogWrite (64 + channel, value) ;
832 }
833
834
835 /*
836  * doGbr:
837  *      gpio gbr channel
838  *      From the analog input
839  *********************************************************************************
840  */
841
842 static void doGbr (int argc, char *argv [])
843 {
844   int channel ;
845
846   if (argc != 3)
847   {
848     fprintf (stderr, "Usage: %s gbr <channel>\n", argv [0]) ;
849     exit (1) ;
850   }
851
852   channel = atoi (argv [2]) ;
853
854   if ((channel < 0) || (channel > 1))
855   {
856     fprintf (stderr, "%s: gbr: Channel number must be 0 or 1\n", argv [0]) ;
857     exit (1) ;
858   }
859
860   if (gertboardAnalogSetup (64) < 0)
861   {
862     fprintf (stderr, "Unable to initialise the Gertboard SPI interface: %s\n", strerror (errno)) ;
863     exit (1) ;
864   }
865
866   printf ("%d\n", analogRead (64 + channel)) ;
867 }
868
869
870 /*
871  * doWrite:
872  *      gpio write pin value
873  *********************************************************************************
874  */
875
876 static void doWrite (int argc, char *argv [])
877 {
878   int pin, val ;
879
880   if (argc != 4)
881   {
882     fprintf (stderr, "Usage: %s write pin value\n", argv [0]) ;
883     exit (1) ;
884   }
885
886   pin = atoi (argv [2]) ;
887
888   /**/ if ((strcasecmp (argv [3], "up") == 0) || (strcasecmp (argv [3], "on") == 0))
889     val = 1 ;
890   else if ((strcasecmp (argv [3], "down") == 0) || (strcasecmp (argv [3], "off") == 0))
891     val = 0 ;
892   else
893     val = atoi (argv [3]) ;
894
895   /**/ if (val == 0)
896     digitalWrite (pin, LOW) ;
897   else
898     digitalWrite (pin, HIGH) ;
899 }
900
901
902 /*
903  * doAwriterite:
904  *      gpio awrite pin value
905  *********************************************************************************
906  */
907
908 static void doAwrite (int argc, char *argv [])
909 {
910   int pin, val ;
911
912   if (argc != 4)
913   {
914     fprintf (stderr, "Usage: %s awrite pin value\n", argv [0]) ;
915     exit (1) ;
916   }
917
918   pin = atoi (argv [2]) ;
919
920   val = atoi (argv [3]) ;
921
922   analogWrite (pin, val) ;
923 }
924
925
926 /*
927  * doWriteByte:
928  *      gpio write value
929  *********************************************************************************
930  */
931
932 static void doWriteByte (int argc, char *argv [])
933 {
934   int val ;
935
936   if (argc != 3)
937   {
938     fprintf (stderr, "Usage: %s wb value\n", argv [0]) ;
939     exit (1) ;
940   }
941
942   val = (int)strtol (argv [2], NULL, 0) ;
943
944   digitalWriteByte (val) ;
945 }
946
947
948 /*
949  * doRead:
950  *      Read a pin and return the value
951  *********************************************************************************
952  */
953
954 void doRead (int argc, char *argv []) 
955 {
956   int pin, val ;
957
958   if (argc != 3)
959   {
960     fprintf (stderr, "Usage: %s read pin\n", argv [0]) ;
961     exit (1) ;
962   }
963
964   pin = atoi (argv [2]) ;
965   val = digitalRead (pin) ;
966
967   printf ("%s\n", val == 0 ? "0" : "1") ;
968 }
969
970
971 /*
972  * doAread:
973  *      Read an analog pin and return the value
974  *********************************************************************************
975  */
976
977 void doAread (int argc, char *argv []) 
978 {
979   if (argc != 3)
980   {
981     fprintf (stderr, "Usage: %s aread pin\n", argv [0]) ;
982     exit (1) ;
983   }
984
985   printf ("%d\n", analogRead (atoi (argv [2]))) ;
986 }
987
988
989 /*
990  * doToggle:
991  *      Toggle an IO pin
992  *********************************************************************************
993  */
994
995 void doToggle (int argc, char *argv [])
996 {
997   int pin ;
998
999   if (argc != 3)
1000   {
1001     fprintf (stderr, "Usage: %s toggle pin\n", argv [0]) ;
1002     exit (1) ;
1003   }
1004
1005   pin = atoi (argv [2]) ;
1006
1007   digitalWrite (pin, !digitalRead (pin)) ;
1008 }
1009
1010
1011 /*
1012  * doPwmTone:
1013  *      Output a tone in a PWM pin
1014  *********************************************************************************
1015  */
1016
1017 void doPwmTone (int argc, char *argv [])
1018 {
1019   int pin, freq ;
1020
1021   if (argc != 4)
1022   {
1023     fprintf (stderr, "Usage: %s pwmTone <pin> <freq>\n", argv [0]) ;
1024     exit (1) ;
1025   }
1026
1027   pin = atoi (argv [2]) ;
1028   freq = atoi (argv [3]) ;
1029
1030   pwmToneWrite (pin, freq) ;
1031 }
1032
1033
1034 /*
1035  * doClock:
1036  *      Output a clock on a pin
1037  *********************************************************************************
1038  */
1039
1040 void doClock (int argc, char *argv [])
1041 {
1042   int pin, freq ;
1043
1044   if (argc != 4)
1045   {
1046     fprintf (stderr, "Usage: %s clock <pin> <freq>\n", argv [0]) ;
1047     exit (1) ;
1048   }
1049
1050   pin = atoi (argv [2]) ;
1051
1052   freq = atoi (argv [3]) ;
1053
1054   gpioClockSet (pin, freq) ;
1055 }
1056
1057
1058 /*
1059  * doPwm:
1060  *      Output a PWM value on a pin
1061  *********************************************************************************
1062  */
1063
1064 void doPwm (int argc, char *argv [])
1065 {
1066   int pin, val ;
1067
1068   if (argc != 4)
1069   {
1070     fprintf (stderr, "Usage: %s pwm <pin> <value>\n", argv [0]) ;
1071     exit (1) ;
1072   }
1073
1074   pin = atoi (argv [2]) ;
1075
1076   val = atoi (argv [3]) ;
1077
1078   pwmWrite (pin, val) ;
1079 }
1080
1081
1082 /*
1083  * doPwmMode: doPwmRange: doPwmClock:
1084  *      Change the PWM mode, range and clock divider values
1085  *********************************************************************************
1086  */
1087
1088 static void doPwmMode (int mode)
1089 {
1090   pwmSetMode (mode) ;
1091 }
1092
1093 static void doPwmRange (int argc, char *argv [])
1094 {
1095   unsigned int range ;
1096
1097   if (argc != 3)
1098   {
1099     fprintf (stderr, "Usage: %s pwmr <range>\n", argv [0]) ;
1100     exit (1) ;
1101   }
1102
1103   range = (unsigned int)strtoul (argv [2], NULL, 10) ;
1104
1105   if (range == 0)
1106   {
1107     fprintf (stderr, "%s: range must be > 0\n", argv [0]) ;
1108     exit (1) ;
1109   }
1110
1111   pwmSetRange (range) ;
1112 }
1113
1114 static void doPwmClock (int argc, char *argv [])
1115 {
1116   unsigned int clock ;
1117
1118   if (argc != 3)
1119   {
1120     fprintf (stderr, "Usage: %s pwmc <clock>\n", argv [0]) ;
1121     exit (1) ;
1122   }
1123
1124   clock = (unsigned int)strtoul (argv [2], NULL, 10) ;
1125
1126   if ((clock < 1) || (clock > 4095))
1127   {
1128     fprintf (stderr, "%s: clock must be between 0 and 4096\n", argv [0]) ;
1129     exit (1) ;
1130   }
1131
1132   pwmSetClock (clock) ;
1133 }
1134
1135
1136 /*
1137  * main:
1138  *      Start here
1139  *********************************************************************************
1140  */
1141
1142 int main (int argc, char *argv [])
1143 {
1144   int i ;
1145   int model, rev, mem, maker, overVolted ;
1146
1147   if (getenv ("WIRINGPI_DEBUG") != NULL)
1148   {
1149     printf ("gpio: wiringPi debug mode enabled\n") ;
1150     wiringPiDebug = TRUE ;
1151   }
1152
1153   if (argc == 1)
1154   {
1155     fprintf (stderr, "%s\n", usage) ;
1156     return 1 ;
1157   }
1158
1159 // Help
1160
1161   if (strcasecmp (argv [1], "-h") == 0)
1162   {
1163     printf ("%s: %s\n", argv [0], usage) ;
1164     return 0 ;
1165   }
1166
1167 // Sort of a special:
1168
1169   if (strcmp (argv [1], "-R") == 0)
1170   {
1171     printf ("%d\n", piBoardRev ()) ;
1172     return 0 ;
1173   }
1174
1175 // Version & Warranty
1176
1177   if (strcmp (argv [1], "-V") == 0)
1178   {
1179     printf ("%d\n", piBoardRev ()) ;
1180     return 0 ;
1181   }
1182
1183   if (strcmp (argv [1], "-v") == 0)
1184   {
1185     printf ("gpio version: %s\n", VERSION) ;
1186     printf ("Copyright (c) 2012-2014 Gordon Henderson\n") ;
1187     printf ("This is free software with ABSOLUTELY NO WARRANTY.\n") ;
1188     printf ("For details type: %s -warranty\n", argv [0]) ;
1189     printf ("\n") ;
1190     piBoardId (&model, &rev, &mem, &maker, &overVolted) ;
1191     if (model == PI_MODEL_UNKNOWN)
1192     {
1193       printf ("Your Raspberry Pi has an unknown model type. Please report this to\n") ;
1194       printf ("    projects@drogon.net\n") ;
1195       printf ("with a copy of your /proc/cpuinfo if possible\n") ;
1196     }
1197     else
1198     {
1199       printf ("Raspberry Pi Details:\n") ;
1200       printf ("  Type: %s, Revision: %s, Memory: %dMB, Maker: %s %s\n", 
1201           piModelNames [model], piRevisionNames [rev], mem, piMakerNames [maker], overVolted ? "[OV]" : "") ;
1202     }
1203     return 0 ;
1204   }
1205
1206   if (strcasecmp (argv [1], "-warranty") == 0)
1207   {
1208     printf ("gpio version: %s\n", VERSION) ;
1209     printf ("Copyright (c) 2012-2014 Gordon Henderson\n") ;
1210     printf ("\n") ;
1211     printf ("    This program is free software; you can redistribute it and/or modify\n") ;
1212     printf ("    it under the terms of the GNU Leser General Public License as published\n") ;
1213     printf ("    by the Free Software Foundation, either version 3 of the License, or\n") ;
1214     printf ("    (at your option) any later version.\n") ;
1215     printf ("\n") ;
1216     printf ("    This program is distributed in the hope that it will be useful,\n") ;
1217     printf ("    but WITHOUT ANY WARRANTY; without even the implied warranty of\n") ;
1218     printf ("    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\n") ;
1219     printf ("    GNU Lesser General Public License for more details.\n") ;
1220     printf ("\n") ;
1221     printf ("    You should have received a copy of the GNU Lesser General Public License\n") ;
1222     printf ("    along with this program. If not, see <http://www.gnu.org/licenses/>.\n") ;
1223     printf ("\n") ;
1224     return 0 ;
1225   }
1226
1227   if (geteuid () != 0)
1228   {
1229     fprintf (stderr, "%s: Must be root to run. Program should be suid root. This is an error.\n", argv [0]) ;
1230     return 1 ;
1231   }
1232
1233 // Initial test for /sys/class/gpio operations:
1234
1235   /**/ if (strcasecmp (argv [1], "exports"    ) == 0)   { doExports     (argc, argv) ;  return 0 ; }
1236   else if (strcasecmp (argv [1], "export"     ) == 0)   { doExport      (argc, argv) ;  return 0 ; }
1237   else if (strcasecmp (argv [1], "edge"       ) == 0)   { doEdge        (argc, argv) ;  return 0 ; }
1238   else if (strcasecmp (argv [1], "unexport"   ) == 0)   { doUnexport    (argc, argv) ;  return 0 ; }
1239   else if (strcasecmp (argv [1], "unexportall") == 0)   { doUnexportall (argv [0]) ;    return 0 ; }
1240
1241 // Check for load command:
1242
1243   if (strcasecmp (argv [1], "load" ) == 0)      { doLoad     (argc, argv) ; return 0 ; }
1244
1245 // Gertboard commands
1246
1247   if (strcasecmp (argv [1], "gbr" ) == 0)       { doGbr (argc, argv) ; return 0 ; }
1248   if (strcasecmp (argv [1], "gbw" ) == 0)       { doGbw (argc, argv) ; return 0 ; }
1249
1250 // Check for -g argument
1251
1252   /**/ if (strcasecmp (argv [1], "-g") == 0)
1253   {
1254     wiringPiSetupGpio () ;
1255
1256     for (i = 2 ; i < argc ; ++i)
1257       argv [i - 1] = argv [i] ;
1258     --argc ;
1259     wpMode = WPI_MODE_GPIO ;
1260   }
1261
1262 // Check for -1 argument
1263
1264   else if (strcasecmp (argv [1], "-1") == 0)
1265   {
1266     wiringPiSetupPhys () ;
1267
1268     for (i = 2 ; i < argc ; ++i)
1269       argv [i - 1] = argv [i] ;
1270     --argc ;
1271     wpMode = WPI_MODE_PHYS ;
1272   }
1273
1274 // Check for -p argument for PiFace
1275
1276   else if (strcasecmp (argv [1], "-p") == 0)
1277   {
1278     piFaceSetup (200) ;
1279
1280     for (i = 2 ; i < argc ; ++i)
1281       argv [i - 1] = argv [i] ;
1282     --argc ;
1283     wpMode = WPI_MODE_PIFACE ;
1284   }
1285
1286 // Default to wiringPi mode
1287
1288   else
1289   {
1290     wiringPiSetup () ;
1291     wpMode = WPI_MODE_PINS ;
1292   }
1293
1294 // Check for -x argument to load in a new extension
1295
1296   if (strcasecmp (argv [1], "-x") == 0)
1297   {
1298     if (argc < 3)
1299     {
1300       fprintf (stderr, "%s: -x missing extension specification.\n", argv [0]) ;
1301       exit (EXIT_FAILURE) ;
1302     }
1303
1304     if (!doExtension (argv [0], argv [2]))      // Prints its own error messages
1305       exit (EXIT_FAILURE) ;
1306
1307     for (i = 3 ; i < argc ; ++i)
1308       argv [i - 2] = argv [i] ;
1309     argc -= 2 ;
1310   }
1311
1312   if (argc <= 1)
1313   {
1314     fprintf (stderr, "%s: no command given\n", argv [0]) ;
1315     exit (EXIT_FAILURE) ;
1316   }
1317
1318 // Core wiringPi functions
1319
1320   /**/ if (strcasecmp (argv [1], "mode"   ) == 0) doMode      (argc, argv) ;
1321   else if (strcasecmp (argv [1], "read"   ) == 0) doRead      (argc, argv) ;
1322   else if (strcasecmp (argv [1], "write"  ) == 0) doWrite     (argc, argv) ;
1323   else if (strcasecmp (argv [1], "pwm"    ) == 0) doPwm       (argc, argv) ;
1324   else if (strcasecmp (argv [1], "awrite" ) == 0) doAwrite    (argc, argv) ;
1325   else if (strcasecmp (argv [1], "aread"  ) == 0) doAread     (argc, argv) ;
1326
1327 // GPIO Nicies
1328
1329   else if (strcasecmp (argv [1], "toggle" ) == 0) doToggle    (argc, argv) ;
1330
1331 // Pi Specifics
1332
1333   else if (strcasecmp (argv [1], "pwm-bal"  ) == 0) doPwmMode    (PWM_MODE_BAL) ;
1334   else if (strcasecmp (argv [1], "pwm-ms"   ) == 0) doPwmMode    (PWM_MODE_MS) ;
1335   else if (strcasecmp (argv [1], "pwmr"     ) == 0) doPwmRange   (argc, argv) ;
1336   else if (strcasecmp (argv [1], "pwmc"     ) == 0) doPwmClock   (argc, argv) ;
1337   else if (strcasecmp (argv [1], "pwmTone"  ) == 0) doPwmTone    (argc, argv) ;
1338   else if (strcasecmp (argv [1], "drive"    ) == 0) doPadDrive   (argc, argv) ;
1339   else if (strcasecmp (argv [1], "usbp"     ) == 0) doUsbP       (argc, argv) ;
1340   else if (strcasecmp (argv [1], "readall"  ) == 0) doReadall    () ;
1341   else if (strcasecmp (argv [1], "nreadall" ) == 0) doReadall    () ;
1342   else if (strcasecmp (argv [1], "pins"     ) == 0) doPins       () ;
1343   else if (strcasecmp (argv [1], "i2cdetect") == 0) doI2Cdetect  (argc, argv) ;
1344   else if (strcasecmp (argv [1], "i2cd"     ) == 0) doI2Cdetect  (argc, argv) ;
1345   else if (strcasecmp (argv [1], "reset"    ) == 0) doReset      (argv [0]) ;
1346   else if (strcasecmp (argv [1], "wb"       ) == 0) doWriteByte  (argc, argv) ;
1347   else if (strcasecmp (argv [1], "clock"    ) == 0) doClock      (argc, argv) ;
1348   else if (strcasecmp (argv [1], "wfi"      ) == 0) doWfi        (argc, argv) ;
1349   else
1350   {
1351     fprintf (stderr, "%s: Unknown command: %s.\n", argv [0], argv [1]) ;
1352     exit (EXIT_FAILURE) ;
1353   }
1354   return 0 ;
1355 }