123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237 |
- /*
- * daemon for version information of embedded systems
- *
- * Copyright (C) 2010 Waldemar Brodkorb <wbx@openadk.org>
- *
- * This program is free software; you can redistribute it and/or
- * modify it under the terms of the GNU General Public License
- * as published by the Free Software Foundation; either version 2
- * of the License, or (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
- *
- */
- #include <arpa/inet.h>
- #include <errno.h>
- #include <fcntl.h>
- #include <getopt.h>
- #include <netinet/in.h>
- #include <signal.h>
- #include <stdio.h>
- #include <stdlib.h>
- #include <string.h>
- #include <sys/socket.h>
- #include <sys/stat.h>
- #include <sys/types.h>
- #include <time.h>
- #include <unistd.h>
- #define VERSION "0.1"
- #define LOGFILE "/tmp/uvd.log"
- #define LOCKFILE "/tmp/uvd.lock"
- #define BUFSIZE 1024
- #define VERFILE "/etc/adkversion"
- void version() {
- fprintf(stdout, "uvd version %s", VERSION);
- exit(EXIT_SUCCESS);
- }
- static const char *log_date(void) {
- static char buf[18];
- time_t t = time(NULL);
- strftime(buf, 18, "%b %d %H:%M:%S", localtime(&t));
- return buf;
- }
- void die(const char *msg) {
- fprintf(stderr, "%s\n", msg);
- exit(EXIT_FAILURE);
- }
- void die_log(FILE *lf, const char *msg) {
- fprintf(lf, "%s: %s\n", log_date(), msg);
- exit(EXIT_FAILURE);
- }
- void log_msg(FILE *lf, const char *msg) {
- fprintf(lf, "%s: %s\n", log_date(), msg);
- fflush(lf);
- }
- void signal_handler(int sig) {
- switch(sig) {
- case SIGHUP:
- break;
- case SIGTERM:
- case SIGINT:
- case SIGQUIT:
- unlink(LOCKFILE);
- exit(EXIT_SUCCESS);
- break;
- default:
- break;
- }
- }
- void usage(int argc, char *argv[]) {
- if (argc >=1) {
- fprintf(stdout, "Usage: %s [ --help | --debug | --version ]\n", argv[0]);
- fprintf(stdout, " Options:\n");
- fprintf(stdout, " --debug | -d start in debug mode, don't fork\n");
- fprintf(stdout, " --version | -v show version\n");
- fprintf(stdout, " --help | -h show help\n");
- fprintf(stdout, "\n");
- }
- exit(EXIT_SUCCESS);
- }
- int main(int argc, char **argv) {
- pid_t pid, sid;
- int lfd, c, ss, n, res;
- FILE *lf = NULL;
- int vf;
- struct sigaction sigact;
- int daemonize = 1;
- int optval = 1;
- char buf[BUFSIZE];
- struct sockaddr_in server = { 0 };
- struct sockaddr_in clientaddr; /* client addr */
- socklen_t clientlen;
- /* options descriptor */
- static struct option longopts[] = {
- { "debug", no_argument, 0, 'd' },
- { "help", no_argument, 0, 'h' },
- { "version", no_argument, 0, 'v' },
- { NULL, 0, NULL, 0 }
- };
- while((c = getopt_long(argc, argv, "dhv", longopts, NULL)) != -1) {
- switch(c) {
- case 'h':
- usage(argc, argv);
- break;
- case 'v':
- version();
- break;
- case 'd':
- daemonize = 0;
- break;
- default:
- usage(argc, argv);
- }
- }
- if (daemonize) {
- /* Fork off the parent process */
- pid = fork();
- if (pid < 0)
- die("Can't fork process");
- /* If we got a pid, we exit the parent process */
- if (pid > 0)
- exit(EXIT_SUCCESS);
- /* Change the file mode mask */
- umask(0);
- }
- /* Open a logfile */
- lf = fopen(LOGFILE, "a");
- if (lf == NULL)
- die("Can't open logfile");
-
- if (daemonize) {
- /* Create a new session for child process */
- if ((sid = setsid()) < 0)
- die_log(lf, "Can't create a new session for child process");
- /* Change the current working directory */
- if ((chdir("/")) < 0)
- die_log(lf, "Can't change working directory to /");
- }
- /* check if daemon already running */
- if ((lfd = open(LOCKFILE, O_RDWR | O_CREAT, 0640) < 0))
- die("Can't open or create lock file");
- if (lockf(lfd, F_TLOCK, 0) == -1)
- die("uvd already running!");
- if (daemonize) {
- /* Close out the standard file descriptors */
- close(STDIN_FILENO);
- close(STDOUT_FILENO);
- close(STDERR_FILENO);
- }
- /* Handle some signals */
- sigact.sa_handler = signal_handler;
- sigact.sa_flags = 0;
- sigaction(SIGHUP, &sigact, NULL);
- sigaction(SIGTERM, &sigact, NULL);
- sigaction(SIGINT, &sigact, NULL);
- sigaction(SIGQUIT, &sigact, NULL);
- log_msg(lf, "uvd started successfully");
- /* create network socket */
- if ((ss = socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP)) == -1)
- die("can not create socket");
- server.sin_addr.s_addr=htonl(INADDR_ANY);
- server.sin_family = AF_INET;
- server.sin_port = htons(4242);
- if (setsockopt (ss, SOL_SOCKET, SO_BROADCAST, (caddr_t) &optval, sizeof (optval)) < 0)
- die("can not set socket option");
- if (setsockopt (ss, SOL_SOCKET, SO_REUSEADDR, (caddr_t) &optval, sizeof (optval)) < 0)
- die("can not set socket option");
- if (bind(ss, (struct sockaddr *) &server, sizeof(server)) < 0)
- die("can not bind socket");
- /* loop forever */
- while(1) {
- clientlen = sizeof(clientaddr);
- n = recvfrom(ss, buf, BUFSIZE , 0, (struct sockaddr *)&clientaddr, &clientlen);
- if (n < 0)
- die_log(lf, "error reading from client");
- buf[n] = 0;
- res = strncmp(buf, "version", n);
- if (res > 0) {
- log_msg(lf, "been asked for version information");
- if ((vf = open(VERFILE, O_RDONLY)) < 0)
- die_log(lf, "unable to open version file");
- ssize_t num;
- do {
- num = read(vf, &buf, sizeof(buf));
- buf[num] = '\0';
- if (sendto(ss, buf, num, 0, (struct sockaddr *) &clientaddr, sizeof(clientaddr)) < 0)
- die_log(lf, "can not send data");
- } while (num > 0);
- close(vf);
- }
- }
- close(ss);
- return(0);
- }
|