13054f218SWarner Losh /*- 2b8f92ce4SWarner Losh * Copyright (c) 2002-2010 M. Warner Losh. 33054f218SWarner Losh * All rights reserved. 43054f218SWarner Losh * 53054f218SWarner Losh * Redistribution and use in source and binary forms, with or without 63054f218SWarner Losh * modification, are permitted provided that the following conditions 73054f218SWarner Losh * are met: 83054f218SWarner Losh * 1. Redistributions of source code must retain the above copyright 93054f218SWarner Losh * notice, this list of conditions and the following disclaimer. 103054f218SWarner Losh * 2. Redistributions in binary form must reproduce the above copyright 113054f218SWarner Losh * notice, this list of conditions and the following disclaimer in the 123054f218SWarner Losh * documentation and/or other materials provided with the distribution. 133054f218SWarner Losh * 143054f218SWarner Losh * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 153054f218SWarner Losh * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 163054f218SWarner Losh * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 173054f218SWarner Losh * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 183054f218SWarner Losh * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 193054f218SWarner Losh * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 203054f218SWarner Losh * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 213054f218SWarner Losh * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 223054f218SWarner Losh * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 233054f218SWarner Losh * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 243054f218SWarner Losh * SUCH DAMAGE. 25b8f92ce4SWarner Losh * 26b8f92ce4SWarner Losh * my_system is a variation on lib/libc/stdlib/system.c: 27b8f92ce4SWarner Losh * 28b8f92ce4SWarner Losh * Copyright (c) 1988, 1993 29b8f92ce4SWarner Losh * The Regents of the University of California. All rights reserved. 30b8f92ce4SWarner Losh * 31b8f92ce4SWarner Losh * Redistribution and use in source and binary forms, with or without 32b8f92ce4SWarner Losh * modification, are permitted provided that the following conditions 33b8f92ce4SWarner Losh * are met: 34b8f92ce4SWarner Losh * 1. Redistributions of source code must retain the above copyright 35b8f92ce4SWarner Losh * notice, this list of conditions and the following disclaimer. 36b8f92ce4SWarner Losh * 2. Redistributions in binary form must reproduce the above copyright 37b8f92ce4SWarner Losh * notice, this list of conditions and the following disclaimer in the 38b8f92ce4SWarner Losh * documentation and/or other materials provided with the distribution. 39b8f92ce4SWarner Losh * 4. Neither the name of the University nor the names of its contributors 40b8f92ce4SWarner Losh * may be used to endorse or promote products derived from this software 41b8f92ce4SWarner Losh * without specific prior written permission. 42b8f92ce4SWarner Losh * 43b8f92ce4SWarner Losh * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 44b8f92ce4SWarner Losh * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 45b8f92ce4SWarner Losh * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 46b8f92ce4SWarner Losh * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 47b8f92ce4SWarner Losh * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 48b8f92ce4SWarner Losh * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 49b8f92ce4SWarner Losh * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 50b8f92ce4SWarner Losh * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 51b8f92ce4SWarner Losh * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 52b8f92ce4SWarner Losh * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 53b8f92ce4SWarner Losh * SUCH DAMAGE. 543054f218SWarner Losh */ 553054f218SWarner Losh 563054f218SWarner Losh /* 573054f218SWarner Losh * DEVD control daemon. 583054f218SWarner Losh */ 593054f218SWarner Losh 603054f218SWarner Losh // TODO list: 613054f218SWarner Losh // o devd.conf and devd man pages need a lot of help: 62251bedd1SWarner Losh // - devd needs to document the unix domain socket 633054f218SWarner Losh // - devd.conf needs more details on the supported statements. 643054f218SWarner Losh 653054f218SWarner Losh #include <sys/cdefs.h> 663054f218SWarner Losh __FBSDID("$FreeBSD$"); 673054f218SWarner Losh 683054f218SWarner Losh #include <sys/param.h> 69251bedd1SWarner Losh #include <sys/socket.h> 70251bedd1SWarner Losh #include <sys/stat.h> 719c180898SWarner Losh #include <sys/sysctl.h> 72251bedd1SWarner Losh #include <sys/types.h> 73b8f92ce4SWarner Losh #include <sys/wait.h> 74251bedd1SWarner Losh #include <sys/un.h> 753054f218SWarner Losh 76*d005340fSEitan Adler #include <cctype> 77*d005340fSEitan Adler #include <cerrno> 78*d005340fSEitan Adler #include <cstdlib> 79*d005340fSEitan Adler #include <cstdio> 80*d005340fSEitan Adler #include <csignal> 81*d005340fSEitan Adler #include <cstring> 82*d005340fSEitan Adler 833054f218SWarner Losh #include <dirent.h> 843054f218SWarner Losh #include <err.h> 853054f218SWarner Losh #include <fcntl.h> 861a0cc6b1SPawel Jakub Dawidek #include <libutil.h> 87b8f92ce4SWarner Losh #include <paths.h> 88e1334f93SIan Lepore #include <poll.h> 895102ef84SWarner Losh #include <regex.h> 903054f218SWarner Losh #include <unistd.h> 913054f218SWarner Losh 928b823408SWarner Losh #include <algorithm> 933054f218SWarner Losh #include <map> 943054f218SWarner Losh #include <string> 95251bedd1SWarner Losh #include <list> 963054f218SWarner Losh #include <vector> 973054f218SWarner Losh 986aeeca8eSWarner Losh #include "devd.h" /* C compatible definitions */ 996aeeca8eSWarner Losh #include "devd.hh" /* C++ class definitions */ 1003054f218SWarner Losh 101251bedd1SWarner Losh #define PIPE "/var/run/devd.pipe" 1023054f218SWarner Losh #define CF "/etc/devd.conf" 1039c180898SWarner Losh #define SYSCTL "hw.bus.devctl_disable" 1043054f218SWarner Losh 1053054f218SWarner Losh using namespace std; 1063054f218SWarner Losh 1073054f218SWarner Losh extern FILE *yyin; 1083054f218SWarner Losh extern int lineno; 1093054f218SWarner Losh 110842ccec5SWarner Losh static const char notify = '!'; 1118b823408SWarner Losh static const char nomatch = '?'; 1128b823408SWarner Losh static const char attach = '+'; 1138b823408SWarner Losh static const char detach = '-'; 1148b823408SWarner Losh 1151a0cc6b1SPawel Jakub Dawidek static struct pidfh *pfh; 1161a0cc6b1SPawel Jakub Dawidek 1175a882d77SWarner Losh int Dflag; 1183054f218SWarner Losh int dflag; 119812f53a7SWarner Losh int nflag; 120fcdcaa88SEitan Adler static volatile sig_atomic_t romeo_must_die = 0; 1213054f218SWarner Losh 1228334958aSJoseph Koshy static const char *configfile = CF; 1238334958aSJoseph Koshy 1243054f218SWarner Losh static void event_loop(void); 1253054f218SWarner Losh static void usage(void); 1263054f218SWarner Losh 1278b823408SWarner Losh template <class T> void 1288b823408SWarner Losh delete_and_clear(vector<T *> &v) 1298b823408SWarner Losh { 1308b823408SWarner Losh typename vector<T *>::const_iterator i; 1313054f218SWarner Losh 132edee691dSEitan Adler for (i = v.begin(); i != v.end(); ++i) 1338b823408SWarner Losh delete *i; 1348b823408SWarner Losh v.clear(); 1358b823408SWarner Losh } 1368b823408SWarner Losh 1373054f218SWarner Losh config cfg; 1383054f218SWarner Losh 1393054f218SWarner Losh event_proc::event_proc() : _prio(-1) 1403054f218SWarner Losh { 1415dfc0f6cSIan Lepore _epsvec.reserve(4); 1423054f218SWarner Losh } 1433054f218SWarner Losh 1443054f218SWarner Losh event_proc::~event_proc() 1453054f218SWarner Losh { 14613f8bb71SAlexander Nedotsukov delete_and_clear(_epsvec); 1473054f218SWarner Losh } 1483054f218SWarner Losh 1493054f218SWarner Losh void 1503054f218SWarner Losh event_proc::add(eps *eps) 1513054f218SWarner Losh { 1523054f218SWarner Losh _epsvec.push_back(eps); 1533054f218SWarner Losh } 1543054f218SWarner Losh 1553054f218SWarner Losh bool 156ef370346SEitan Adler event_proc::matches(config &c) const 1573054f218SWarner Losh { 1583054f218SWarner Losh vector<eps *>::const_iterator i; 1593054f218SWarner Losh 160edee691dSEitan Adler for (i = _epsvec.begin(); i != _epsvec.end(); ++i) 1613054f218SWarner Losh if (!(*i)->do_match(c)) 1623054f218SWarner Losh return (false); 1633054f218SWarner Losh return (true); 1643054f218SWarner Losh } 1653054f218SWarner Losh 1663054f218SWarner Losh bool 167ef370346SEitan Adler event_proc::run(config &c) const 1683054f218SWarner Losh { 1693054f218SWarner Losh vector<eps *>::const_iterator i; 1703054f218SWarner Losh 171edee691dSEitan Adler for (i = _epsvec.begin(); i != _epsvec.end(); ++i) 1723054f218SWarner Losh if (!(*i)->do_action(c)) 1733054f218SWarner Losh return (false); 1743054f218SWarner Losh return (true); 1753054f218SWarner Losh } 1763054f218SWarner Losh 1773054f218SWarner Losh action::action(const char *cmd) 1783054f218SWarner Losh : _cmd(cmd) 1793054f218SWarner Losh { 1803054f218SWarner Losh // nothing 1813054f218SWarner Losh } 1823054f218SWarner Losh 1833054f218SWarner Losh action::~action() 1843054f218SWarner Losh { 1853054f218SWarner Losh // nothing 1863054f218SWarner Losh } 1873054f218SWarner Losh 188b8f92ce4SWarner Losh static int 189b8f92ce4SWarner Losh my_system(const char *command) 190b8f92ce4SWarner Losh { 191b8f92ce4SWarner Losh pid_t pid, savedpid; 192b8f92ce4SWarner Losh int pstat; 193b8f92ce4SWarner Losh struct sigaction ign, intact, quitact; 194b8f92ce4SWarner Losh sigset_t newsigblock, oldsigblock; 195b8f92ce4SWarner Losh 196b8f92ce4SWarner Losh if (!command) /* just checking... */ 197b8f92ce4SWarner Losh return(1); 198b8f92ce4SWarner Losh 199b8f92ce4SWarner Losh /* 200b8f92ce4SWarner Losh * Ignore SIGINT and SIGQUIT, block SIGCHLD. Remember to save 201b8f92ce4SWarner Losh * existing signal dispositions. 202b8f92ce4SWarner Losh */ 203b8f92ce4SWarner Losh ign.sa_handler = SIG_IGN; 204b8f92ce4SWarner Losh ::sigemptyset(&ign.sa_mask); 205b8f92ce4SWarner Losh ign.sa_flags = 0; 206b8f92ce4SWarner Losh ::sigaction(SIGINT, &ign, &intact); 207b8f92ce4SWarner Losh ::sigaction(SIGQUIT, &ign, &quitact); 208b8f92ce4SWarner Losh ::sigemptyset(&newsigblock); 209b8f92ce4SWarner Losh ::sigaddset(&newsigblock, SIGCHLD); 210b8f92ce4SWarner Losh ::sigprocmask(SIG_BLOCK, &newsigblock, &oldsigblock); 211b8f92ce4SWarner Losh switch (pid = ::fork()) { 212b8f92ce4SWarner Losh case -1: /* error */ 213b8f92ce4SWarner Losh break; 214b8f92ce4SWarner Losh case 0: /* child */ 215b8f92ce4SWarner Losh /* 216b8f92ce4SWarner Losh * Restore original signal dispositions and exec the command. 217b8f92ce4SWarner Losh */ 218b8f92ce4SWarner Losh ::sigaction(SIGINT, &intact, NULL); 219b8f92ce4SWarner Losh ::sigaction(SIGQUIT, &quitact, NULL); 220b8f92ce4SWarner Losh ::sigprocmask(SIG_SETMASK, &oldsigblock, NULL); 221b8f92ce4SWarner Losh /* 222b8f92ce4SWarner Losh * Close the PID file, and all other open descriptors. 223b8f92ce4SWarner Losh * Inherit std{in,out,err} only. 224b8f92ce4SWarner Losh */ 225b8f92ce4SWarner Losh cfg.close_pidfile(); 226b8f92ce4SWarner Losh ::closefrom(3); 227b8f92ce4SWarner Losh ::execl(_PATH_BSHELL, "sh", "-c", command, (char *)NULL); 228b8f92ce4SWarner Losh ::_exit(127); 229b8f92ce4SWarner Losh default: /* parent */ 230b8f92ce4SWarner Losh savedpid = pid; 231b8f92ce4SWarner Losh do { 232b8f92ce4SWarner Losh pid = ::wait4(savedpid, &pstat, 0, (struct rusage *)0); 233b8f92ce4SWarner Losh } while (pid == -1 && errno == EINTR); 234b8f92ce4SWarner Losh break; 235b8f92ce4SWarner Losh } 236b8f92ce4SWarner Losh ::sigaction(SIGINT, &intact, NULL); 237b8f92ce4SWarner Losh ::sigaction(SIGQUIT, &quitact, NULL); 238b8f92ce4SWarner Losh ::sigprocmask(SIG_SETMASK, &oldsigblock, NULL); 239b8f92ce4SWarner Losh return (pid == -1 ? -1 : pstat); 240b8f92ce4SWarner Losh } 241b8f92ce4SWarner Losh 2423054f218SWarner Losh bool 2435102ef84SWarner Losh action::do_action(config &c) 2443054f218SWarner Losh { 2455dfc0f6cSIan Lepore string s = c.expand_string(_cmd.c_str()); 2465a882d77SWarner Losh if (Dflag) 2478b823408SWarner Losh fprintf(stderr, "Executing '%s'\n", s.c_str()); 248b8f92ce4SWarner Losh my_system(s.c_str()); 2493054f218SWarner Losh return (true); 2503054f218SWarner Losh } 2513054f218SWarner Losh 2525dfc0f6cSIan Lepore match::match(config &c, const char *var, const char *re) : 2535dfc0f6cSIan Lepore _inv(re[0] == '!'), 2545dfc0f6cSIan Lepore _var(var), 2555dfc0f6cSIan Lepore _re(c.expand_string(_inv ? re + 1 : re, "^", "$")) 2563054f218SWarner Losh { 2579cb4a7bdSWarner Losh regcomp(&_regex, _re.c_str(), REG_EXTENDED | REG_NOSUB | REG_ICASE); 2583054f218SWarner Losh } 2593054f218SWarner Losh 2603054f218SWarner Losh match::~match() 2613054f218SWarner Losh { 2625102ef84SWarner Losh regfree(&_regex); 2633054f218SWarner Losh } 2643054f218SWarner Losh 2653054f218SWarner Losh bool 2665102ef84SWarner Losh match::do_match(config &c) 2673054f218SWarner Losh { 2685bf08422SUlf Lilleengen const string &value = c.get_variable(_var); 2695102ef84SWarner Losh bool retval; 2705102ef84SWarner Losh 2715a882d77SWarner Losh if (Dflag) 2720321b694SHiroki Sato fprintf(stderr, "Testing %s=%s against %s, invert=%d\n", 2730321b694SHiroki Sato _var.c_str(), value.c_str(), _re.c_str(), _inv); 2748b823408SWarner Losh 2755102ef84SWarner Losh retval = (regexec(&_regex, value.c_str(), 0, NULL, 0) == 0); 2760321b694SHiroki Sato if (_inv == 1) 2770321b694SHiroki Sato retval = (retval == 0) ? 1 : 0; 2780321b694SHiroki Sato 2795102ef84SWarner Losh return retval; 2803054f218SWarner Losh } 2813054f218SWarner Losh 282cd70782bSWarner Losh #include <sys/sockio.h> 283cd70782bSWarner Losh #include <net/if.h> 284cd70782bSWarner Losh #include <net/if_media.h> 285cd70782bSWarner Losh 28694d2d4ebSBrooks Davis media::media(config &, const char *var, const char *type) 287cd70782bSWarner Losh : _var(var), _type(-1) 288cd70782bSWarner Losh { 289cd70782bSWarner Losh static struct ifmedia_description media_types[] = { 290cd70782bSWarner Losh { IFM_ETHER, "Ethernet" }, 291cd70782bSWarner Losh { IFM_TOKEN, "Tokenring" }, 292cd70782bSWarner Losh { IFM_FDDI, "FDDI" }, 293cd70782bSWarner Losh { IFM_IEEE80211, "802.11" }, 294cd70782bSWarner Losh { IFM_ATM, "ATM" }, 295cd70782bSWarner Losh { -1, "unknown" }, 296cd70782bSWarner Losh { 0, NULL }, 297cd70782bSWarner Losh }; 298edee691dSEitan Adler for (int i = 0; media_types[i].ifmt_string != NULL; ++i) 299cd70782bSWarner Losh if (strcasecmp(type, media_types[i].ifmt_string) == 0) { 300cd70782bSWarner Losh _type = media_types[i].ifmt_word; 301cd70782bSWarner Losh break; 302cd70782bSWarner Losh } 303cd70782bSWarner Losh } 304cd70782bSWarner Losh 305cd70782bSWarner Losh media::~media() 306cd70782bSWarner Losh { 307cd70782bSWarner Losh } 308cd70782bSWarner Losh 309cd70782bSWarner Losh bool 310cd70782bSWarner Losh media::do_match(config &c) 311cd70782bSWarner Losh { 31209c47e76SWarner Losh string value; 313cd70782bSWarner Losh struct ifmediareq ifmr; 314cd70782bSWarner Losh bool retval; 315cd70782bSWarner Losh int s; 316cd70782bSWarner Losh 31709c47e76SWarner Losh // Since we can be called from both a device attach/detach 31809c47e76SWarner Losh // context where device-name is defined and what we want, 31909c47e76SWarner Losh // as well as from a link status context, where subsystem is 32009c47e76SWarner Losh // the name of interest, first try device-name and fall back 32109c47e76SWarner Losh // to subsystem if none exists. 32209c47e76SWarner Losh value = c.get_variable("device-name"); 323a6393aa1SEitan Adler if (value.empty()) 324de02eec8SWarner Losh value = c.get_variable("subsystem"); 325cd70782bSWarner Losh if (Dflag) 326cd70782bSWarner Losh fprintf(stderr, "Testing media type of %s against 0x%x\n", 327cd70782bSWarner Losh value.c_str(), _type); 328cd70782bSWarner Losh 329cd70782bSWarner Losh retval = false; 330cd70782bSWarner Losh 331cd70782bSWarner Losh s = socket(PF_INET, SOCK_DGRAM, 0); 332cd70782bSWarner Losh if (s >= 0) { 333cd70782bSWarner Losh memset(&ifmr, 0, sizeof(ifmr)); 334cd70782bSWarner Losh strncpy(ifmr.ifm_name, value.c_str(), sizeof(ifmr.ifm_name)); 335cd70782bSWarner Losh 336cd70782bSWarner Losh if (ioctl(s, SIOCGIFMEDIA, (caddr_t)&ifmr) >= 0 && 337cd70782bSWarner Losh ifmr.ifm_status & IFM_AVALID) { 338cd70782bSWarner Losh if (Dflag) 339cd70782bSWarner Losh fprintf(stderr, "%s has media type 0x%x\n", 340cd70782bSWarner Losh value.c_str(), IFM_TYPE(ifmr.ifm_active)); 341cd70782bSWarner Losh retval = (IFM_TYPE(ifmr.ifm_active) == _type); 342cd70782bSWarner Losh } else if (_type == -1) { 343cd70782bSWarner Losh if (Dflag) 344cd70782bSWarner Losh fprintf(stderr, "%s has unknown media type\n", 345cd70782bSWarner Losh value.c_str()); 346cd70782bSWarner Losh retval = true; 347cd70782bSWarner Losh } 348cd70782bSWarner Losh close(s); 349cd70782bSWarner Losh } 350cd70782bSWarner Losh 351cd70782bSWarner Losh return retval; 352cd70782bSWarner Losh } 353cd70782bSWarner Losh 3543054f218SWarner Losh const string var_list::bogus = "_$_$_$_$_B_O_G_U_S_$_$_$_$_"; 3553054f218SWarner Losh const string var_list::nothing = ""; 3563054f218SWarner Losh 3573054f218SWarner Losh const string & 3583054f218SWarner Losh var_list::get_variable(const string &var) const 3593054f218SWarner Losh { 3603054f218SWarner Losh map<string, string>::const_iterator i; 3613054f218SWarner Losh 3623054f218SWarner Losh i = _vars.find(var); 3633054f218SWarner Losh if (i == _vars.end()) 3648b823408SWarner Losh return (var_list::bogus); 3653054f218SWarner Losh return (i->second); 3663054f218SWarner Losh } 3673054f218SWarner Losh 3683054f218SWarner Losh bool 3693054f218SWarner Losh var_list::is_set(const string &var) const 3703054f218SWarner Losh { 3713054f218SWarner Losh return (_vars.find(var) != _vars.end()); 3723054f218SWarner Losh } 3733054f218SWarner Losh 3743054f218SWarner Losh void 3753054f218SWarner Losh var_list::set_variable(const string &var, const string &val) 3763054f218SWarner Losh { 3775a882d77SWarner Losh if (Dflag) 378528d3f79SWarner Losh fprintf(stderr, "setting %s=%s\n", var.c_str(), val.c_str()); 3793054f218SWarner Losh _vars[var] = val; 3803054f218SWarner Losh } 3813054f218SWarner Losh 3823054f218SWarner Losh void 3833054f218SWarner Losh config::reset(void) 3843054f218SWarner Losh { 3853054f218SWarner Losh _dir_list.clear(); 3868b823408SWarner Losh delete_and_clear(_var_list_table); 3878b823408SWarner Losh delete_and_clear(_attach_list); 3888b823408SWarner Losh delete_and_clear(_detach_list); 3898b823408SWarner Losh delete_and_clear(_nomatch_list); 390842ccec5SWarner Losh delete_and_clear(_notify_list); 3913054f218SWarner Losh } 3923054f218SWarner Losh 3933054f218SWarner Losh void 3943054f218SWarner Losh config::parse_one_file(const char *fn) 3953054f218SWarner Losh { 3965a882d77SWarner Losh if (Dflag) 3970f3311c5SPoul-Henning Kamp fprintf(stderr, "Parsing %s\n", fn); 3983054f218SWarner Losh yyin = fopen(fn, "r"); 3993054f218SWarner Losh if (yyin == NULL) 4003054f218SWarner Losh err(1, "Cannot open config file %s", fn); 4014a3050fcSMaxim Konovalov lineno = 1; 4023054f218SWarner Losh if (yyparse() != 0) 4033054f218SWarner Losh errx(1, "Cannot parse %s at line %d", fn, lineno); 4043054f218SWarner Losh fclose(yyin); 4053054f218SWarner Losh } 4063054f218SWarner Losh 4073054f218SWarner Losh void 4083054f218SWarner Losh config::parse_files_in_dir(const char *dirname) 4093054f218SWarner Losh { 4103054f218SWarner Losh DIR *dirp; 4113054f218SWarner Losh struct dirent *dp; 4123054f218SWarner Losh char path[PATH_MAX]; 4133054f218SWarner Losh 4145a882d77SWarner Losh if (Dflag) 4150f3311c5SPoul-Henning Kamp fprintf(stderr, "Parsing files in %s\n", dirname); 4163054f218SWarner Losh dirp = opendir(dirname); 4173054f218SWarner Losh if (dirp == NULL) 4183054f218SWarner Losh return; 4193054f218SWarner Losh readdir(dirp); /* Skip . */ 4203054f218SWarner Losh readdir(dirp); /* Skip .. */ 4213054f218SWarner Losh while ((dp = readdir(dirp)) != NULL) { 4223054f218SWarner Losh if (strcmp(dp->d_name + dp->d_namlen - 5, ".conf") == 0) { 4233054f218SWarner Losh snprintf(path, sizeof(path), "%s/%s", 4243054f218SWarner Losh dirname, dp->d_name); 4253054f218SWarner Losh parse_one_file(path); 4263054f218SWarner Losh } 4273054f218SWarner Losh } 4285983e891SKevin Lo closedir(dirp); 4293054f218SWarner Losh } 4303054f218SWarner Losh 4318b823408SWarner Losh class epv_greater { 4328b823408SWarner Losh public: 433ef370346SEitan Adler int operator()(event_proc *const&l1, event_proc *const&l2) const 4348b823408SWarner Losh { 4358b823408SWarner Losh return (l1->get_priority() > l2->get_priority()); 4368b823408SWarner Losh } 4378b823408SWarner Losh }; 4388b823408SWarner Losh 4398b823408SWarner Losh void 4408b823408SWarner Losh config::sort_vector(vector<event_proc *> &v) 4418b823408SWarner Losh { 4420c04d185SDimitry Andric stable_sort(v.begin(), v.end(), epv_greater()); 4438b823408SWarner Losh } 4448b823408SWarner Losh 4453054f218SWarner Losh void 4463054f218SWarner Losh config::parse(void) 4473054f218SWarner Losh { 4483054f218SWarner Losh vector<string>::const_iterator i; 4493054f218SWarner Losh 4508334958aSJoseph Koshy parse_one_file(configfile); 451edee691dSEitan Adler for (i = _dir_list.begin(); i != _dir_list.end(); ++i) 4523054f218SWarner Losh parse_files_in_dir((*i).c_str()); 4538b823408SWarner Losh sort_vector(_attach_list); 4548b823408SWarner Losh sort_vector(_detach_list); 4558b823408SWarner Losh sort_vector(_nomatch_list); 456842ccec5SWarner Losh sort_vector(_notify_list); 4573054f218SWarner Losh } 4583054f218SWarner Losh 4593054f218SWarner Losh void 4601a0cc6b1SPawel Jakub Dawidek config::open_pidfile() 4613054f218SWarner Losh { 4621a0cc6b1SPawel Jakub Dawidek pid_t otherpid; 4633054f218SWarner Losh 4645a3b1a3dSEitan Adler if (_pidfile.empty()) 4653054f218SWarner Losh return; 4661a0cc6b1SPawel Jakub Dawidek pfh = pidfile_open(_pidfile.c_str(), 0600, &otherpid); 4671a0cc6b1SPawel Jakub Dawidek if (pfh == NULL) { 4681a0cc6b1SPawel Jakub Dawidek if (errno == EEXIST) 4691a0cc6b1SPawel Jakub Dawidek errx(1, "devd already running, pid: %d", (int)otherpid); 4701a0cc6b1SPawel Jakub Dawidek warn("cannot open pid file"); 4711a0cc6b1SPawel Jakub Dawidek } 4721a0cc6b1SPawel Jakub Dawidek } 4731a0cc6b1SPawel Jakub Dawidek 4741a0cc6b1SPawel Jakub Dawidek void 4751a0cc6b1SPawel Jakub Dawidek config::write_pidfile() 4761a0cc6b1SPawel Jakub Dawidek { 4771a0cc6b1SPawel Jakub Dawidek 4781a0cc6b1SPawel Jakub Dawidek pidfile_write(pfh); 4791a0cc6b1SPawel Jakub Dawidek } 4801a0cc6b1SPawel Jakub Dawidek 4811a0cc6b1SPawel Jakub Dawidek void 482b8f92ce4SWarner Losh config::close_pidfile() 483b8f92ce4SWarner Losh { 484b8f92ce4SWarner Losh 485b8f92ce4SWarner Losh pidfile_close(pfh); 486b8f92ce4SWarner Losh } 487b8f92ce4SWarner Losh 488b8f92ce4SWarner Losh void 4891a0cc6b1SPawel Jakub Dawidek config::remove_pidfile() 4901a0cc6b1SPawel Jakub Dawidek { 4911a0cc6b1SPawel Jakub Dawidek 4921a0cc6b1SPawel Jakub Dawidek pidfile_remove(pfh); 4933054f218SWarner Losh } 4943054f218SWarner Losh 4953054f218SWarner Losh void 4963054f218SWarner Losh config::add_attach(int prio, event_proc *p) 4973054f218SWarner Losh { 4983054f218SWarner Losh p->set_priority(prio); 4993054f218SWarner Losh _attach_list.push_back(p); 5003054f218SWarner Losh } 5013054f218SWarner Losh 5023054f218SWarner Losh void 5033054f218SWarner Losh config::add_detach(int prio, event_proc *p) 5043054f218SWarner Losh { 5053054f218SWarner Losh p->set_priority(prio); 5063054f218SWarner Losh _detach_list.push_back(p); 5073054f218SWarner Losh } 5083054f218SWarner Losh 5093054f218SWarner Losh void 5103054f218SWarner Losh config::add_directory(const char *dir) 5113054f218SWarner Losh { 5123054f218SWarner Losh _dir_list.push_back(string(dir)); 5133054f218SWarner Losh } 5143054f218SWarner Losh 5153054f218SWarner Losh void 5163054f218SWarner Losh config::add_nomatch(int prio, event_proc *p) 5173054f218SWarner Losh { 5183054f218SWarner Losh p->set_priority(prio); 5193054f218SWarner Losh _nomatch_list.push_back(p); 5203054f218SWarner Losh } 5213054f218SWarner Losh 5223054f218SWarner Losh void 523842ccec5SWarner Losh config::add_notify(int prio, event_proc *p) 524842ccec5SWarner Losh { 525842ccec5SWarner Losh p->set_priority(prio); 526842ccec5SWarner Losh _notify_list.push_back(p); 527842ccec5SWarner Losh } 528842ccec5SWarner Losh 529842ccec5SWarner Losh void 5303054f218SWarner Losh config::set_pidfile(const char *fn) 5313054f218SWarner Losh { 5325a3b1a3dSEitan Adler _pidfile = fn; 5333054f218SWarner Losh } 5343054f218SWarner Losh 5353054f218SWarner Losh void 5363054f218SWarner Losh config::push_var_table() 5373054f218SWarner Losh { 5383054f218SWarner Losh var_list *vl; 5393054f218SWarner Losh 5403054f218SWarner Losh vl = new var_list(); 5413054f218SWarner Losh _var_list_table.push_back(vl); 5425a882d77SWarner Losh if (Dflag) 5438b823408SWarner Losh fprintf(stderr, "Pushing table\n"); 5443054f218SWarner Losh } 5453054f218SWarner Losh 5463054f218SWarner Losh void 5473054f218SWarner Losh config::pop_var_table() 5483054f218SWarner Losh { 5493054f218SWarner Losh delete _var_list_table.back(); 5503054f218SWarner Losh _var_list_table.pop_back(); 5515a882d77SWarner Losh if (Dflag) 5528b823408SWarner Losh fprintf(stderr, "Popping table\n"); 5533054f218SWarner Losh } 5543054f218SWarner Losh 5553054f218SWarner Losh void 5563054f218SWarner Losh config::set_variable(const char *var, const char *val) 5573054f218SWarner Losh { 5583054f218SWarner Losh _var_list_table.back()->set_variable(var, val); 5593054f218SWarner Losh } 5603054f218SWarner Losh 5613054f218SWarner Losh const string & 5623054f218SWarner Losh config::get_variable(const string &var) 5633054f218SWarner Losh { 5643054f218SWarner Losh vector<var_list *>::reverse_iterator i; 5653054f218SWarner Losh 566edee691dSEitan Adler for (i = _var_list_table.rbegin(); i != _var_list_table.rend(); ++i) { 5673054f218SWarner Losh if ((*i)->is_set(var)) 5688b823408SWarner Losh return ((*i)->get_variable(var)); 5693054f218SWarner Losh } 5703054f218SWarner Losh return (var_list::nothing); 5713054f218SWarner Losh } 5723054f218SWarner Losh 5738b823408SWarner Losh bool 574ef370346SEitan Adler config::is_id_char(char ch) const 5758b823408SWarner Losh { 5768b823408SWarner Losh return (ch != '\0' && (isalpha(ch) || isdigit(ch) || ch == '_' || 5778b823408SWarner Losh ch == '-')); 5788b823408SWarner Losh } 5798b823408SWarner Losh 5805102ef84SWarner Losh void 5816cabd675SWarner Losh config::expand_one(const char *&src, string &dst) 5823054f218SWarner Losh { 5835102ef84SWarner Losh int count; 5845bf08422SUlf Lilleengen string buffer; 5855102ef84SWarner Losh 5868b823408SWarner Losh src++; 5875102ef84SWarner Losh // $$ -> $ 5885102ef84SWarner Losh if (*src == '$') { 5897d9e9c60SEitan Adler dst += *src++; 5905102ef84SWarner Losh return; 5915102ef84SWarner Losh } 5925102ef84SWarner Losh 5935102ef84SWarner Losh // $(foo) -> $(foo) 5948b823408SWarner Losh // Not sure if I want to support this or not, so for now we just pass 5958b823408SWarner Losh // it through. 5965102ef84SWarner Losh if (*src == '(') { 5977d9e9c60SEitan Adler dst += '$'; 5985102ef84SWarner Losh count = 1; 5996cabd675SWarner Losh /* If the string ends before ) is matched , return. */ 6006cabd675SWarner Losh while (count > 0 && *src) { 6015102ef84SWarner Losh if (*src == ')') 6025102ef84SWarner Losh count--; 6035102ef84SWarner Losh else if (*src == '(') 6045102ef84SWarner Losh count++; 6057d9e9c60SEitan Adler dst += *src++; 6065102ef84SWarner Losh } 6075102ef84SWarner Losh return; 6085102ef84SWarner Losh } 6095102ef84SWarner Losh 6104cb9d1beSEitan Adler // $[^A-Za-z] -> $\1 6115102ef84SWarner Losh if (!isalpha(*src)) { 6127d9e9c60SEitan Adler dst += '$'; 6137d9e9c60SEitan Adler dst += *src++; 6145102ef84SWarner Losh return; 6155102ef84SWarner Losh } 6165102ef84SWarner Losh 6175102ef84SWarner Losh // $var -> replace with value 6186cabd675SWarner Losh do { 6197d9e9c60SEitan Adler buffer += *src++; 62066ff6821SWarner Losh } while (is_id_char(*src)); 6215a3b1a3dSEitan Adler dst.append(get_variable(buffer)); 6225102ef84SWarner Losh } 6235102ef84SWarner Losh 6245102ef84SWarner Losh const string 6255dfc0f6cSIan Lepore config::expand_string(const char *src, const char *prepend, const char *append) 6265102ef84SWarner Losh { 6275dfc0f6cSIan Lepore const char *var_at; 6286cabd675SWarner Losh string dst; 6295102ef84SWarner Losh 6305dfc0f6cSIan Lepore /* 6315dfc0f6cSIan Lepore * 128 bytes is enough for 2427 of 2438 expansions that happen 6325dfc0f6cSIan Lepore * while parsing config files, as tested on 2013-01-30. 6335dfc0f6cSIan Lepore */ 6345dfc0f6cSIan Lepore dst.reserve(128); 6355dfc0f6cSIan Lepore 6365dfc0f6cSIan Lepore if (prepend != NULL) 6375dfc0f6cSIan Lepore dst = prepend; 6385dfc0f6cSIan Lepore 6395dfc0f6cSIan Lepore for (;;) { 6405dfc0f6cSIan Lepore var_at = strchr(src, '$'); 6415dfc0f6cSIan Lepore if (var_at == NULL) { 6425dfc0f6cSIan Lepore dst.append(src); 6435dfc0f6cSIan Lepore break; 6445102ef84SWarner Losh } 6455dfc0f6cSIan Lepore dst.append(src, var_at - src); 6465dfc0f6cSIan Lepore src = var_at; 6475dfc0f6cSIan Lepore expand_one(src, dst); 6485dfc0f6cSIan Lepore } 6495dfc0f6cSIan Lepore 6505dfc0f6cSIan Lepore if (append != NULL) 6515dfc0f6cSIan Lepore dst.append(append); 6525102ef84SWarner Losh 6536cabd675SWarner Losh return (dst); 6543054f218SWarner Losh } 6553054f218SWarner Losh 6568b823408SWarner Losh bool 65711fd1366SEitan Adler config::chop_var(char *&buffer, char *&lhs, char *&rhs) const 6588b823408SWarner Losh { 6598b823408SWarner Losh char *walker; 6608b823408SWarner Losh 6618b823408SWarner Losh if (*buffer == '\0') 6628b823408SWarner Losh return (false); 6638b823408SWarner Losh walker = lhs = buffer; 6648b823408SWarner Losh while (is_id_char(*walker)) 6658b823408SWarner Losh walker++; 6668b823408SWarner Losh if (*walker != '=') 6678b823408SWarner Losh return (false); 6688b823408SWarner Losh walker++; // skip = 6698b823408SWarner Losh if (*walker == '"') { 6708b823408SWarner Losh walker++; // skip " 6718b823408SWarner Losh rhs = walker; 6728b823408SWarner Losh while (*walker && *walker != '"') 6738b823408SWarner Losh walker++; 6748b823408SWarner Losh if (*walker != '"') 6758b823408SWarner Losh return (false); 6768b823408SWarner Losh rhs[-2] = '\0'; 6778b823408SWarner Losh *walker++ = '\0'; 6788b823408SWarner Losh } else { 6798b823408SWarner Losh rhs = walker; 6808b823408SWarner Losh while (*walker && !isspace(*walker)) 6818b823408SWarner Losh walker++; 6828b823408SWarner Losh if (*walker != '\0') 6838b823408SWarner Losh *walker++ = '\0'; 6848b823408SWarner Losh rhs[-1] = '\0'; 6858b823408SWarner Losh } 686b1ee04b1SWarner Losh while (isspace(*walker)) 687b1ee04b1SWarner Losh walker++; 6888b823408SWarner Losh buffer = walker; 6898b823408SWarner Losh return (true); 6908b823408SWarner Losh } 6918b823408SWarner Losh 6928b823408SWarner Losh 6938b823408SWarner Losh char * 6948b823408SWarner Losh config::set_vars(char *buffer) 6958b823408SWarner Losh { 6968b823408SWarner Losh char *lhs; 6978b823408SWarner Losh char *rhs; 6988b823408SWarner Losh 6998b823408SWarner Losh while (1) { 7008b823408SWarner Losh if (!chop_var(buffer, lhs, rhs)) 7018b823408SWarner Losh break; 7028b823408SWarner Losh set_variable(lhs, rhs); 7038b823408SWarner Losh } 7048b823408SWarner Losh return (buffer); 7058b823408SWarner Losh } 7068b823408SWarner Losh 7078b823408SWarner Losh void 7088b823408SWarner Losh config::find_and_execute(char type) 7098b823408SWarner Losh { 7108b823408SWarner Losh vector<event_proc *> *l; 7118b823408SWarner Losh vector<event_proc *>::const_iterator i; 71294d2d4ebSBrooks Davis const char *s; 7138b823408SWarner Losh 7148b823408SWarner Losh switch (type) { 7158b823408SWarner Losh default: 7168b823408SWarner Losh return; 717842ccec5SWarner Losh case notify: 718842ccec5SWarner Losh l = &_notify_list; 719842ccec5SWarner Losh s = "notify"; 720842ccec5SWarner Losh break; 7218b823408SWarner Losh case nomatch: 7228b823408SWarner Losh l = &_nomatch_list; 7238b823408SWarner Losh s = "nomatch"; 7248b823408SWarner Losh break; 7258b823408SWarner Losh case attach: 7268b823408SWarner Losh l = &_attach_list; 7278b823408SWarner Losh s = "attach"; 7288b823408SWarner Losh break; 7298b823408SWarner Losh case detach: 7308b823408SWarner Losh l = &_detach_list; 7318b823408SWarner Losh s = "detach"; 7328b823408SWarner Losh break; 7338b823408SWarner Losh } 7345a882d77SWarner Losh if (Dflag) 7358b823408SWarner Losh fprintf(stderr, "Processing %s event\n", s); 736edee691dSEitan Adler for (i = l->begin(); i != l->end(); ++i) { 7378b823408SWarner Losh if ((*i)->matches(*this)) { 7388b823408SWarner Losh (*i)->run(*this); 7398b823408SWarner Losh break; 7408b823408SWarner Losh } 7418b823408SWarner Losh } 7428b823408SWarner Losh 7438b823408SWarner Losh } 7448b823408SWarner Losh 7453054f218SWarner Losh 7463054f218SWarner Losh static void 7478b823408SWarner Losh process_event(char *buffer) 7483054f218SWarner Losh { 7493054f218SWarner Losh char type; 7503054f218SWarner Losh char *sp; 7513054f218SWarner Losh 7528b823408SWarner Losh sp = buffer + 1; 7535a882d77SWarner Losh if (Dflag) 7548b823408SWarner Losh fprintf(stderr, "Processing event '%s'\n", buffer); 7553054f218SWarner Losh type = *buffer++; 7568b823408SWarner Losh cfg.push_var_table(); 7578b823408SWarner Losh // No match doesn't have a device, and the format is a little 7588b823408SWarner Losh // different, so handle it separately. 759842ccec5SWarner Losh switch (type) { 760842ccec5SWarner Losh case notify: 761842ccec5SWarner Losh sp = cfg.set_vars(sp); 762842ccec5SWarner Losh break; 763842ccec5SWarner Losh case nomatch: 764528d3f79SWarner Losh //? at location pnp-info on bus 765528d3f79SWarner Losh sp = strchr(sp, ' '); 766528d3f79SWarner Losh if (sp == NULL) 767528d3f79SWarner Losh return; /* Can't happen? */ 768528d3f79SWarner Losh *sp++ = '\0'; 7699b6cb520SWarner Losh while (isspace(*sp)) 7709b6cb520SWarner Losh sp++; 771842ccec5SWarner Losh if (strncmp(sp, "at ", 3) == 0) 772842ccec5SWarner Losh sp += 3; 773842ccec5SWarner Losh sp = cfg.set_vars(sp); 7749b6cb520SWarner Losh while (isspace(*sp)) 7759b6cb520SWarner Losh sp++; 776842ccec5SWarner Losh if (strncmp(sp, "on ", 3) == 0) 777842ccec5SWarner Losh cfg.set_variable("bus", sp + 3); 778842ccec5SWarner Losh break; 779842ccec5SWarner Losh case attach: /*FALLTHROUGH*/ 780842ccec5SWarner Losh case detach: 7818b823408SWarner Losh sp = strchr(sp, ' '); 7823054f218SWarner Losh if (sp == NULL) 7833054f218SWarner Losh return; /* Can't happen? */ 7848b823408SWarner Losh *sp++ = '\0'; 7858b823408SWarner Losh cfg.set_variable("device-name", buffer); 7869b6cb520SWarner Losh while (isspace(*sp)) 7879b6cb520SWarner Losh sp++; 7888b823408SWarner Losh if (strncmp(sp, "at ", 3) == 0) 7898b823408SWarner Losh sp += 3; 7908b823408SWarner Losh sp = cfg.set_vars(sp); 7919b6cb520SWarner Losh while (isspace(*sp)) 7929b6cb520SWarner Losh sp++; 7938b823408SWarner Losh if (strncmp(sp, "on ", 3) == 0) 7948b823408SWarner Losh cfg.set_variable("bus", sp + 3); 795842ccec5SWarner Losh break; 796b1ee04b1SWarner Losh } 797b1ee04b1SWarner Losh 7988b823408SWarner Losh cfg.find_and_execute(type); 7998b823408SWarner Losh cfg.pop_var_table(); 8003054f218SWarner Losh } 8013054f218SWarner Losh 802251bedd1SWarner Losh int 803251bedd1SWarner Losh create_socket(const char *name) 804251bedd1SWarner Losh { 805251bedd1SWarner Losh int fd, slen; 806251bedd1SWarner Losh struct sockaddr_un sun; 807251bedd1SWarner Losh 808251bedd1SWarner Losh if ((fd = socket(PF_LOCAL, SOCK_STREAM, 0)) < 0) 809251bedd1SWarner Losh err(1, "socket"); 810251bedd1SWarner Losh bzero(&sun, sizeof(sun)); 811251bedd1SWarner Losh sun.sun_family = AF_UNIX; 812251bedd1SWarner Losh strlcpy(sun.sun_path, name, sizeof(sun.sun_path)); 813251bedd1SWarner Losh slen = SUN_LEN(&sun); 814251bedd1SWarner Losh unlink(name); 815e60fa014SJoe Marcus Clarke if (fcntl(fd, F_SETFL, O_NONBLOCK) < 0) 816e60fa014SJoe Marcus Clarke err(1, "fcntl"); 8175d586d29SDimitry Andric if (::bind(fd, (struct sockaddr *) & sun, slen) < 0) 818251bedd1SWarner Losh err(1, "bind"); 819251bedd1SWarner Losh listen(fd, 4); 82083818929SJoe Marcus Clarke chown(name, 0, 0); /* XXX - root.wheel */ 821e60fa014SJoe Marcus Clarke chmod(name, 0666); 822251bedd1SWarner Losh return (fd); 823251bedd1SWarner Losh } 824251bedd1SWarner Losh 825e1334f93SIan Lepore unsigned int max_clients = 10; /* Default, can be overriden on cmdline. */ 826e1334f93SIan Lepore unsigned int num_clients; 827251bedd1SWarner Losh list<int> clients; 828251bedd1SWarner Losh 829251bedd1SWarner Losh void 830251bedd1SWarner Losh notify_clients(const char *data, int len) 831251bedd1SWarner Losh { 832e1334f93SIan Lepore list<int>::iterator i; 833251bedd1SWarner Losh 834e1334f93SIan Lepore /* 835e1334f93SIan Lepore * Deliver the data to all clients. Throw clients overboard at the 836e1334f93SIan Lepore * first sign of trouble. This reaps clients who've died or closed 837e1334f93SIan Lepore * their sockets, and also clients who are alive but failing to keep up 838e1334f93SIan Lepore * (or who are maliciously not reading, to consume buffer space in 839e1334f93SIan Lepore * kernel memory or tie up the limited number of available connections). 840e1334f93SIan Lepore */ 841e1334f93SIan Lepore for (i = clients.begin(); i != clients.end(); ) { 842e1334f93SIan Lepore if (write(*i, data, len) != len) { 843e1334f93SIan Lepore --num_clients; 844251bedd1SWarner Losh close(*i); 845e1334f93SIan Lepore i = clients.erase(i); 846e1334f93SIan Lepore } else 847e1334f93SIan Lepore ++i; 848251bedd1SWarner Losh } 849251bedd1SWarner Losh } 850251bedd1SWarner Losh 851e1334f93SIan Lepore void 852e1334f93SIan Lepore check_clients(void) 853e1334f93SIan Lepore { 854e1334f93SIan Lepore int s; 855e1334f93SIan Lepore struct pollfd pfd; 856e1334f93SIan Lepore list<int>::iterator i; 857e1334f93SIan Lepore 858e1334f93SIan Lepore /* 859e1334f93SIan Lepore * Check all existing clients to see if any of them have disappeared. 860e1334f93SIan Lepore * Normally we reap clients when we get an error trying to send them an 861e1334f93SIan Lepore * event. This check eliminates the problem of an ever-growing list of 862e1334f93SIan Lepore * zombie clients because we're never writing to them on a system 863e1334f93SIan Lepore * without frequent device-change activity. 864e1334f93SIan Lepore */ 865e1334f93SIan Lepore pfd.events = 0; 866e1334f93SIan Lepore for (i = clients.begin(); i != clients.end(); ) { 867e1334f93SIan Lepore pfd.fd = *i; 868e1334f93SIan Lepore s = poll(&pfd, 1, 0); 869e1334f93SIan Lepore if ((s < 0 && s != EINTR ) || 870e1334f93SIan Lepore (s > 0 && (pfd.revents & POLLHUP))) { 871e1334f93SIan Lepore --num_clients; 872e1334f93SIan Lepore close(*i); 873e1334f93SIan Lepore i = clients.erase(i); 874e1334f93SIan Lepore } else 875e1334f93SIan Lepore ++i; 876e1334f93SIan Lepore } 877251bedd1SWarner Losh } 878251bedd1SWarner Losh 879251bedd1SWarner Losh void 880251bedd1SWarner Losh new_client(int fd) 881251bedd1SWarner Losh { 882251bedd1SWarner Losh int s; 883251bedd1SWarner Losh 884e1334f93SIan Lepore /* 885e1334f93SIan Lepore * First go reap any zombie clients, then accept the connection, and 886e1334f93SIan Lepore * shut down the read side to stop clients from consuming kernel memory 887e1334f93SIan Lepore * by sending large buffers full of data we'll never read. 888e1334f93SIan Lepore */ 889e1334f93SIan Lepore check_clients(); 890251bedd1SWarner Losh s = accept(fd, NULL, NULL); 891e1334f93SIan Lepore if (s != -1) { 892e1334f93SIan Lepore shutdown(s, SHUT_RD); 893251bedd1SWarner Losh clients.push_back(s); 894e1334f93SIan Lepore ++num_clients; 895e1334f93SIan Lepore } 896251bedd1SWarner Losh } 897251bedd1SWarner Losh 8983054f218SWarner Losh static void 8993054f218SWarner Losh event_loop(void) 9003054f218SWarner Losh { 9013054f218SWarner Losh int rv; 9023054f218SWarner Losh int fd; 9033054f218SWarner Losh char buffer[DEVCTL_MAXBUF]; 9045a882d77SWarner Losh int once = 0; 905251bedd1SWarner Losh int server_fd, max_fd; 906e1334f93SIan Lepore int accepting; 9075a882d77SWarner Losh timeval tv; 9085a882d77SWarner Losh fd_set fds; 9093054f218SWarner Losh 910901610f4SPawel Jakub Dawidek fd = open(PATH_DEVCTL, O_RDONLY | O_CLOEXEC); 9113054f218SWarner Losh if (fd == -1) 912251bedd1SWarner Losh err(1, "Can't open devctl device %s", PATH_DEVCTL); 913251bedd1SWarner Losh server_fd = create_socket(PIPE); 914e1334f93SIan Lepore accepting = 1; 915251bedd1SWarner Losh max_fd = max(fd, server_fd) + 1; 91667ae3bf1SEitan Adler while (!romeo_must_die) { 9175a882d77SWarner Losh if (!once && !dflag && !nflag) { 9185a882d77SWarner Losh // Check to see if we have any events pending. 9195a882d77SWarner Losh tv.tv_sec = 0; 9205a882d77SWarner Losh tv.tv_usec = 0; 9215a882d77SWarner Losh FD_ZERO(&fds); 9225a882d77SWarner Losh FD_SET(fd, &fds); 9235a882d77SWarner Losh rv = select(fd + 1, &fds, &fds, &fds, &tv); 9245a882d77SWarner Losh // No events -> we've processed all pending events 925f3059505SWarner Losh if (rv == 0) { 9265a882d77SWarner Losh if (Dflag) 9275a882d77SWarner Losh fprintf(stderr, "Calling daemon\n"); 9281a0cc6b1SPawel Jakub Dawidek cfg.remove_pidfile(); 9291a0cc6b1SPawel Jakub Dawidek cfg.open_pidfile(); 9305a882d77SWarner Losh daemon(0, 0); 9311a0cc6b1SPawel Jakub Dawidek cfg.write_pidfile(); 9325a882d77SWarner Losh once++; 9335a882d77SWarner Losh } 9345a882d77SWarner Losh } 935e1334f93SIan Lepore /* 936e1334f93SIan Lepore * When we've already got the max number of clients, stop 937e1334f93SIan Lepore * accepting new connections (don't put server_fd in the set), 938e1334f93SIan Lepore * shrink the accept() queue to reject connections quickly, and 939e1334f93SIan Lepore * poll the existing clients more often, so that we notice more 940e1334f93SIan Lepore * quickly when any of them disappear to free up client slots. 941e1334f93SIan Lepore */ 942251bedd1SWarner Losh FD_ZERO(&fds); 943251bedd1SWarner Losh FD_SET(fd, &fds); 944e1334f93SIan Lepore if (num_clients < max_clients) { 945e1334f93SIan Lepore if (!accepting) { 946e1334f93SIan Lepore listen(server_fd, max_clients); 947e1334f93SIan Lepore accepting = 1; 948e1334f93SIan Lepore } 949251bedd1SWarner Losh FD_SET(server_fd, &fds); 950e1334f93SIan Lepore tv.tv_sec = 60; 951e1334f93SIan Lepore tv.tv_usec = 0; 952e1334f93SIan Lepore } else { 953e1334f93SIan Lepore if (accepting) { 954e1334f93SIan Lepore listen(server_fd, 0); 955e1334f93SIan Lepore accepting = 0; 956e1334f93SIan Lepore } 957e1334f93SIan Lepore tv.tv_sec = 2; 958e1334f93SIan Lepore tv.tv_usec = 0; 959e1334f93SIan Lepore } 960e1334f93SIan Lepore rv = select(max_fd, &fds, NULL, NULL, &tv); 961251bedd1SWarner Losh if (rv == -1) { 962251bedd1SWarner Losh if (errno == EINTR) 963251bedd1SWarner Losh continue; 964251bedd1SWarner Losh err(1, "select"); 965e1334f93SIan Lepore } else if (rv == 0) 966e1334f93SIan Lepore check_clients(); 967251bedd1SWarner Losh if (FD_ISSET(fd, &fds)) { 9683054f218SWarner Losh rv = read(fd, buffer, sizeof(buffer) - 1); 9693054f218SWarner Losh if (rv > 0) { 970251bedd1SWarner Losh notify_clients(buffer, rv); 9713054f218SWarner Losh buffer[rv] = '\0'; 9723054f218SWarner Losh while (buffer[--rv] == '\n') 9733054f218SWarner Losh buffer[rv] = '\0'; 9743054f218SWarner Losh process_event(buffer); 9753054f218SWarner Losh } else if (rv < 0) { 9763054f218SWarner Losh if (errno != EINTR) 9773054f218SWarner Losh break; 9783054f218SWarner Losh } else { 9793054f218SWarner Losh /* EOF */ 9803054f218SWarner Losh break; 9813054f218SWarner Losh } 9823054f218SWarner Losh } 983251bedd1SWarner Losh if (FD_ISSET(server_fd, &fds)) 984251bedd1SWarner Losh new_client(server_fd); 985251bedd1SWarner Losh } 9863054f218SWarner Losh close(fd); 9873054f218SWarner Losh } 9883054f218SWarner Losh 9893054f218SWarner Losh /* 9903054f218SWarner Losh * functions that the parser uses. 9913054f218SWarner Losh */ 9923054f218SWarner Losh void 9933054f218SWarner Losh add_attach(int prio, event_proc *p) 9943054f218SWarner Losh { 9953054f218SWarner Losh cfg.add_attach(prio, p); 9963054f218SWarner Losh } 9973054f218SWarner Losh 9983054f218SWarner Losh void 9993054f218SWarner Losh add_detach(int prio, event_proc *p) 10003054f218SWarner Losh { 10013054f218SWarner Losh cfg.add_detach(prio, p); 10023054f218SWarner Losh } 10033054f218SWarner Losh 10043054f218SWarner Losh void 10053054f218SWarner Losh add_directory(const char *dir) 10063054f218SWarner Losh { 10073054f218SWarner Losh cfg.add_directory(dir); 10083054f218SWarner Losh free(const_cast<char *>(dir)); 10093054f218SWarner Losh } 10103054f218SWarner Losh 10113054f218SWarner Losh void 10123054f218SWarner Losh add_nomatch(int prio, event_proc *p) 10133054f218SWarner Losh { 10143054f218SWarner Losh cfg.add_nomatch(prio, p); 10153054f218SWarner Losh } 10163054f218SWarner Losh 1017842ccec5SWarner Losh void 1018842ccec5SWarner Losh add_notify(int prio, event_proc *p) 1019842ccec5SWarner Losh { 1020842ccec5SWarner Losh cfg.add_notify(prio, p); 1021842ccec5SWarner Losh } 1022842ccec5SWarner Losh 10233054f218SWarner Losh event_proc * 10243054f218SWarner Losh add_to_event_proc(event_proc *ep, eps *eps) 10253054f218SWarner Losh { 10263054f218SWarner Losh if (ep == NULL) 10273054f218SWarner Losh ep = new event_proc(); 10283054f218SWarner Losh ep->add(eps); 10293054f218SWarner Losh return (ep); 10303054f218SWarner Losh } 10313054f218SWarner Losh 10323054f218SWarner Losh eps * 10333054f218SWarner Losh new_action(const char *cmd) 10343054f218SWarner Losh { 10353054f218SWarner Losh eps *e = new action(cmd); 10363054f218SWarner Losh free(const_cast<char *>(cmd)); 10373054f218SWarner Losh return (e); 10383054f218SWarner Losh } 10393054f218SWarner Losh 10403054f218SWarner Losh eps * 10413054f218SWarner Losh new_match(const char *var, const char *re) 10423054f218SWarner Losh { 10435102ef84SWarner Losh eps *e = new match(cfg, var, re); 10443054f218SWarner Losh free(const_cast<char *>(var)); 10453054f218SWarner Losh free(const_cast<char *>(re)); 10463054f218SWarner Losh return (e); 10473054f218SWarner Losh } 10483054f218SWarner Losh 1049cd70782bSWarner Losh eps * 1050cd70782bSWarner Losh new_media(const char *var, const char *re) 1051cd70782bSWarner Losh { 1052cd70782bSWarner Losh eps *e = new media(cfg, var, re); 1053cd70782bSWarner Losh free(const_cast<char *>(var)); 1054cd70782bSWarner Losh free(const_cast<char *>(re)); 1055cd70782bSWarner Losh return (e); 1056cd70782bSWarner Losh } 1057cd70782bSWarner Losh 10583054f218SWarner Losh void 10593054f218SWarner Losh set_pidfile(const char *name) 10603054f218SWarner Losh { 10613054f218SWarner Losh cfg.set_pidfile(name); 10623054f218SWarner Losh free(const_cast<char *>(name)); 10633054f218SWarner Losh } 10643054f218SWarner Losh 10653054f218SWarner Losh void 10663054f218SWarner Losh set_variable(const char *var, const char *val) 10673054f218SWarner Losh { 10683054f218SWarner Losh cfg.set_variable(var, val); 10693054f218SWarner Losh free(const_cast<char *>(var)); 10703054f218SWarner Losh free(const_cast<char *>(val)); 10713054f218SWarner Losh } 10723054f218SWarner Losh 10733054f218SWarner Losh 10743054f218SWarner Losh 10753054f218SWarner Losh static void 10763054f218SWarner Losh gensighand(int) 10773054f218SWarner Losh { 1078fcdcaa88SEitan Adler romeo_must_die = 1; 10793054f218SWarner Losh } 10803054f218SWarner Losh 10813054f218SWarner Losh static void 10823054f218SWarner Losh usage() 10833054f218SWarner Losh { 1084e1334f93SIan Lepore fprintf(stderr, "usage: %s [-Ddn] [-l connlimit] [-f file]\n", 1085e1334f93SIan Lepore getprogname()); 10863054f218SWarner Losh exit(1); 10873054f218SWarner Losh } 10883054f218SWarner Losh 10899c180898SWarner Losh static void 10909c180898SWarner Losh check_devd_enabled() 10919c180898SWarner Losh { 10929c180898SWarner Losh int val = 0; 10939c180898SWarner Losh size_t len; 10949c180898SWarner Losh 10959c180898SWarner Losh len = sizeof(val); 109679f92e25SWarner Losh if (sysctlbyname(SYSCTL, &val, &len, NULL, 0) != 0) 10979c180898SWarner Losh errx(1, "devctl sysctl missing from kernel!"); 10989c180898SWarner Losh if (val) { 10999c180898SWarner Losh warnx("Setting " SYSCTL " to 0"); 11009c180898SWarner Losh val = 0; 11019c180898SWarner Losh sysctlbyname(SYSCTL, NULL, NULL, &val, sizeof(val)); 11029c180898SWarner Losh } 11039c180898SWarner Losh } 11049c180898SWarner Losh 11053054f218SWarner Losh /* 11063054f218SWarner Losh * main 11073054f218SWarner Losh */ 11083054f218SWarner Losh int 11093054f218SWarner Losh main(int argc, char **argv) 11103054f218SWarner Losh { 11113054f218SWarner Losh int ch; 11123054f218SWarner Losh 11139c180898SWarner Losh check_devd_enabled(); 1114e1334f93SIan Lepore while ((ch = getopt(argc, argv, "Ddf:l:n")) != -1) { 11153054f218SWarner Losh switch (ch) { 11165a882d77SWarner Losh case 'D': 11175a882d77SWarner Losh Dflag++; 11185a882d77SWarner Losh break; 11193054f218SWarner Losh case 'd': 11203054f218SWarner Losh dflag++; 11213054f218SWarner Losh break; 11228334958aSJoseph Koshy case 'f': 11238334958aSJoseph Koshy configfile = optarg; 11248334958aSJoseph Koshy break; 1125e1334f93SIan Lepore case 'l': 1126e1334f93SIan Lepore max_clients = MAX(1, strtoul(optarg, NULL, 0)); 1127e1334f93SIan Lepore break; 11285a882d77SWarner Losh case 'n': 11295a882d77SWarner Losh nflag++; 11305a882d77SWarner Losh break; 11313054f218SWarner Losh default: 11323054f218SWarner Losh usage(); 11333054f218SWarner Losh } 11343054f218SWarner Losh } 11353054f218SWarner Losh 11363054f218SWarner Losh cfg.parse(); 1137fd6a8f23SWarner Losh if (!dflag && nflag) { 11381a0cc6b1SPawel Jakub Dawidek cfg.open_pidfile(); 11393054f218SWarner Losh daemon(0, 0); 11401a0cc6b1SPawel Jakub Dawidek cfg.write_pidfile(); 1141fd6a8f23SWarner Losh } 1142f32a7686SWarner Losh signal(SIGPIPE, SIG_IGN); 11433054f218SWarner Losh signal(SIGHUP, gensighand); 11443054f218SWarner Losh signal(SIGINT, gensighand); 11453054f218SWarner Losh signal(SIGTERM, gensighand); 11463054f218SWarner Losh event_loop(); 11473054f218SWarner Losh return (0); 11483054f218SWarner Losh } 1149