|
@@ -0,0 +1,237 @@
|
|
|
+
|
|
|
+ * 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;
|
|
|
+ socklen_t clientlen;
|
|
|
+
|
|
|
+
|
|
|
+ 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) {
|
|
|
+
|
|
|
+ pid = fork();
|
|
|
+ if (pid < 0)
|
|
|
+ die("Can't fork process");
|
|
|
+
|
|
|
+
|
|
|
+ if (pid > 0)
|
|
|
+ exit(EXIT_SUCCESS);
|
|
|
+
|
|
|
+
|
|
|
+ umask(0);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ lf = fopen(LOGFILE, "a");
|
|
|
+ if (lf == NULL)
|
|
|
+ die("Can't open logfile");
|
|
|
+
|
|
|
+ if (daemonize) {
|
|
|
+
|
|
|
+ if ((sid = setsid()) < 0);
|
|
|
+ die_log(lf, "Can't create a new session for child process");
|
|
|
+
|
|
|
+
|
|
|
+ if ((chdir("/")) < 0)
|
|
|
+ die_log(lf, "Can't change working directory to /");
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ 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(STDIN_FILENO);
|
|
|
+ close(STDOUT_FILENO);
|
|
|
+ close(STDERR_FILENO);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ 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");
|
|
|
+
|
|
|
+
|
|
|
+ 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");
|
|
|
+
|
|
|
+
|
|
|
+ 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);
|
|
|
+}
|