From 5c8e9386f879394f81aa24460de7a820563657dc Mon Sep 17 00:00:00 2001 From: makefu Date: Thu, 20 Dec 2012 10:46:35 +0700 Subject: //claws -> //god/claws claws provides interaction with the real world --- god/claws/.gitignore | 1 + god/claws/Makefile | 18 ++ god/claws/README.md | 12 ++ god/claws/communication.h | 25 +++ god/claws/rcontrol.c | 479 ++++++++++++++++++++++++++++++++++++++++++++++ 5 files changed, 535 insertions(+) create mode 100644 god/claws/.gitignore create mode 100644 god/claws/Makefile create mode 100644 god/claws/README.md create mode 100644 god/claws/communication.h create mode 100644 god/claws/rcontrol.c (limited to 'god') diff --git a/god/claws/.gitignore b/god/claws/.gitignore new file mode 100644 index 00000000..5da5924f --- /dev/null +++ b/god/claws/.gitignore @@ -0,0 +1 @@ +rcontrol diff --git a/god/claws/Makefile b/god/claws/Makefile new file mode 100644 index 00000000..abfae5eb --- /dev/null +++ b/god/claws/Makefile @@ -0,0 +1,18 @@ +# Makefile for the rcontrol program +# We set a setuid bit as this tools needs root privilleges to open the FT232 device. + +BINARY=./rcontrol + +install: all + ln -snf $$PWD/$(BINARY) ../bin/ + +all:$(BINARY) + chown root.root $(BINARY) + chmod +s $(BINARY) + +$(BINARY): rcontrol.c + #punani install libftdi1 libftdi-dev + gcc -Wall -lusb -lftdi rcontrol.c -o $(BINARY) + +clean: + rm -f $(BINARY) diff --git a/god/claws/README.md b/god/claws/README.md new file mode 100644 index 00000000..00c90a85 --- /dev/null +++ b/god/claws/README.md @@ -0,0 +1,12 @@ +# CLAWS + +Claws is a tool to manage some obscure Hardware Relay Board currently attached +to the krebs infrastructure + +# Installation + + make install + +# Usage + + rcontrol --help diff --git a/god/claws/communication.h b/god/claws/communication.h new file mode 100644 index 00000000..db4b2c36 --- /dev/null +++ b/god/claws/communication.h @@ -0,0 +1,25 @@ +/* These are the command codes of the relay card. If you change something here you must + * recompile the firmware and the control tool. */ +#ifndef _COMMUNICATION_H +#define _COMMUNICATION_H + +#define COMMAND_RELAY_ON 0x01 +#define COMMAND_RELAY_OFF 0x02 +#define COMMAND_RELAY_TOGGLE 0x04 +#define COMMAND_RELAY_SET 0x08 +#define COMMAND_RELAY_TIME_ON 0x10 +#define COMMAND_RELAY_TIME_OFF 0x20 +#define COMMAND_RELAY_TIME_CYCLIC 0x40 +#define COMMAND_GET_STATUS 0x80 +#define COMMAND_DEL_TIMERS 0x81 +#define COMMAND_SETUP_REMOTE 0x82 + + +#define RESPONSE_OK 0xff +#define RESPONSE_INVALID_COMMAND 0xfe +#define RESPONSE_INVALID_ARGUMENT 0xfd +#define RESPONSE_TRANSMISSION_ERROR 0xfc + +#define COMMANDO_LENGTH 4 + +#endif diff --git a/god/claws/rcontrol.c b/god/claws/rcontrol.c new file mode 100644 index 00000000..972c97d0 --- /dev/null +++ b/god/claws/rcontrol.c @@ -0,0 +1,479 @@ +/* +----------------------------------------------------------------------+ + * | relay control program | + * | by mgr, 2007 | + * | last change: 2009-01-05 | + * | | + * | This program is used to control the relay card version 1.0. For more | + * | information have a look at the project homepage. | + * | You will need libftdi in order to compile this tool. | + * | | + * | NOTE: For some reason the -l option causes a program crash if I | + * | compile this program with -O2. On top of that this code seems to | + * | be quite optimal, the program size gets a little larger with -O2, | + * | so I suggest you just compile it without optimization. | + * +----------------------------------------------------------------------+ + */ + +#include +#include +#include +#include +#include +#include +#include +#include // libftdi + +/* Notice that if you experiment with the baud rate, you will have to adapt + * the firmware, too. Also I do not recommend it, as 9600 Bauds are completely + * sufficient for this application. */ +#define BAUD 9600 + +/* If you are using more than one FT232-based pieces of hardware at once, + * we need a way to uniquely address any given one. This is done by the + * serial of the specific device which you can pass to this tool or specify + * here. If no serial is specified (NULL), the first found device is opened. */ +#define DEFAULT_FT_SERIAL "A6TMRSS6" + +#define VERSION "1.0" + +#define OPTION_ADDRESS 0x01 +#define OPTION_INTERVAL 0x04 +#define OPTION_CYCLIC 0x08 +#define OPTION_ON 0x10 +#define OPTION_OFF 0x20 +#define OPTION_TOGGLE 0x40 +#define OPTION_DEL_TIMERS 0x80 +#define OPTION_LIST_DEVICES 0x200 + +#define EXIT_CODE_OK 0 +#define EXIT_CODE_FAILURE 1 + +#include "communication.h" + +/* function prototypes */ +void usage(char* name); +void version(void); +const char* card_strerror(int error); +int valid_argument(const char* str); +void exit_gracefully(struct ftdi_context* ftdic, char exit_code); + + +int main(int argc, char **argv) +{ + int ret=0, int_argument=0, option_flags=0, long_index=0, i=0, num_ops=0; + char c=0; + unsigned char buf[COMMANDO_LENGTH], char_argument=0, operation=0; + const char* ft_serial=DEFAULT_FT_SERIAL; + double double_argument; + char buf0[64], buf1[64], buf2[64]; + time_t start_time; + + //struct ftdi_eeprom eeprom; + + struct ftdi_context ftdic; + struct ftdi_device_list *devlist=NULL, *curdev=NULL; + + static struct option long_options[] = + { + {"help", 0, 0, '?'}, + {"version", 0, 0, 'V'}, + {"on", 1, 0, 'o'}, + {"off", 1, 0, 'f'}, + {"toggle", 1, 0, 't'}, + {"set", 1, 0, 's'}, + {"status", 0, 0, 'S'}, + {"interval", 1, 0, 'v'}, + {"cyclic", 0, 0, 'c'}, + {"address", 1, 0, 'a'}, + {"delete-timers", 0, 0, 'd'}, + {"list-devices", 0, 0, 'l'}, + {0, 0, 0, 0 } + }; + + /* fetch the command line options */ + while ((c = getopt_long_only(argc, argv, "?Vo:f:t:s:Sv:ca:dl", long_options, + &long_index)) != -1) + { + switch (c) + { + case 'o': case 'f': case 't': + int_argument = atoi(optarg); + + if (int_argument > 6 || int_argument < 1 || !valid_argument(optarg)) + { + fprintf(stderr, "%s: -s: invalid value `%s' (1-6 is valid)\n", + *argv, optarg); + return EXIT_CODE_FAILURE; + } + + char_argument = (unsigned char)int_argument; + + if (c == 't') + { + operation = COMMAND_RELAY_TOGGLE; + option_flags |= OPTION_TOGGLE; + } else if (c == 'o') + { + operation = COMMAND_RELAY_ON; + option_flags |= OPTION_ON; + } else + { + operation = COMMAND_RELAY_OFF; + option_flags |= OPTION_OFF; + } + + num_ops++; + + break; + + case 's': + int_argument = atoi(optarg); + + if (int_argument > (1 << 6)-1 || int_argument < 0 || !valid_argument(optarg)) + { + fprintf(stderr, "%s: -s: invalid value `%s'\n", *argv, optarg); + return EXIT_CODE_FAILURE; + } + + char_argument = (unsigned char)int_argument; + operation = COMMAND_RELAY_SET; + num_ops++; + break; + + case 'S': + operation = COMMAND_GET_STATUS; + num_ops++; + break; + + case 'v': + double_argument = atof(optarg); + int_argument = (int)(double_argument*60) /10; + + if (int_argument < 1 || int_argument > (1 << 16)-1) + { + fprintf(stderr, "%s: -i: invalid interval `%s'\n", *argv, optarg); + return EXIT_CODE_FAILURE; + } + + option_flags |= OPTION_INTERVAL; + break; + + case 'c': + option_flags |= OPTION_CYCLIC; + break; + + case 'd': + operation = COMMAND_DEL_TIMERS; + option_flags |= OPTION_DEL_TIMERS; + num_ops++; + break; + + case 'a': + if (strlen(optarg) != 8) + { + fprintf(stderr, "%s: -s: invalid serial number `%s'\n", *argv, optarg); + return EXIT_CODE_FAILURE; + } + + ft_serial = optarg; + option_flags |= OPTION_ADDRESS; + break; + + case 'l': + option_flags |= OPTION_LIST_DEVICES; + break; + + case 'V': + version(); + break; + case '?': default: + usage(*argv); + break; + } + } + + /* check whether the command line options are valid */ + if ((option_flags & OPTION_INTERVAL)) + { + if (option_flags & OPTION_DEL_TIMERS) + { + fprintf(stderr, "%s: -d cannot be mixed with timing options\n", *argv); + usage(*argv); + } + + if (option_flags & OPTION_CYCLIC) + { + operation = COMMAND_RELAY_TIME_CYCLIC; + } else if (option_flags & OPTION_ON) + { + operation = COMMAND_RELAY_TIME_ON; + } else if (option_flags & OPTION_OFF) + { + operation = COMMAND_RELAY_TIME_OFF; + } else + { + fprintf(stderr, "%s: -v: you must also specify an operation (-o or -f)\n", *argv); + usage(*argv); + } + } + + if (!operation && !(option_flags & OPTION_LIST_DEVICES)) + { + usage(*argv); + } + + if (((option_flags & OPTION_DEL_TIMERS) && (operation != COMMAND_DEL_TIMERS))) + { + fprintf(stderr, "%s: invalid mixture of options\n", *argv); + usage(*argv); + } + + if (num_ops > 1) + { + fprintf(stderr, "%s: more than one operation specified\n", *argv); + usage(*argv); + } + + if (ftdi_init(&ftdic) < 0) + { + fprintf(stderr, "%s: unable to initialize FTDI context: %d (%s)\n", *argv, ret, + ftdi_get_error_string(&ftdic)); + return EXIT_CODE_FAILURE; + } + + /* list all found FT232 devices */ + if (option_flags & OPTION_LIST_DEVICES) + { + printf("scanning for FT232 devices...\n" + "you can address the devices using `%s -a '\n", *argv); + + if ((ret = ftdi_usb_find_all(&ftdic, &devlist, 0x0403, 0x6001)) < 0) + { + fprintf(stderr, "%s: unable to scan devices: %d (%s)\n", *argv, ret, + ftdi_get_error_string(&ftdic)); + exit(EXIT_CODE_FAILURE); + } + + if (ret == 0) + { + printf(" no devices found :(\n"); + return EXIT_CODE_OK; + } + + for (i=0, curdev = devlist; curdev != NULL; i++) + { + if (ftdi_usb_get_strings(&ftdic, curdev->dev, buf0, sizeof(buf0)/sizeof(char), + buf1, sizeof(buf1)/sizeof(char), buf2, sizeof(buf2)/sizeof(char)) < 0) + { + fprintf(stderr, "unable to fetch information for device #%i: %s\n", i, + ftdi_get_error_string(&ftdic)); + // continue caused an endless loop in case of an error + break; + } + + printf("\ndevice #%i%s:\n" + " manufacturer: %s\n" + " device: %s\n" + " serial: %s\n", i, (i == 0 ? " (default)" : ""), + (buf0 != NULL ? buf0 : "n/a"), (buf1 != NULL ? buf1 : "n/a"), + (buf2 != NULL ? buf2 : "n/a")); + + curdev = curdev->next; + } + + ftdi_list_free(&devlist); + return EXIT_CODE_OK; + } + + /* Try to open the specified device. If that fails, we take a long shot + * and open the first found FT232 device and assume its the relay card. + * We don't do this if an address was specified with the -a option. */ + if ((ret = ftdi_usb_open_desc(&ftdic, 0x0403, 0x6001, NULL, ft_serial)) < 0) + { + fprintf(stderr, "%s: unable to open ftdi device: %d (%s)\n", *argv, ret, + ftdi_get_error_string(&ftdic)); + exit(EXIT_CODE_FAILURE); + } + + /* get rid of any data still floating around the buffer */ + ftdi_usb_reset(&ftdic); + ftdi_usb_purge_buffers(&ftdic); + + + if ((ret = ftdi_set_baudrate(&ftdic, BAUD)) < 0) + { + fprintf(stderr, "%s: unable to set baudrate: %d (%s)\n", *argv, ret, + ftdi_get_error_string(&ftdic)); + exit_gracefully(&ftdic, EXIT_CODE_FAILURE); + } + + if ((ret = ftdi_set_line_property(&ftdic, 8, 2, NONE)) < 0) + { + fprintf(stderr, "%s: unable to set line property: %d (%s)\n", *argv, ret, + ftdi_get_error_string(&ftdic)); + exit_gracefully(&ftdic, EXIT_CODE_FAILURE); + } + + if ((ret = ftdi_setflowctrl(&ftdic, SIO_RTS_CTS_HS)) < 0) { + fprintf(stderr, "%s: unable to setup flow control: %d (%s)\n", *argv, ret, + ftdi_get_error_string(&ftdic)); + exit_gracefully(&ftdic, EXIT_CODE_FAILURE); + } + + /*if ((ret = ftdi_set_latency_timer(&ftdic, 10)) < 0) + { + fprintf(stderr, "%s: unable to set latency timer: %d (%s)\n", *argv, ret, + ftdi_get_error_string(&ftdic)); + exit_gracefully(&ftdic, EXIT_CODE_FAILURE); + }*/ + + buf[0] = operation; + buf[2] = 0; buf[3] = 0; buf[4] = 0; + + switch (operation) + { + case COMMAND_RELAY_SET: + buf[1] = char_argument; + break; + + case COMMAND_RELAY_ON: case COMMAND_RELAY_OFF: + case COMMAND_RELAY_TOGGLE: + buf[1] = (1 << (char_argument-1)); + break; + + case COMMAND_RELAY_TIME_ON: + case COMMAND_RELAY_TIME_OFF: + case COMMAND_RELAY_TIME_CYCLIC: + buf[1] = char_argument-1; + buf[2] = (int_argument & 0xff); // low byte + buf[3] = (int_argument >> 8); // high byte + break; + + default: + break; + } + + /* These values might not make much sense are vital to the correct + * funtion of this program, so better don't touch them. */ + ftdi_write_data_set_chunksize(&ftdic, 1); + ftdi_read_data_set_chunksize(&ftdic, 4); + + /* send the command */ + if (ftdi_write_data(&ftdic, buf, COMMANDO_LENGTH) != COMMANDO_LENGTH) + { + fprintf(stderr, "%s: unable to send command: %s\n", *argv, + ftdi_get_error_string(&ftdic)); + exit_gracefully(&ftdic, EXIT_CODE_FAILURE); + } + + /* Read the card's response. */ + start_time = time(NULL); + while ((ret = ftdi_read_data(&ftdic, buf, 1)) == 0) { + usleep(500); + if (time(NULL)-start_time >= 2) { + fprintf(stderr, "%s: unable to read card response, the operation might have " + "failed\n", *argv); + exit_gracefully(&ftdic, EXIT_FAILURE); + } + } + + if (operation == COMMAND_GET_STATUS && buf[0] <= ((1 << 7)-1)) + { + printf("relay status: %i (0b%s%s%s%s%s%s)\n", buf[0], + (buf[0] & (1 << 5)) ? "1" : "0", + (buf[0] & (1 << 4)) ? "1" : "0", + (buf[0] & (1 << 3)) ? "1" : "0", + (buf[0] & (1 << 2)) ? "1" : "0", + (buf[0] & (1 << 1)) ? "1" : "0", + (buf[0] & (1 << 0)) ? "1" : "0"); + exit_gracefully(&ftdic, EXIT_CODE_OK); + } + + if (buf[0] != RESPONSE_OK) + { + fprintf(stderr, "%s: relay card returned: %s\n", *argv, card_strerror(buf[0])); + exit_gracefully(&ftdic, EXIT_FAILURE); + } + + /* we can exit now */ + exit_gracefully(&ftdic, EXIT_CODE_OK); + return 0; // to make the compiler happy +} + +void exit_gracefully(struct ftdi_context* ftdic, char exit_code) +{ + ftdi_usb_purge_buffers(ftdic); + ftdi_usb_close(ftdic); + ftdi_deinit(ftdic); + + exit(exit_code); +} + +int valid_argument(const char* str) +{ + int i; + + for (i=0; i: ......... switch relay on\n" + " -f/--off : ........ switch relay off \n" + " -t/--toggle : ..... toggle relay \n" + " -s/--set : ....... set the status of all relays to \n" + " -S/--status: ........... get the current relay status\n" + " -v/--interval .. specify a timing interval (in minutes)\n" + " -c/--cyclic: ........... makes a timing operation (-v) cyclic\n" + " -a/--address .. addresses a specific card if multiple are installed\n" + " -d/--delete-timers ..... deletes all active timers\n" + " -l/--list-devices ...... lists all found FT232 devices\n\n"); + exit(0); +} + +void version(void) +{ + printf("\nThis is the relay control program version %s ($Revision: 26 $)\n" + "----------------------------------------------------------------\n" + " written by Michael Gross, 2007\n" + " binary compiled: %s %s\n\n" + "This program can be redistributed under the terms of the GNU GPL version 2\n" + "or later. For more information about this software and the hardware, visit\n" + "my homepage at http://www.coremelt.net. As usual with free software, there\n" + "is ABSOLUTELY NO WARRANTY. For details, refer to the GPL.\n\n", VERSION, + __DATE__, __TIME__); + + exit(0); +} -- cgit v1.2.3