First support PL2303-HXD GPIO 2 & 3
[pub/pl2303-ft232-gpio.git] / main.c
diff --git a/main.c b/main.c
index f24e40f..6dbb9d0 100644 (file)
--- a/main.c
+++ b/main.c
@@ -9,23 +9,19 @@
 #include <termios.h>
 #include <fcntl.h>
 #include <signal.h>
-#include <usb.h>
+#include <libusb.h>
 #include <getopt.h>
 
 
 #define _GNU_SOURCE
 #include <getopt.h>
 
-
-#define I_VENDOR_NUM        0x67b
-#define I_PRODUCT_NUM       0x2303
-
-
-#define VENDOR_READ_REQUEST_TYPE       0xc0
-#define VENDOR_READ_REQUEST            0x01
-
-#define VENDOR_WRITE_REQUEST_TYPE      0x40
-#define VENDOR_WRITE_REQUEST           0x01
+int get_device_vid();
+int get_device_pid();
+void check_handle(libusb_device_handle **h, int vid, int pid, const char* manuf, const char* product, const char* serial);
+int gpio_read(libusb_device_handle *h, int gpio);
+void gpio_in(libusb_device_handle *h, int gpio, int pullup);
+void gpio_out(libusb_device_handle *h, int gpio, int value);
 
 
 void handle_error(int ret)
@@ -37,111 +33,35 @@ void handle_error(int ret)
        }
 }
 
-
-/* Get current GPIO register from PL2303 */
-char gpio_read_reg(usb_dev_handle *h)
-{
-       char buf;
-       int bytes = usb_control_msg(
-               h,             // handle obtained with usb_open()
-               VENDOR_READ_REQUEST_TYPE, // bRequestType
-               VENDOR_READ_REQUEST,      // bRequest
-               0x0081,              // wValue
-               0,              // wIndex
-               &buf,             // pointer to destination buffer
-               1,  // wLength
-               100
-               );
-       handle_error(bytes);
-       return buf;
-}
-
-void gpio_write_reg(usb_dev_handle *h, unsigned char reg)
-{
-       int bytes = usb_control_msg(
-               h,             // handle obtained with usb_open()
-               VENDOR_WRITE_REQUEST_TYPE, // bRequestType
-               VENDOR_WRITE_REQUEST,      // bRequest
-               1,              // wValue
-               reg,              // wIndex
-               0,             // pointer to destination buffer
-               0,  // wLength
-               6000
-               );
-       handle_error(bytes);
-       
-}
-
-int gpio_dir_shift(int gpio) {
-       if (gpio == 0) 
-               return 4;
-       if (gpio == 1) 
-               return 5;
-       return 4; /* default to 0 */
-}
-
-int gpio_val_shift(int gpio) {
-       if (gpio == 0) 
-               return 6;
-       if (gpio == 1) 
-               return 7;
-       return 6; /* default to 0 */
-}
-
-
-void gpio_out(usb_dev_handle *h, int gpio, int value)
-{
-       int shift_dir = gpio_dir_shift(gpio);
-       int shift_val = gpio_val_shift(gpio);
-       unsigned char reg = gpio_read_reg(h);
-       reg |= (1 << shift_dir);
-       reg &= ~(1 << shift_val);
-       reg |= (value << shift_val);
-       gpio_write_reg(h, reg);
-}
-
-void gpio_in(usb_dev_handle *h, int gpio, int pullup)
-{
-       int shift_dir = gpio_dir_shift(gpio);
-       int shift_val = gpio_val_shift(gpio);
-
-       unsigned char reg = gpio_read_reg(h);
-       reg &= ~(1 << shift_dir);
-       reg &= ~(1 << shift_val);
-       reg |= (pullup << shift_val);
-       gpio_write_reg(h, reg);
-}
-
-int gpio_read(usb_dev_handle *h, int gpio)
-{
-       unsigned char r = gpio_read_reg(h);
-       int shift = gpio_val_shift(gpio);
-       return (r & (1<<shift));
-}
-
 static struct option long_options[] =
 {
        /* These options set a flag. */
-       {"help",    no_argument,       0, 'h'},
-       {"gpio",    required_argument, 0, 'g'},
-       {"in",      optional_argument, 0, 'i'},
-       {"out",     required_argument, 0, 'o'},
-       {"sleep",   required_argument, 0, 's'},
-       {"read",    no_argument,       0, 'r'},
+       {"help",    no_argument,             0, 'h'},
+       {"gpio",    required_argument,       0, 'g'},
+       {"in",      optional_argument,       0, 'i'},
+       {"out",     required_argument,       0, 'o'},
+       {"sleep",   required_argument,       0, 's'},
+       {"read",    no_argument,             0, 'r'},
+       {"product", required_argument,       0, 'p'},
+       {"manuf",   required_argument,       0, 'm'},
+       {"serial",  required_argument,       0, 'n'},
        {0, 0, 0, 0}
 };
   
 void usage(const char *self)
 {
-       printf("PL2303HXA userspace GPIO control tool\n"
+       printf("PL2303HXA/CP2103 userspace GPIO control tool\n"
               "(c) Andrew 'Necromant' Andrianov 2014, License: GPLv3\n"
               "Usage: %s [action1] [action2] ...\n"
               "Options are: \n"
-              "\t -g/--gpio  n  - select GPIO, n=0, 1\n"
-              "\t -i/--in       - configure GPIO as input\n"
-              "\t -o/--out v    - configure GPIO as output with value v\n"
-              "\t -r/--read v   - Read current GPIO value\n\n"
-              "\t -s/--sleep v  - Delay for v ms\n\n"
+              "\t --product=blah    - Use device with this 'product' string\n"
+              "\t --serial=blah     - Use device with this 'serial' string\n"
+              "\t --manuf=blah      - Use device with this 'manufacturer' string\n"
+              "\t -g/--gpio  n      - select GPIO, n=0, 1\n"
+              "\t -i/--in           - configure GPIO as input\n"
+              "\t -o/--out v        - configure GPIO as output with value v\n"
+              "\t -r/--read v       - Read current GPIO value\n\n"
+              "\t -s/--sleep v      - Delay for v ms\n\n"
               "Examples: \n"
               "\t%s --gpio=1 --out 1\n"
               "\t%s --gpio=0 --out 0 --gpio=1 --in\n\n"
@@ -151,29 +71,14 @@ void usage(const char *self)
               "\n", self, self, self, self);
 }
 
-extern usb_dev_handle *nc_usb_open(int vendor, int product, char *vendor_name, char *product_name, char *serial);
-void check_handle(usb_dev_handle **h)
-{
-       if (*h)
-               return;
-
-       /* TODO: Make a proper way to select different PL2303 devices. */
-       *h = nc_usb_open(I_VENDOR_NUM, I_PRODUCT_NUM, NULL, NULL, NULL);
-       if (!(*h)) {
-               fprintf(stderr, "No PL2303 USB device %04x:%04x found ;(\n", I_VENDOR_NUM, I_PRODUCT_NUM);
-               exit(1);
-       }
-
-       /* We don't set config or claim interface, pl2303 kernel driver does 
-        * that for us. 
-        */
-}
-
 int main(int argc, char* argv[])
 {
-       char c;
-       usb_dev_handle *h = NULL;
+       int c;
+       libusb_device_handle *h = NULL;
        int gpio=0; 
+       const char *product = NULL; 
+       const char *manuf = NULL;
+       const char *serial = NULL;  
        if (argc == 1) 
        {
                usage(argv[0]);
@@ -191,6 +96,15 @@ int main(int argc, char* argv[])
                
                switch (c)
                {
+               case 'p':
+                       product = optarg;
+                       break;
+               case 'm':
+                       manuf = optarg;
+                       break;
+               case 'n':
+                       serial = optarg;
+                       break;
                case 'h':
                        usage(argv[0]);
                        exit(1);
@@ -201,7 +115,7 @@ int main(int argc, char* argv[])
                case 'i':
                {
                        int v=0; 
-                       check_handle(&h);
+                       check_handle(&h, get_device_vid(), get_device_pid(), manuf, product, serial);
                        if (optarg)
                                v = atoi(optarg);
                        gpio_in(h, gpio, v); 
@@ -210,7 +124,7 @@ int main(int argc, char* argv[])
                case 'o':
                {
                        int v=0; 
-                       check_handle(&h);
+                       check_handle(&h, get_device_vid(), get_device_pid(), manuf, product, serial);
                        if (optarg)
                                v = atoi(optarg);
                        gpio_out(h, gpio, v); 
@@ -224,7 +138,7 @@ int main(int argc, char* argv[])
                }
                case 'r': 
                {
-                       check_handle(&h);
+                       check_handle(&h, get_device_vid(), get_device_pid(), manuf, product, serial);
                        printf("%d\n", gpio_read(h, gpio) ? 1 : 0); 
                        break;
                }