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 76d005340fSEitan Adler #include <cctype> 77d005340fSEitan Adler #include <cerrno> 78d005340fSEitan Adler #include <cstdlib> 79d005340fSEitan Adler #include <cstdio> 80d005340fSEitan Adler #include <csignal> 81d005340fSEitan Adler #include <cstring> 82d005340fSEitan 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> 90*6d1014a3SAlan Somers #include <syslog.h> 913054f218SWarner Losh #include <unistd.h> 923054f218SWarner Losh 938b823408SWarner Losh #include <algorithm> 943054f218SWarner Losh #include <map> 953054f218SWarner Losh #include <string> 96251bedd1SWarner Losh #include <list> 973054f218SWarner Losh #include <vector> 983054f218SWarner Losh 996aeeca8eSWarner Losh #include "devd.h" /* C compatible definitions */ 1006aeeca8eSWarner Losh #include "devd.hh" /* C++ class definitions */ 1013054f218SWarner Losh 102251bedd1SWarner Losh #define PIPE "/var/run/devd.pipe" 1033054f218SWarner Losh #define CF "/etc/devd.conf" 1049c180898SWarner Losh #define SYSCTL "hw.bus.devctl_disable" 1053054f218SWarner Losh 1063054f218SWarner Losh using namespace std; 1073054f218SWarner Losh 1083054f218SWarner Losh extern FILE *yyin; 1093054f218SWarner Losh extern int lineno; 1103054f218SWarner Losh 111842ccec5SWarner Losh static const char notify = '!'; 1128b823408SWarner Losh static const char nomatch = '?'; 1138b823408SWarner Losh static const char attach = '+'; 1148b823408SWarner Losh static const char detach = '-'; 1158b823408SWarner Losh 1161a0cc6b1SPawel Jakub Dawidek static struct pidfh *pfh; 1171a0cc6b1SPawel Jakub Dawidek 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 124*6d1014a3SAlan Somers static void devdlog(int priority, const char* message, ...); 1253054f218SWarner Losh static void event_loop(void); 1263054f218SWarner Losh static void usage(void); 1273054f218SWarner Losh 1288b823408SWarner Losh template <class T> void 1298b823408SWarner Losh delete_and_clear(vector<T *> &v) 1308b823408SWarner Losh { 1318b823408SWarner Losh typename vector<T *>::const_iterator i; 1323054f218SWarner Losh 133edee691dSEitan Adler for (i = v.begin(); i != v.end(); ++i) 1348b823408SWarner Losh delete *i; 1358b823408SWarner Losh v.clear(); 1368b823408SWarner Losh } 1378b823408SWarner Losh 1383054f218SWarner Losh config cfg; 1393054f218SWarner Losh 1403054f218SWarner Losh event_proc::event_proc() : _prio(-1) 1413054f218SWarner Losh { 1425dfc0f6cSIan Lepore _epsvec.reserve(4); 1433054f218SWarner Losh } 1443054f218SWarner Losh 1453054f218SWarner Losh event_proc::~event_proc() 1463054f218SWarner Losh { 14713f8bb71SAlexander Nedotsukov delete_and_clear(_epsvec); 1483054f218SWarner Losh } 1493054f218SWarner Losh 1503054f218SWarner Losh void 1513054f218SWarner Losh event_proc::add(eps *eps) 1523054f218SWarner Losh { 1533054f218SWarner Losh _epsvec.push_back(eps); 1543054f218SWarner Losh } 1553054f218SWarner Losh 1563054f218SWarner Losh bool 157ef370346SEitan Adler event_proc::matches(config &c) const 1583054f218SWarner Losh { 1593054f218SWarner Losh vector<eps *>::const_iterator i; 1603054f218SWarner Losh 161edee691dSEitan Adler for (i = _epsvec.begin(); i != _epsvec.end(); ++i) 1623054f218SWarner Losh if (!(*i)->do_match(c)) 1633054f218SWarner Losh return (false); 1643054f218SWarner Losh return (true); 1653054f218SWarner Losh } 1663054f218SWarner Losh 1673054f218SWarner Losh bool 168ef370346SEitan Adler event_proc::run(config &c) const 1693054f218SWarner Losh { 1703054f218SWarner Losh vector<eps *>::const_iterator i; 1713054f218SWarner Losh 172edee691dSEitan Adler for (i = _epsvec.begin(); i != _epsvec.end(); ++i) 1733054f218SWarner Losh if (!(*i)->do_action(c)) 1743054f218SWarner Losh return (false); 1753054f218SWarner Losh return (true); 1763054f218SWarner Losh } 1773054f218SWarner Losh 1783054f218SWarner Losh action::action(const char *cmd) 1793054f218SWarner Losh : _cmd(cmd) 1803054f218SWarner Losh { 1813054f218SWarner Losh // nothing 1823054f218SWarner Losh } 1833054f218SWarner Losh 1843054f218SWarner Losh action::~action() 1853054f218SWarner Losh { 1863054f218SWarner Losh // nothing 1873054f218SWarner Losh } 1883054f218SWarner Losh 189b8f92ce4SWarner Losh static int 190b8f92ce4SWarner Losh my_system(const char *command) 191b8f92ce4SWarner Losh { 192b8f92ce4SWarner Losh pid_t pid, savedpid; 193b8f92ce4SWarner Losh int pstat; 194b8f92ce4SWarner Losh struct sigaction ign, intact, quitact; 195b8f92ce4SWarner Losh sigset_t newsigblock, oldsigblock; 196b8f92ce4SWarner Losh 197b8f92ce4SWarner Losh if (!command) /* just checking... */ 198b8f92ce4SWarner Losh return(1); 199b8f92ce4SWarner Losh 200b8f92ce4SWarner Losh /* 201b8f92ce4SWarner Losh * Ignore SIGINT and SIGQUIT, block SIGCHLD. Remember to save 202b8f92ce4SWarner Losh * existing signal dispositions. 203b8f92ce4SWarner Losh */ 204b8f92ce4SWarner Losh ign.sa_handler = SIG_IGN; 205b8f92ce4SWarner Losh ::sigemptyset(&ign.sa_mask); 206b8f92ce4SWarner Losh ign.sa_flags = 0; 207b8f92ce4SWarner Losh ::sigaction(SIGINT, &ign, &intact); 208b8f92ce4SWarner Losh ::sigaction(SIGQUIT, &ign, &quitact); 209b8f92ce4SWarner Losh ::sigemptyset(&newsigblock); 210b8f92ce4SWarner Losh ::sigaddset(&newsigblock, SIGCHLD); 211b8f92ce4SWarner Losh ::sigprocmask(SIG_BLOCK, &newsigblock, &oldsigblock); 212b8f92ce4SWarner Losh switch (pid = ::fork()) { 213b8f92ce4SWarner Losh case -1: /* error */ 214b8f92ce4SWarner Losh break; 215b8f92ce4SWarner Losh case 0: /* child */ 216b8f92ce4SWarner Losh /* 217b8f92ce4SWarner Losh * Restore original signal dispositions and exec the command. 218b8f92ce4SWarner Losh */ 219b8f92ce4SWarner Losh ::sigaction(SIGINT, &intact, NULL); 220b8f92ce4SWarner Losh ::sigaction(SIGQUIT, &quitact, NULL); 221b8f92ce4SWarner Losh ::sigprocmask(SIG_SETMASK, &oldsigblock, NULL); 222b8f92ce4SWarner Losh /* 223b8f92ce4SWarner Losh * Close the PID file, and all other open descriptors. 224b8f92ce4SWarner Losh * Inherit std{in,out,err} only. 225b8f92ce4SWarner Losh */ 226b8f92ce4SWarner Losh cfg.close_pidfile(); 227b8f92ce4SWarner Losh ::closefrom(3); 228b8f92ce4SWarner Losh ::execl(_PATH_BSHELL, "sh", "-c", command, (char *)NULL); 229b8f92ce4SWarner Losh ::_exit(127); 230b8f92ce4SWarner Losh default: /* parent */ 231b8f92ce4SWarner Losh savedpid = pid; 232b8f92ce4SWarner Losh do { 233b8f92ce4SWarner Losh pid = ::wait4(savedpid, &pstat, 0, (struct rusage *)0); 234b8f92ce4SWarner Losh } while (pid == -1 && errno == EINTR); 235b8f92ce4SWarner Losh break; 236b8f92ce4SWarner Losh } 237b8f92ce4SWarner Losh ::sigaction(SIGINT, &intact, NULL); 238b8f92ce4SWarner Losh ::sigaction(SIGQUIT, &quitact, NULL); 239b8f92ce4SWarner Losh ::sigprocmask(SIG_SETMASK, &oldsigblock, NULL); 240b8f92ce4SWarner Losh return (pid == -1 ? -1 : pstat); 241b8f92ce4SWarner Losh } 242b8f92ce4SWarner Losh 2433054f218SWarner Losh bool 2445102ef84SWarner Losh action::do_action(config &c) 2453054f218SWarner Losh { 2465dfc0f6cSIan Lepore string s = c.expand_string(_cmd.c_str()); 247*6d1014a3SAlan Somers devdlog(LOG_NOTICE, "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 271*6d1014a3SAlan Somers /* 272*6d1014a3SAlan Somers * This function gets called WAY too often to justify calling syslog() 273*6d1014a3SAlan Somers * each time, even at LOG_DEBUG. Because if syslogd isn't running, it 274*6d1014a3SAlan Somers * can consume excessive amounts of systime inside of connect(). Only 275*6d1014a3SAlan Somers * log when we're in -d mode. 276*6d1014a3SAlan Somers */ 277*6d1014a3SAlan Somers if (dflag) { 278*6d1014a3SAlan Somers devdlog(LOG_DEBUG, "Testing %s=%s against %s, invert=%d\n", 2790321b694SHiroki Sato _var.c_str(), value.c_str(), _re.c_str(), _inv); 280*6d1014a3SAlan Somers } 2818b823408SWarner Losh 2825102ef84SWarner Losh retval = (regexec(&_regex, value.c_str(), 0, NULL, 0) == 0); 2830321b694SHiroki Sato if (_inv == 1) 2840321b694SHiroki Sato retval = (retval == 0) ? 1 : 0; 2850321b694SHiroki Sato 2865102ef84SWarner Losh return retval; 2873054f218SWarner Losh } 2883054f218SWarner Losh 289cd70782bSWarner Losh #include <sys/sockio.h> 290cd70782bSWarner Losh #include <net/if.h> 291cd70782bSWarner Losh #include <net/if_media.h> 292cd70782bSWarner Losh 29394d2d4ebSBrooks Davis media::media(config &, const char *var, const char *type) 294cd70782bSWarner Losh : _var(var), _type(-1) 295cd70782bSWarner Losh { 296cd70782bSWarner Losh static struct ifmedia_description media_types[] = { 297cd70782bSWarner Losh { IFM_ETHER, "Ethernet" }, 298cd70782bSWarner Losh { IFM_TOKEN, "Tokenring" }, 299cd70782bSWarner Losh { IFM_FDDI, "FDDI" }, 300cd70782bSWarner Losh { IFM_IEEE80211, "802.11" }, 301cd70782bSWarner Losh { IFM_ATM, "ATM" }, 302cd70782bSWarner Losh { -1, "unknown" }, 303cd70782bSWarner Losh { 0, NULL }, 304cd70782bSWarner Losh }; 305edee691dSEitan Adler for (int i = 0; media_types[i].ifmt_string != NULL; ++i) 306cd70782bSWarner Losh if (strcasecmp(type, media_types[i].ifmt_string) == 0) { 307cd70782bSWarner Losh _type = media_types[i].ifmt_word; 308cd70782bSWarner Losh break; 309cd70782bSWarner Losh } 310cd70782bSWarner Losh } 311cd70782bSWarner Losh 312cd70782bSWarner Losh media::~media() 313cd70782bSWarner Losh { 314cd70782bSWarner Losh } 315cd70782bSWarner Losh 316cd70782bSWarner Losh bool 317cd70782bSWarner Losh media::do_match(config &c) 318cd70782bSWarner Losh { 31909c47e76SWarner Losh string value; 320cd70782bSWarner Losh struct ifmediareq ifmr; 321cd70782bSWarner Losh bool retval; 322cd70782bSWarner Losh int s; 323cd70782bSWarner Losh 32409c47e76SWarner Losh // Since we can be called from both a device attach/detach 32509c47e76SWarner Losh // context where device-name is defined and what we want, 32609c47e76SWarner Losh // as well as from a link status context, where subsystem is 32709c47e76SWarner Losh // the name of interest, first try device-name and fall back 32809c47e76SWarner Losh // to subsystem if none exists. 32909c47e76SWarner Losh value = c.get_variable("device-name"); 330a6393aa1SEitan Adler if (value.empty()) 331de02eec8SWarner Losh value = c.get_variable("subsystem"); 332*6d1014a3SAlan Somers devdlog(LOG_DEBUG, "Testing media type of %s against 0x%x\n", 333cd70782bSWarner Losh value.c_str(), _type); 334cd70782bSWarner Losh 335cd70782bSWarner Losh retval = false; 336cd70782bSWarner Losh 337cd70782bSWarner Losh s = socket(PF_INET, SOCK_DGRAM, 0); 338cd70782bSWarner Losh if (s >= 0) { 339cd70782bSWarner Losh memset(&ifmr, 0, sizeof(ifmr)); 340cd70782bSWarner Losh strncpy(ifmr.ifm_name, value.c_str(), sizeof(ifmr.ifm_name)); 341cd70782bSWarner Losh 342cd70782bSWarner Losh if (ioctl(s, SIOCGIFMEDIA, (caddr_t)&ifmr) >= 0 && 343cd70782bSWarner Losh ifmr.ifm_status & IFM_AVALID) { 344*6d1014a3SAlan Somers devdlog(LOG_DEBUG, "%s has media type 0x%x\n", 345cd70782bSWarner Losh value.c_str(), IFM_TYPE(ifmr.ifm_active)); 346cd70782bSWarner Losh retval = (IFM_TYPE(ifmr.ifm_active) == _type); 347cd70782bSWarner Losh } else if (_type == -1) { 348*6d1014a3SAlan Somers devdlog(LOG_DEBUG, "%s has unknown media type\n", 349cd70782bSWarner Losh value.c_str()); 350cd70782bSWarner Losh retval = true; 351cd70782bSWarner Losh } 352cd70782bSWarner Losh close(s); 353cd70782bSWarner Losh } 354cd70782bSWarner Losh 355cd70782bSWarner Losh return retval; 356cd70782bSWarner Losh } 357cd70782bSWarner Losh 3583054f218SWarner Losh const string var_list::bogus = "_$_$_$_$_B_O_G_U_S_$_$_$_$_"; 3593054f218SWarner Losh const string var_list::nothing = ""; 3603054f218SWarner Losh 3613054f218SWarner Losh const string & 3623054f218SWarner Losh var_list::get_variable(const string &var) const 3633054f218SWarner Losh { 3643054f218SWarner Losh map<string, string>::const_iterator i; 3653054f218SWarner Losh 3663054f218SWarner Losh i = _vars.find(var); 3673054f218SWarner Losh if (i == _vars.end()) 3688b823408SWarner Losh return (var_list::bogus); 3693054f218SWarner Losh return (i->second); 3703054f218SWarner Losh } 3713054f218SWarner Losh 3723054f218SWarner Losh bool 3733054f218SWarner Losh var_list::is_set(const string &var) const 3743054f218SWarner Losh { 3753054f218SWarner Losh return (_vars.find(var) != _vars.end()); 3763054f218SWarner Losh } 3773054f218SWarner Losh 3783054f218SWarner Losh void 3793054f218SWarner Losh var_list::set_variable(const string &var, const string &val) 3803054f218SWarner Losh { 381*6d1014a3SAlan Somers /* 382*6d1014a3SAlan Somers * This function gets called WAY too often to justify calling syslog() 383*6d1014a3SAlan Somers * each time, even at LOG_DEBUG. Because if syslogd isn't running, it 384*6d1014a3SAlan Somers * can consume excessive amounts of systime inside of connect(). Only 385*6d1014a3SAlan Somers * log when we're in -d mode. 386*6d1014a3SAlan Somers */ 387*6d1014a3SAlan Somers if (dflag) 388*6d1014a3SAlan Somers devdlog(LOG_DEBUG, "setting %s=%s\n", var.c_str(), val.c_str()); 3893054f218SWarner Losh _vars[var] = val; 3903054f218SWarner Losh } 3913054f218SWarner Losh 3923054f218SWarner Losh void 3933054f218SWarner Losh config::reset(void) 3943054f218SWarner Losh { 3953054f218SWarner Losh _dir_list.clear(); 3968b823408SWarner Losh delete_and_clear(_var_list_table); 3978b823408SWarner Losh delete_and_clear(_attach_list); 3988b823408SWarner Losh delete_and_clear(_detach_list); 3998b823408SWarner Losh delete_and_clear(_nomatch_list); 400842ccec5SWarner Losh delete_and_clear(_notify_list); 4013054f218SWarner Losh } 4023054f218SWarner Losh 4033054f218SWarner Losh void 4043054f218SWarner Losh config::parse_one_file(const char *fn) 4053054f218SWarner Losh { 406*6d1014a3SAlan Somers devdlog(LOG_DEBUG, "Parsing %s\n", fn); 4073054f218SWarner Losh yyin = fopen(fn, "r"); 4083054f218SWarner Losh if (yyin == NULL) 4093054f218SWarner Losh err(1, "Cannot open config file %s", fn); 4104a3050fcSMaxim Konovalov lineno = 1; 4113054f218SWarner Losh if (yyparse() != 0) 4123054f218SWarner Losh errx(1, "Cannot parse %s at line %d", fn, lineno); 4133054f218SWarner Losh fclose(yyin); 4143054f218SWarner Losh } 4153054f218SWarner Losh 4163054f218SWarner Losh void 4173054f218SWarner Losh config::parse_files_in_dir(const char *dirname) 4183054f218SWarner Losh { 4193054f218SWarner Losh DIR *dirp; 4203054f218SWarner Losh struct dirent *dp; 4213054f218SWarner Losh char path[PATH_MAX]; 4223054f218SWarner Losh 423*6d1014a3SAlan Somers devdlog(LOG_DEBUG, "Parsing files in %s\n", dirname); 4243054f218SWarner Losh dirp = opendir(dirname); 4253054f218SWarner Losh if (dirp == NULL) 4263054f218SWarner Losh return; 4273054f218SWarner Losh readdir(dirp); /* Skip . */ 4283054f218SWarner Losh readdir(dirp); /* Skip .. */ 4293054f218SWarner Losh while ((dp = readdir(dirp)) != NULL) { 4303054f218SWarner Losh if (strcmp(dp->d_name + dp->d_namlen - 5, ".conf") == 0) { 4313054f218SWarner Losh snprintf(path, sizeof(path), "%s/%s", 4323054f218SWarner Losh dirname, dp->d_name); 4333054f218SWarner Losh parse_one_file(path); 4343054f218SWarner Losh } 4353054f218SWarner Losh } 4365983e891SKevin Lo closedir(dirp); 4373054f218SWarner Losh } 4383054f218SWarner Losh 4398b823408SWarner Losh class epv_greater { 4408b823408SWarner Losh public: 441ef370346SEitan Adler int operator()(event_proc *const&l1, event_proc *const&l2) const 4428b823408SWarner Losh { 4438b823408SWarner Losh return (l1->get_priority() > l2->get_priority()); 4448b823408SWarner Losh } 4458b823408SWarner Losh }; 4468b823408SWarner Losh 4478b823408SWarner Losh void 4488b823408SWarner Losh config::sort_vector(vector<event_proc *> &v) 4498b823408SWarner Losh { 4500c04d185SDimitry Andric stable_sort(v.begin(), v.end(), epv_greater()); 4518b823408SWarner Losh } 4528b823408SWarner Losh 4533054f218SWarner Losh void 4543054f218SWarner Losh config::parse(void) 4553054f218SWarner Losh { 4563054f218SWarner Losh vector<string>::const_iterator i; 4573054f218SWarner Losh 4588334958aSJoseph Koshy parse_one_file(configfile); 459edee691dSEitan Adler for (i = _dir_list.begin(); i != _dir_list.end(); ++i) 4603054f218SWarner Losh parse_files_in_dir((*i).c_str()); 4618b823408SWarner Losh sort_vector(_attach_list); 4628b823408SWarner Losh sort_vector(_detach_list); 4638b823408SWarner Losh sort_vector(_nomatch_list); 464842ccec5SWarner Losh sort_vector(_notify_list); 4653054f218SWarner Losh } 4663054f218SWarner Losh 4673054f218SWarner Losh void 4681a0cc6b1SPawel Jakub Dawidek config::open_pidfile() 4693054f218SWarner Losh { 4701a0cc6b1SPawel Jakub Dawidek pid_t otherpid; 4713054f218SWarner Losh 4725a3b1a3dSEitan Adler if (_pidfile.empty()) 4733054f218SWarner Losh return; 4741a0cc6b1SPawel Jakub Dawidek pfh = pidfile_open(_pidfile.c_str(), 0600, &otherpid); 4751a0cc6b1SPawel Jakub Dawidek if (pfh == NULL) { 4761a0cc6b1SPawel Jakub Dawidek if (errno == EEXIST) 4771a0cc6b1SPawel Jakub Dawidek errx(1, "devd already running, pid: %d", (int)otherpid); 4781a0cc6b1SPawel Jakub Dawidek warn("cannot open pid file"); 4791a0cc6b1SPawel Jakub Dawidek } 4801a0cc6b1SPawel Jakub Dawidek } 4811a0cc6b1SPawel Jakub Dawidek 4821a0cc6b1SPawel Jakub Dawidek void 4831a0cc6b1SPawel Jakub Dawidek config::write_pidfile() 4841a0cc6b1SPawel Jakub Dawidek { 4851a0cc6b1SPawel Jakub Dawidek 4861a0cc6b1SPawel Jakub Dawidek pidfile_write(pfh); 4871a0cc6b1SPawel Jakub Dawidek } 4881a0cc6b1SPawel Jakub Dawidek 4891a0cc6b1SPawel Jakub Dawidek void 490b8f92ce4SWarner Losh config::close_pidfile() 491b8f92ce4SWarner Losh { 492b8f92ce4SWarner Losh 493b8f92ce4SWarner Losh pidfile_close(pfh); 494b8f92ce4SWarner Losh } 495b8f92ce4SWarner Losh 496b8f92ce4SWarner Losh void 4971a0cc6b1SPawel Jakub Dawidek config::remove_pidfile() 4981a0cc6b1SPawel Jakub Dawidek { 4991a0cc6b1SPawel Jakub Dawidek 5001a0cc6b1SPawel Jakub Dawidek pidfile_remove(pfh); 5013054f218SWarner Losh } 5023054f218SWarner Losh 5033054f218SWarner Losh void 5043054f218SWarner Losh config::add_attach(int prio, event_proc *p) 5053054f218SWarner Losh { 5063054f218SWarner Losh p->set_priority(prio); 5073054f218SWarner Losh _attach_list.push_back(p); 5083054f218SWarner Losh } 5093054f218SWarner Losh 5103054f218SWarner Losh void 5113054f218SWarner Losh config::add_detach(int prio, event_proc *p) 5123054f218SWarner Losh { 5133054f218SWarner Losh p->set_priority(prio); 5143054f218SWarner Losh _detach_list.push_back(p); 5153054f218SWarner Losh } 5163054f218SWarner Losh 5173054f218SWarner Losh void 5183054f218SWarner Losh config::add_directory(const char *dir) 5193054f218SWarner Losh { 5203054f218SWarner Losh _dir_list.push_back(string(dir)); 5213054f218SWarner Losh } 5223054f218SWarner Losh 5233054f218SWarner Losh void 5243054f218SWarner Losh config::add_nomatch(int prio, event_proc *p) 5253054f218SWarner Losh { 5263054f218SWarner Losh p->set_priority(prio); 5273054f218SWarner Losh _nomatch_list.push_back(p); 5283054f218SWarner Losh } 5293054f218SWarner Losh 5303054f218SWarner Losh void 531842ccec5SWarner Losh config::add_notify(int prio, event_proc *p) 532842ccec5SWarner Losh { 533842ccec5SWarner Losh p->set_priority(prio); 534842ccec5SWarner Losh _notify_list.push_back(p); 535842ccec5SWarner Losh } 536842ccec5SWarner Losh 537842ccec5SWarner Losh void 5383054f218SWarner Losh config::set_pidfile(const char *fn) 5393054f218SWarner Losh { 5405a3b1a3dSEitan Adler _pidfile = fn; 5413054f218SWarner Losh } 5423054f218SWarner Losh 5433054f218SWarner Losh void 5443054f218SWarner Losh config::push_var_table() 5453054f218SWarner Losh { 5463054f218SWarner Losh var_list *vl; 5473054f218SWarner Losh 5483054f218SWarner Losh vl = new var_list(); 5493054f218SWarner Losh _var_list_table.push_back(vl); 550*6d1014a3SAlan Somers devdlog(LOG_DEBUG, "Pushing table\n"); 5513054f218SWarner Losh } 5523054f218SWarner Losh 5533054f218SWarner Losh void 5543054f218SWarner Losh config::pop_var_table() 5553054f218SWarner Losh { 5563054f218SWarner Losh delete _var_list_table.back(); 5573054f218SWarner Losh _var_list_table.pop_back(); 558*6d1014a3SAlan Somers devdlog(LOG_DEBUG, "Popping table\n"); 5593054f218SWarner Losh } 5603054f218SWarner Losh 5613054f218SWarner Losh void 5623054f218SWarner Losh config::set_variable(const char *var, const char *val) 5633054f218SWarner Losh { 5643054f218SWarner Losh _var_list_table.back()->set_variable(var, val); 5653054f218SWarner Losh } 5663054f218SWarner Losh 5673054f218SWarner Losh const string & 5683054f218SWarner Losh config::get_variable(const string &var) 5693054f218SWarner Losh { 5703054f218SWarner Losh vector<var_list *>::reverse_iterator i; 5713054f218SWarner Losh 572edee691dSEitan Adler for (i = _var_list_table.rbegin(); i != _var_list_table.rend(); ++i) { 5733054f218SWarner Losh if ((*i)->is_set(var)) 5748b823408SWarner Losh return ((*i)->get_variable(var)); 5753054f218SWarner Losh } 5763054f218SWarner Losh return (var_list::nothing); 5773054f218SWarner Losh } 5783054f218SWarner Losh 5798b823408SWarner Losh bool 580ef370346SEitan Adler config::is_id_char(char ch) const 5818b823408SWarner Losh { 5828b823408SWarner Losh return (ch != '\0' && (isalpha(ch) || isdigit(ch) || ch == '_' || 5838b823408SWarner Losh ch == '-')); 5848b823408SWarner Losh } 5858b823408SWarner Losh 5865102ef84SWarner Losh void 5876cabd675SWarner Losh config::expand_one(const char *&src, string &dst) 5883054f218SWarner Losh { 5895102ef84SWarner Losh int count; 5905bf08422SUlf Lilleengen string buffer; 5915102ef84SWarner Losh 5928b823408SWarner Losh src++; 5935102ef84SWarner Losh // $$ -> $ 5945102ef84SWarner Losh if (*src == '$') { 5957d9e9c60SEitan Adler dst += *src++; 5965102ef84SWarner Losh return; 5975102ef84SWarner Losh } 5985102ef84SWarner Losh 5995102ef84SWarner Losh // $(foo) -> $(foo) 6008b823408SWarner Losh // Not sure if I want to support this or not, so for now we just pass 6018b823408SWarner Losh // it through. 6025102ef84SWarner Losh if (*src == '(') { 6037d9e9c60SEitan Adler dst += '$'; 6045102ef84SWarner Losh count = 1; 6056cabd675SWarner Losh /* If the string ends before ) is matched , return. */ 6066cabd675SWarner Losh while (count > 0 && *src) { 6075102ef84SWarner Losh if (*src == ')') 6085102ef84SWarner Losh count--; 6095102ef84SWarner Losh else if (*src == '(') 6105102ef84SWarner Losh count++; 6117d9e9c60SEitan Adler dst += *src++; 6125102ef84SWarner Losh } 6135102ef84SWarner Losh return; 6145102ef84SWarner Losh } 6155102ef84SWarner Losh 6164cb9d1beSEitan Adler // $[^A-Za-z] -> $\1 6175102ef84SWarner Losh if (!isalpha(*src)) { 6187d9e9c60SEitan Adler dst += '$'; 6197d9e9c60SEitan Adler dst += *src++; 6205102ef84SWarner Losh return; 6215102ef84SWarner Losh } 6225102ef84SWarner Losh 6235102ef84SWarner Losh // $var -> replace with value 6246cabd675SWarner Losh do { 6257d9e9c60SEitan Adler buffer += *src++; 62666ff6821SWarner Losh } while (is_id_char(*src)); 6275a3b1a3dSEitan Adler dst.append(get_variable(buffer)); 6285102ef84SWarner Losh } 6295102ef84SWarner Losh 6305102ef84SWarner Losh const string 6315dfc0f6cSIan Lepore config::expand_string(const char *src, const char *prepend, const char *append) 6325102ef84SWarner Losh { 6335dfc0f6cSIan Lepore const char *var_at; 6346cabd675SWarner Losh string dst; 6355102ef84SWarner Losh 6365dfc0f6cSIan Lepore /* 6375dfc0f6cSIan Lepore * 128 bytes is enough for 2427 of 2438 expansions that happen 6385dfc0f6cSIan Lepore * while parsing config files, as tested on 2013-01-30. 6395dfc0f6cSIan Lepore */ 6405dfc0f6cSIan Lepore dst.reserve(128); 6415dfc0f6cSIan Lepore 6425dfc0f6cSIan Lepore if (prepend != NULL) 6435dfc0f6cSIan Lepore dst = prepend; 6445dfc0f6cSIan Lepore 6455dfc0f6cSIan Lepore for (;;) { 6465dfc0f6cSIan Lepore var_at = strchr(src, '$'); 6475dfc0f6cSIan Lepore if (var_at == NULL) { 6485dfc0f6cSIan Lepore dst.append(src); 6495dfc0f6cSIan Lepore break; 6505102ef84SWarner Losh } 6515dfc0f6cSIan Lepore dst.append(src, var_at - src); 6525dfc0f6cSIan Lepore src = var_at; 6535dfc0f6cSIan Lepore expand_one(src, dst); 6545dfc0f6cSIan Lepore } 6555dfc0f6cSIan Lepore 6565dfc0f6cSIan Lepore if (append != NULL) 6575dfc0f6cSIan Lepore dst.append(append); 6585102ef84SWarner Losh 6596cabd675SWarner Losh return (dst); 6603054f218SWarner Losh } 6613054f218SWarner Losh 6628b823408SWarner Losh bool 66311fd1366SEitan Adler config::chop_var(char *&buffer, char *&lhs, char *&rhs) const 6648b823408SWarner Losh { 6658b823408SWarner Losh char *walker; 6668b823408SWarner Losh 6678b823408SWarner Losh if (*buffer == '\0') 6688b823408SWarner Losh return (false); 6698b823408SWarner Losh walker = lhs = buffer; 6708b823408SWarner Losh while (is_id_char(*walker)) 6718b823408SWarner Losh walker++; 6728b823408SWarner Losh if (*walker != '=') 6738b823408SWarner Losh return (false); 6748b823408SWarner Losh walker++; // skip = 6758b823408SWarner Losh if (*walker == '"') { 6768b823408SWarner Losh walker++; // skip " 6778b823408SWarner Losh rhs = walker; 6788b823408SWarner Losh while (*walker && *walker != '"') 6798b823408SWarner Losh walker++; 6808b823408SWarner Losh if (*walker != '"') 6818b823408SWarner Losh return (false); 6828b823408SWarner Losh rhs[-2] = '\0'; 6838b823408SWarner Losh *walker++ = '\0'; 6848b823408SWarner Losh } else { 6858b823408SWarner Losh rhs = walker; 6868b823408SWarner Losh while (*walker && !isspace(*walker)) 6878b823408SWarner Losh walker++; 6888b823408SWarner Losh if (*walker != '\0') 6898b823408SWarner Losh *walker++ = '\0'; 6908b823408SWarner Losh rhs[-1] = '\0'; 6918b823408SWarner Losh } 692b1ee04b1SWarner Losh while (isspace(*walker)) 693b1ee04b1SWarner Losh walker++; 6948b823408SWarner Losh buffer = walker; 6958b823408SWarner Losh return (true); 6968b823408SWarner Losh } 6978b823408SWarner Losh 6988b823408SWarner Losh 6998b823408SWarner Losh char * 7008b823408SWarner Losh config::set_vars(char *buffer) 7018b823408SWarner Losh { 7028b823408SWarner Losh char *lhs; 7038b823408SWarner Losh char *rhs; 7048b823408SWarner Losh 7058b823408SWarner Losh while (1) { 7068b823408SWarner Losh if (!chop_var(buffer, lhs, rhs)) 7078b823408SWarner Losh break; 7088b823408SWarner Losh set_variable(lhs, rhs); 7098b823408SWarner Losh } 7108b823408SWarner Losh return (buffer); 7118b823408SWarner Losh } 7128b823408SWarner Losh 7138b823408SWarner Losh void 7148b823408SWarner Losh config::find_and_execute(char type) 7158b823408SWarner Losh { 7168b823408SWarner Losh vector<event_proc *> *l; 7178b823408SWarner Losh vector<event_proc *>::const_iterator i; 71894d2d4ebSBrooks Davis const char *s; 7198b823408SWarner Losh 7208b823408SWarner Losh switch (type) { 7218b823408SWarner Losh default: 7228b823408SWarner Losh return; 723842ccec5SWarner Losh case notify: 724842ccec5SWarner Losh l = &_notify_list; 725842ccec5SWarner Losh s = "notify"; 726842ccec5SWarner Losh break; 7278b823408SWarner Losh case nomatch: 7288b823408SWarner Losh l = &_nomatch_list; 7298b823408SWarner Losh s = "nomatch"; 7308b823408SWarner Losh break; 7318b823408SWarner Losh case attach: 7328b823408SWarner Losh l = &_attach_list; 7338b823408SWarner Losh s = "attach"; 7348b823408SWarner Losh break; 7358b823408SWarner Losh case detach: 7368b823408SWarner Losh l = &_detach_list; 7378b823408SWarner Losh s = "detach"; 7388b823408SWarner Losh break; 7398b823408SWarner Losh } 740*6d1014a3SAlan Somers devdlog(LOG_DEBUG, "Processing %s event\n", s); 741edee691dSEitan Adler for (i = l->begin(); i != l->end(); ++i) { 7428b823408SWarner Losh if ((*i)->matches(*this)) { 7438b823408SWarner Losh (*i)->run(*this); 7448b823408SWarner Losh break; 7458b823408SWarner Losh } 7468b823408SWarner Losh } 7478b823408SWarner Losh 7488b823408SWarner Losh } 7498b823408SWarner Losh 7503054f218SWarner Losh 7513054f218SWarner Losh static void 7528b823408SWarner Losh process_event(char *buffer) 7533054f218SWarner Losh { 7543054f218SWarner Losh char type; 7553054f218SWarner Losh char *sp; 7563054f218SWarner Losh 7578b823408SWarner Losh sp = buffer + 1; 758*6d1014a3SAlan Somers devdlog(LOG_DEBUG, "Processing event '%s'\n", buffer); 7593054f218SWarner Losh type = *buffer++; 7608b823408SWarner Losh cfg.push_var_table(); 7618b823408SWarner Losh // No match doesn't have a device, and the format is a little 7628b823408SWarner Losh // different, so handle it separately. 763842ccec5SWarner Losh switch (type) { 764842ccec5SWarner Losh case notify: 765842ccec5SWarner Losh sp = cfg.set_vars(sp); 766842ccec5SWarner Losh break; 767842ccec5SWarner Losh case nomatch: 768528d3f79SWarner Losh //? at location pnp-info on bus 769528d3f79SWarner Losh sp = strchr(sp, ' '); 770528d3f79SWarner Losh if (sp == NULL) 771528d3f79SWarner Losh return; /* Can't happen? */ 772528d3f79SWarner Losh *sp++ = '\0'; 7739b6cb520SWarner Losh while (isspace(*sp)) 7749b6cb520SWarner Losh sp++; 775842ccec5SWarner Losh if (strncmp(sp, "at ", 3) == 0) 776842ccec5SWarner Losh sp += 3; 777842ccec5SWarner Losh sp = cfg.set_vars(sp); 7789b6cb520SWarner Losh while (isspace(*sp)) 7799b6cb520SWarner Losh sp++; 780842ccec5SWarner Losh if (strncmp(sp, "on ", 3) == 0) 781842ccec5SWarner Losh cfg.set_variable("bus", sp + 3); 782842ccec5SWarner Losh break; 783842ccec5SWarner Losh case attach: /*FALLTHROUGH*/ 784842ccec5SWarner Losh case detach: 7858b823408SWarner Losh sp = strchr(sp, ' '); 7863054f218SWarner Losh if (sp == NULL) 7873054f218SWarner Losh return; /* Can't happen? */ 7888b823408SWarner Losh *sp++ = '\0'; 7898b823408SWarner Losh cfg.set_variable("device-name", buffer); 7909b6cb520SWarner Losh while (isspace(*sp)) 7919b6cb520SWarner Losh sp++; 7928b823408SWarner Losh if (strncmp(sp, "at ", 3) == 0) 7938b823408SWarner Losh sp += 3; 7948b823408SWarner Losh sp = cfg.set_vars(sp); 7959b6cb520SWarner Losh while (isspace(*sp)) 7969b6cb520SWarner Losh sp++; 7978b823408SWarner Losh if (strncmp(sp, "on ", 3) == 0) 7988b823408SWarner Losh cfg.set_variable("bus", sp + 3); 799842ccec5SWarner Losh break; 800b1ee04b1SWarner Losh } 801b1ee04b1SWarner Losh 8028b823408SWarner Losh cfg.find_and_execute(type); 8038b823408SWarner Losh cfg.pop_var_table(); 8043054f218SWarner Losh } 8053054f218SWarner Losh 806251bedd1SWarner Losh int 807251bedd1SWarner Losh create_socket(const char *name) 808251bedd1SWarner Losh { 809251bedd1SWarner Losh int fd, slen; 810251bedd1SWarner Losh struct sockaddr_un sun; 811251bedd1SWarner Losh 812251bedd1SWarner Losh if ((fd = socket(PF_LOCAL, SOCK_STREAM, 0)) < 0) 813251bedd1SWarner Losh err(1, "socket"); 814251bedd1SWarner Losh bzero(&sun, sizeof(sun)); 815251bedd1SWarner Losh sun.sun_family = AF_UNIX; 816251bedd1SWarner Losh strlcpy(sun.sun_path, name, sizeof(sun.sun_path)); 817251bedd1SWarner Losh slen = SUN_LEN(&sun); 818251bedd1SWarner Losh unlink(name); 819e60fa014SJoe Marcus Clarke if (fcntl(fd, F_SETFL, O_NONBLOCK) < 0) 820e60fa014SJoe Marcus Clarke err(1, "fcntl"); 8215d586d29SDimitry Andric if (::bind(fd, (struct sockaddr *) & sun, slen) < 0) 822251bedd1SWarner Losh err(1, "bind"); 823251bedd1SWarner Losh listen(fd, 4); 82483818929SJoe Marcus Clarke chown(name, 0, 0); /* XXX - root.wheel */ 825e60fa014SJoe Marcus Clarke chmod(name, 0666); 826251bedd1SWarner Losh return (fd); 827251bedd1SWarner Losh } 828251bedd1SWarner Losh 829e1334f93SIan Lepore unsigned int max_clients = 10; /* Default, can be overriden on cmdline. */ 830e1334f93SIan Lepore unsigned int num_clients; 831251bedd1SWarner Losh list<int> clients; 832251bedd1SWarner Losh 833251bedd1SWarner Losh void 834251bedd1SWarner Losh notify_clients(const char *data, int len) 835251bedd1SWarner Losh { 836e1334f93SIan Lepore list<int>::iterator i; 837251bedd1SWarner Losh 838e1334f93SIan Lepore /* 839e1334f93SIan Lepore * Deliver the data to all clients. Throw clients overboard at the 840e1334f93SIan Lepore * first sign of trouble. This reaps clients who've died or closed 841e1334f93SIan Lepore * their sockets, and also clients who are alive but failing to keep up 842e1334f93SIan Lepore * (or who are maliciously not reading, to consume buffer space in 843e1334f93SIan Lepore * kernel memory or tie up the limited number of available connections). 844e1334f93SIan Lepore */ 845e1334f93SIan Lepore for (i = clients.begin(); i != clients.end(); ) { 846e1334f93SIan Lepore if (write(*i, data, len) != len) { 847e1334f93SIan Lepore --num_clients; 848251bedd1SWarner Losh close(*i); 849e1334f93SIan Lepore i = clients.erase(i); 850*6d1014a3SAlan Somers devdlog(LOG_WARNING, "notify_clients: write() failed; " 851*6d1014a3SAlan Somers "dropping unresponsive client\n"); 852e1334f93SIan Lepore } else 853e1334f93SIan Lepore ++i; 854251bedd1SWarner Losh } 855251bedd1SWarner Losh } 856251bedd1SWarner Losh 857e1334f93SIan Lepore void 858e1334f93SIan Lepore check_clients(void) 859e1334f93SIan Lepore { 860e1334f93SIan Lepore int s; 861e1334f93SIan Lepore struct pollfd pfd; 862e1334f93SIan Lepore list<int>::iterator i; 863e1334f93SIan Lepore 864e1334f93SIan Lepore /* 865e1334f93SIan Lepore * Check all existing clients to see if any of them have disappeared. 866e1334f93SIan Lepore * Normally we reap clients when we get an error trying to send them an 867e1334f93SIan Lepore * event. This check eliminates the problem of an ever-growing list of 868e1334f93SIan Lepore * zombie clients because we're never writing to them on a system 869e1334f93SIan Lepore * without frequent device-change activity. 870e1334f93SIan Lepore */ 871e1334f93SIan Lepore pfd.events = 0; 872e1334f93SIan Lepore for (i = clients.begin(); i != clients.end(); ) { 873e1334f93SIan Lepore pfd.fd = *i; 874e1334f93SIan Lepore s = poll(&pfd, 1, 0); 875e1334f93SIan Lepore if ((s < 0 && s != EINTR ) || 876e1334f93SIan Lepore (s > 0 && (pfd.revents & POLLHUP))) { 877e1334f93SIan Lepore --num_clients; 878e1334f93SIan Lepore close(*i); 879e1334f93SIan Lepore i = clients.erase(i); 880*6d1014a3SAlan Somers devdlog(LOG_NOTICE, "check_clients: " 881*6d1014a3SAlan Somers "dropping disconnected client\n"); 882e1334f93SIan Lepore } else 883e1334f93SIan Lepore ++i; 884e1334f93SIan Lepore } 885251bedd1SWarner Losh } 886251bedd1SWarner Losh 887251bedd1SWarner Losh void 888251bedd1SWarner Losh new_client(int fd) 889251bedd1SWarner Losh { 890251bedd1SWarner Losh int s; 891251bedd1SWarner Losh 892e1334f93SIan Lepore /* 893e1334f93SIan Lepore * First go reap any zombie clients, then accept the connection, and 894e1334f93SIan Lepore * shut down the read side to stop clients from consuming kernel memory 895e1334f93SIan Lepore * by sending large buffers full of data we'll never read. 896e1334f93SIan Lepore */ 897e1334f93SIan Lepore check_clients(); 898251bedd1SWarner Losh s = accept(fd, NULL, NULL); 899e1334f93SIan Lepore if (s != -1) { 900e1334f93SIan Lepore shutdown(s, SHUT_RD); 901251bedd1SWarner Losh clients.push_back(s); 902e1334f93SIan Lepore ++num_clients; 903e1334f93SIan Lepore } 904251bedd1SWarner Losh } 905251bedd1SWarner Losh 9063054f218SWarner Losh static void 9073054f218SWarner Losh event_loop(void) 9083054f218SWarner Losh { 9093054f218SWarner Losh int rv; 9103054f218SWarner Losh int fd; 9113054f218SWarner Losh char buffer[DEVCTL_MAXBUF]; 9125a882d77SWarner Losh int once = 0; 913251bedd1SWarner Losh int server_fd, max_fd; 914e1334f93SIan Lepore int accepting; 9155a882d77SWarner Losh timeval tv; 9165a882d77SWarner Losh fd_set fds; 9173054f218SWarner Losh 918901610f4SPawel Jakub Dawidek fd = open(PATH_DEVCTL, O_RDONLY | O_CLOEXEC); 9193054f218SWarner Losh if (fd == -1) 920251bedd1SWarner Losh err(1, "Can't open devctl device %s", PATH_DEVCTL); 921251bedd1SWarner Losh server_fd = create_socket(PIPE); 922e1334f93SIan Lepore accepting = 1; 923251bedd1SWarner Losh max_fd = max(fd, server_fd) + 1; 92467ae3bf1SEitan Adler while (!romeo_must_die) { 9255a882d77SWarner Losh if (!once && !dflag && !nflag) { 9265a882d77SWarner Losh // Check to see if we have any events pending. 9275a882d77SWarner Losh tv.tv_sec = 0; 9285a882d77SWarner Losh tv.tv_usec = 0; 9295a882d77SWarner Losh FD_ZERO(&fds); 9305a882d77SWarner Losh FD_SET(fd, &fds); 9315a882d77SWarner Losh rv = select(fd + 1, &fds, &fds, &fds, &tv); 9325a882d77SWarner Losh // No events -> we've processed all pending events 933f3059505SWarner Losh if (rv == 0) { 934*6d1014a3SAlan Somers devdlog(LOG_DEBUG, "Calling daemon\n"); 9351a0cc6b1SPawel Jakub Dawidek cfg.remove_pidfile(); 9361a0cc6b1SPawel Jakub Dawidek cfg.open_pidfile(); 9375a882d77SWarner Losh daemon(0, 0); 9381a0cc6b1SPawel Jakub Dawidek cfg.write_pidfile(); 9395a882d77SWarner Losh once++; 9405a882d77SWarner Losh } 9415a882d77SWarner Losh } 942e1334f93SIan Lepore /* 943e1334f93SIan Lepore * When we've already got the max number of clients, stop 944e1334f93SIan Lepore * accepting new connections (don't put server_fd in the set), 945e1334f93SIan Lepore * shrink the accept() queue to reject connections quickly, and 946e1334f93SIan Lepore * poll the existing clients more often, so that we notice more 947e1334f93SIan Lepore * quickly when any of them disappear to free up client slots. 948e1334f93SIan Lepore */ 949251bedd1SWarner Losh FD_ZERO(&fds); 950251bedd1SWarner Losh FD_SET(fd, &fds); 951e1334f93SIan Lepore if (num_clients < max_clients) { 952e1334f93SIan Lepore if (!accepting) { 953e1334f93SIan Lepore listen(server_fd, max_clients); 954e1334f93SIan Lepore accepting = 1; 955e1334f93SIan Lepore } 956251bedd1SWarner Losh FD_SET(server_fd, &fds); 957e1334f93SIan Lepore tv.tv_sec = 60; 958e1334f93SIan Lepore tv.tv_usec = 0; 959e1334f93SIan Lepore } else { 960e1334f93SIan Lepore if (accepting) { 961e1334f93SIan Lepore listen(server_fd, 0); 962e1334f93SIan Lepore accepting = 0; 963e1334f93SIan Lepore } 964e1334f93SIan Lepore tv.tv_sec = 2; 965e1334f93SIan Lepore tv.tv_usec = 0; 966e1334f93SIan Lepore } 967e1334f93SIan Lepore rv = select(max_fd, &fds, NULL, NULL, &tv); 968251bedd1SWarner Losh if (rv == -1) { 969251bedd1SWarner Losh if (errno == EINTR) 970251bedd1SWarner Losh continue; 971251bedd1SWarner Losh err(1, "select"); 972e1334f93SIan Lepore } else if (rv == 0) 973e1334f93SIan Lepore check_clients(); 974251bedd1SWarner Losh if (FD_ISSET(fd, &fds)) { 9753054f218SWarner Losh rv = read(fd, buffer, sizeof(buffer) - 1); 9763054f218SWarner Losh if (rv > 0) { 977*6d1014a3SAlan Somers if (rv == sizeof(buffer) - 1) { 978*6d1014a3SAlan Somers devdlog(LOG_WARNING, "Warning: " 979*6d1014a3SAlan Somers "available event data exceeded " 980*6d1014a3SAlan Somers "buffer space\n"); 981*6d1014a3SAlan Somers } 982251bedd1SWarner Losh notify_clients(buffer, rv); 9833054f218SWarner Losh buffer[rv] = '\0'; 9843054f218SWarner Losh while (buffer[--rv] == '\n') 9853054f218SWarner Losh buffer[rv] = '\0'; 9863054f218SWarner Losh process_event(buffer); 9873054f218SWarner Losh } else if (rv < 0) { 9883054f218SWarner Losh if (errno != EINTR) 9893054f218SWarner Losh break; 9903054f218SWarner Losh } else { 9913054f218SWarner Losh /* EOF */ 9923054f218SWarner Losh break; 9933054f218SWarner Losh } 9943054f218SWarner Losh } 995251bedd1SWarner Losh if (FD_ISSET(server_fd, &fds)) 996251bedd1SWarner Losh new_client(server_fd); 997251bedd1SWarner Losh } 9983054f218SWarner Losh close(fd); 9993054f218SWarner Losh } 10003054f218SWarner Losh 10013054f218SWarner Losh /* 10023054f218SWarner Losh * functions that the parser uses. 10033054f218SWarner Losh */ 10043054f218SWarner Losh void 10053054f218SWarner Losh add_attach(int prio, event_proc *p) 10063054f218SWarner Losh { 10073054f218SWarner Losh cfg.add_attach(prio, p); 10083054f218SWarner Losh } 10093054f218SWarner Losh 10103054f218SWarner Losh void 10113054f218SWarner Losh add_detach(int prio, event_proc *p) 10123054f218SWarner Losh { 10133054f218SWarner Losh cfg.add_detach(prio, p); 10143054f218SWarner Losh } 10153054f218SWarner Losh 10163054f218SWarner Losh void 10173054f218SWarner Losh add_directory(const char *dir) 10183054f218SWarner Losh { 10193054f218SWarner Losh cfg.add_directory(dir); 10203054f218SWarner Losh free(const_cast<char *>(dir)); 10213054f218SWarner Losh } 10223054f218SWarner Losh 10233054f218SWarner Losh void 10243054f218SWarner Losh add_nomatch(int prio, event_proc *p) 10253054f218SWarner Losh { 10263054f218SWarner Losh cfg.add_nomatch(prio, p); 10273054f218SWarner Losh } 10283054f218SWarner Losh 1029842ccec5SWarner Losh void 1030842ccec5SWarner Losh add_notify(int prio, event_proc *p) 1031842ccec5SWarner Losh { 1032842ccec5SWarner Losh cfg.add_notify(prio, p); 1033842ccec5SWarner Losh } 1034842ccec5SWarner Losh 10353054f218SWarner Losh event_proc * 10363054f218SWarner Losh add_to_event_proc(event_proc *ep, eps *eps) 10373054f218SWarner Losh { 10383054f218SWarner Losh if (ep == NULL) 10393054f218SWarner Losh ep = new event_proc(); 10403054f218SWarner Losh ep->add(eps); 10413054f218SWarner Losh return (ep); 10423054f218SWarner Losh } 10433054f218SWarner Losh 10443054f218SWarner Losh eps * 10453054f218SWarner Losh new_action(const char *cmd) 10463054f218SWarner Losh { 10473054f218SWarner Losh eps *e = new action(cmd); 10483054f218SWarner Losh free(const_cast<char *>(cmd)); 10493054f218SWarner Losh return (e); 10503054f218SWarner Losh } 10513054f218SWarner Losh 10523054f218SWarner Losh eps * 10533054f218SWarner Losh new_match(const char *var, const char *re) 10543054f218SWarner Losh { 10555102ef84SWarner Losh eps *e = new match(cfg, var, re); 10563054f218SWarner Losh free(const_cast<char *>(var)); 10573054f218SWarner Losh free(const_cast<char *>(re)); 10583054f218SWarner Losh return (e); 10593054f218SWarner Losh } 10603054f218SWarner Losh 1061cd70782bSWarner Losh eps * 1062cd70782bSWarner Losh new_media(const char *var, const char *re) 1063cd70782bSWarner Losh { 1064cd70782bSWarner Losh eps *e = new media(cfg, var, re); 1065cd70782bSWarner Losh free(const_cast<char *>(var)); 1066cd70782bSWarner Losh free(const_cast<char *>(re)); 1067cd70782bSWarner Losh return (e); 1068cd70782bSWarner Losh } 1069cd70782bSWarner Losh 10703054f218SWarner Losh void 10713054f218SWarner Losh set_pidfile(const char *name) 10723054f218SWarner Losh { 10733054f218SWarner Losh cfg.set_pidfile(name); 10743054f218SWarner Losh free(const_cast<char *>(name)); 10753054f218SWarner Losh } 10763054f218SWarner Losh 10773054f218SWarner Losh void 10783054f218SWarner Losh set_variable(const char *var, const char *val) 10793054f218SWarner Losh { 10803054f218SWarner Losh cfg.set_variable(var, val); 10813054f218SWarner Losh free(const_cast<char *>(var)); 10823054f218SWarner Losh free(const_cast<char *>(val)); 10833054f218SWarner Losh } 10843054f218SWarner Losh 10853054f218SWarner Losh 10863054f218SWarner Losh 10873054f218SWarner Losh static void 10883054f218SWarner Losh gensighand(int) 10893054f218SWarner Losh { 1090fcdcaa88SEitan Adler romeo_must_die = 1; 10913054f218SWarner Losh } 10923054f218SWarner Losh 1093*6d1014a3SAlan Somers /* 1094*6d1014a3SAlan Somers * Local logging function. Prints to syslog if we're daemonized; syslog 1095*6d1014a3SAlan Somers * otherwise. 1096*6d1014a3SAlan Somers */ 1097*6d1014a3SAlan Somers static void 1098*6d1014a3SAlan Somers devdlog(int priority, const char* fmt, ...) 1099*6d1014a3SAlan Somers { 1100*6d1014a3SAlan Somers va_list argp; 1101*6d1014a3SAlan Somers 1102*6d1014a3SAlan Somers va_start(argp, fmt); 1103*6d1014a3SAlan Somers if (dflag) 1104*6d1014a3SAlan Somers vfprintf(stderr, fmt, argp); 1105*6d1014a3SAlan Somers else 1106*6d1014a3SAlan Somers vsyslog(priority, fmt, argp); 1107*6d1014a3SAlan Somers va_end(argp); 1108*6d1014a3SAlan Somers } 1109*6d1014a3SAlan Somers 11103054f218SWarner Losh static void 11113054f218SWarner Losh usage() 11123054f218SWarner Losh { 1113*6d1014a3SAlan Somers fprintf(stderr, "usage: %s [-dn] [-l connlimit] [-f file]\n", 1114e1334f93SIan Lepore getprogname()); 11153054f218SWarner Losh exit(1); 11163054f218SWarner Losh } 11173054f218SWarner Losh 11189c180898SWarner Losh static void 11199c180898SWarner Losh check_devd_enabled() 11209c180898SWarner Losh { 11219c180898SWarner Losh int val = 0; 11229c180898SWarner Losh size_t len; 11239c180898SWarner Losh 11249c180898SWarner Losh len = sizeof(val); 112579f92e25SWarner Losh if (sysctlbyname(SYSCTL, &val, &len, NULL, 0) != 0) 11269c180898SWarner Losh errx(1, "devctl sysctl missing from kernel!"); 11279c180898SWarner Losh if (val) { 11289c180898SWarner Losh warnx("Setting " SYSCTL " to 0"); 11299c180898SWarner Losh val = 0; 11309c180898SWarner Losh sysctlbyname(SYSCTL, NULL, NULL, &val, sizeof(val)); 11319c180898SWarner Losh } 11329c180898SWarner Losh } 11339c180898SWarner Losh 11343054f218SWarner Losh /* 11353054f218SWarner Losh * main 11363054f218SWarner Losh */ 11373054f218SWarner Losh int 11383054f218SWarner Losh main(int argc, char **argv) 11393054f218SWarner Losh { 11403054f218SWarner Losh int ch; 11413054f218SWarner Losh 11429c180898SWarner Losh check_devd_enabled(); 1143*6d1014a3SAlan Somers while ((ch = getopt(argc, argv, "df:l:n")) != -1) { 11443054f218SWarner Losh switch (ch) { 11453054f218SWarner Losh case 'd': 11463054f218SWarner Losh dflag++; 11473054f218SWarner Losh break; 11488334958aSJoseph Koshy case 'f': 11498334958aSJoseph Koshy configfile = optarg; 11508334958aSJoseph Koshy break; 1151e1334f93SIan Lepore case 'l': 1152e1334f93SIan Lepore max_clients = MAX(1, strtoul(optarg, NULL, 0)); 1153e1334f93SIan Lepore break; 11545a882d77SWarner Losh case 'n': 11555a882d77SWarner Losh nflag++; 11565a882d77SWarner Losh break; 11573054f218SWarner Losh default: 11583054f218SWarner Losh usage(); 11593054f218SWarner Losh } 11603054f218SWarner Losh } 11613054f218SWarner Losh 11623054f218SWarner Losh cfg.parse(); 1163fd6a8f23SWarner Losh if (!dflag && nflag) { 11641a0cc6b1SPawel Jakub Dawidek cfg.open_pidfile(); 11653054f218SWarner Losh daemon(0, 0); 11661a0cc6b1SPawel Jakub Dawidek cfg.write_pidfile(); 1167fd6a8f23SWarner Losh } 1168f32a7686SWarner Losh signal(SIGPIPE, SIG_IGN); 11693054f218SWarner Losh signal(SIGHUP, gensighand); 11703054f218SWarner Losh signal(SIGINT, gensighand); 11713054f218SWarner Losh signal(SIGTERM, gensighand); 11723054f218SWarner Losh event_loop(); 11733054f218SWarner Losh return (0); 11743054f218SWarner Losh } 1175