main.c

  1 // SPDX-License-Identifier: GPL-2.0+
  2 /*
  3  * Copyright (C) 2021 Felix Fietkau <nbd@nbd.name>
  4  */
  5 #include <sys/wait.h>
  6 #include <stdio.h>
  7 #include <unistd.h>
  8 #include <stdint.h>
  9 
 10 #include <libubox/uloop.h>
 11 
 12 #include "qosify.h"
 13 
 14 static int usage(const char *progname)
 15 {
 16         fprintf(stderr, "Usage: %s [options]\n"
 17                 "Options:\n"
 18                 "       -l <file>       Load defaults from <file>\n"
 19                 "       -o              only load program/maps without running as daemon\n"
 20                 "\n", progname);
 21 
 22         return 1;
 23 }
 24 
 25 int qosify_run_cmd(char *cmd, bool ignore_error)
 26 {
 27         char *argv[] = { "sh", "-c", cmd, NULL };
 28         bool first = true;
 29         int status = -1;
 30         char buf[512];
 31         int fds[2];
 32         FILE *f;
 33         int pid;
 34 
 35         if (pipe(fds))
 36                 return -1;
 37 
 38         pid = fork();
 39         if (!pid) {
 40                 close(fds[0]);
 41                 if (fds[1] != STDOUT_FILENO)
 42                         dup2(fds[1], STDOUT_FILENO);
 43                 if (fds[1] != STDERR_FILENO)
 44                         dup2(fds[1], STDERR_FILENO);
 45                 if (fds[1] > STDERR_FILENO)
 46                         close(fds[1]);
 47                 execv("/bin/sh", argv);
 48                 exit(1);
 49         }
 50 
 51         if (pid < 0)
 52                 return -1;
 53 
 54         close(fds[1]);
 55         f = fdopen(fds[0], "r");
 56         if (!f) {
 57                 close(fds[0]);
 58                 goto out;
 59         }
 60 
 61         while (fgets(buf, sizeof(buf), f) != NULL) {
 62                 if (!strlen(buf))
 63                         break;
 64                 if (ignore_error)
 65                         continue;
 66                 if (first) {
 67                         ULOG_WARN("Command: %s\n", cmd);
 68                         first = false;
 69                 }
 70                 ULOG_WARN("%s%s", buf, strchr(buf, '\n') ? "" : "\n");
 71         }
 72 
 73         fclose(f);
 74 
 75 out:
 76         while (waitpid(pid, &status, 0) < 0)
 77                 if (errno != EINTR)
 78                         break;
 79 
 80         return status;
 81 }
 82 
 83 
 84 int main(int argc, char **argv)
 85 {
 86         const char *load_file = NULL;
 87         bool oneshot = false;
 88         int ch;
 89 
 90         while ((ch = getopt(argc, argv, "fl:o")) != -1) {
 91                 switch (ch) {
 92                 case 'f':
 93                         break;
 94                 case 'l':
 95                         load_file = optarg;
 96                         break;
 97                 case 'o':
 98                         oneshot = true;
 99                         break;
100                 default:
101                         return usage(argv[0]);
102                 }
103         }
104 
105         if (qosify_loader_init())
106                 return 2;
107 
108         if (qosify_map_init())
109                 return 2;
110 
111         if (qosify_map_load_file(load_file))
112                 return 2;
113 
114         if (oneshot)
115                 return 0;
116 
117         ulog_open(ULOG_SYSLOG, LOG_DAEMON, "qosify");
118         uloop_init();
119 
120         if (qosify_ubus_init() ||
121             qosify_iface_init())
122                 return 2;
123 
124         qosify_dns_init();
125 
126         uloop_run();
127 
128         qosify_ubus_stop();
129         qosify_iface_stop();
130 
131         uloop_done();
132 
133         return 0;
134 }
135
This page was automatically generated by LXR 0.3.1. • OpenWrt
OpenWrt.org Cross Reference

Sources/qosify/main.c
