Files
Robotd/robotd.c

132 lines
3.2 KiB
C

#include <Python.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <signal.h>
#include <unistd.h>
#include <errno.h>
#include <syslog.h>
#include <stdbool.h>
#include <mraa.h>
#define GPIO_PIN_4 4
#define GPIO_PIN_13 13
mraa_result_t mr;
mraa_gpio_context pin13, pin4;
void clean_up() {
syslog(LOG_DAEMON | LOG_WARNING, "Cleaning up.");
mr = mraa_gpio_write(pin4, 0);
// TODO: MRAA error handling
closelog();
}
void sig_handler(int signal) {
// Turn off the red LED
syslog(LOG_DAEMON | LOG_WARNING, "Robot daemon interrupted.");
clean_up();
exit(EXIT_FAILURE);
}
int main(int argc, char ** argv) {
int rv;
bool done = false;
sighandler_t sh;
// Turn the process into a daemon.
rv = daemon(0, 1);
if (rv != 0) {
perror("daemon(0, 1)");
return EXIT_FAILURE;
}
// Set SIGINT signal handler
sh = signal(SIGINT, sig_handler);
if (sh == SIG_ERR) {
perror("sh = signal(SIGINT, sig_handler)");
return EXIT_FAILURE;
}
// Open a syslog connection
openlog(argv[0], LOG_NDELAY | LOG_PID, LOG_DAEMON);
syslog(LOG_DAEMON | LOG_NOTICE, "Started robot daemon.");
// Open PIN 13 as a PULLUP input pin for the push button
pin13 = mraa_gpio_init(GPIO_PIN_13);
mr = mraa_gpio_dir(pin13, MRAA_GPIO_IN);
mr = mraa_gpio_mode(pin13, MRAA_GPIO_PULLUP);
syslog(LOG_DAEMON | LOG_NOTICE, "Push button ready.");
// TODO: MRAA error handling
// Open PIN 4 as an output pin for the red indicator LED
pin4 = mraa_gpio_init(GPIO_PIN_4);
mr = mraa_gpio_dir(pin4, MRAA_GPIO_OUT);
syslog(LOG_DAEMON | LOG_NOTICE, "Red LED ready.");
// TODO: MRAA error handling
// Start pin processing loop
while (!done) {
// Keep the red LED on and read from the push button pin
mr = mraa_gpio_write(pin4, 1);
rv = mraa_gpio_read(pin13);
// TODO: MRAA error handling
if (rv == 0) {
// If the button is pressed then stop reading and go on with the program
syslog(LOG_DAEMON | LOG_NOTICE, "Button push detected.");
done = true;
}
}
// Turn off the red LED
mr = mraa_gpio_write(pin4, 0);
// TODO: MRAA error handling
// Open the source code file for the Python script
syslog(LOG_DAEMON | LOG_NOTICE, "Opening Python source code.");
FILE * code = fopen("/home/root/Robotd/robot.py", "r");
if (code == NULL) {
perror("FILE * code = fopen(\"/home/root/Robotd/robot.py\", \"r\")");
syslog(LOG_DAEMON | LOG_NOTICE, "Failed to load Python program.");
clean_up();
return EXIT_FAILURE;
}
// Return SIGINT handler to it's default state
sh = signal(SIGINT, SIG_DFL);
if (sh == SIG_ERR) {
perror("sh = signal(SIGINT, SIG_DLF)");
syslog(LOG_DAEMON | LOG_NOTICE, "Failed to disable signal handler.");
clean_up();
return EXIT_FAILURE;
}
// Execute the Python code
syslog(LOG_DAEMON | LOG_NOTICE, "Performing Python call.");
Py_SetProgramName(argv[0]);
Py_Initialize();
PyRun_SimpleFile(code, "/home/root/Robotd/robot.py");
Py_Finalize();
// TODO: Python error handling
// Close the source code file
fclose(code);
// Clean up and leave
syslog(LOG_DAEMON | LOG_NOTICE, "Finishing.");
closelog();
//execlp("shutdown", "shutdown", "-Ph", "now", (char *)NULL);
return EXIT_SUCCESS;
}