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> 82*967ecf4aSAlan Somers #include <cstdarg> 83d005340fSEitan Adler 843054f218SWarner Losh #include <dirent.h> 853054f218SWarner Losh #include <err.h> 863054f218SWarner Losh #include <fcntl.h> 871a0cc6b1SPawel Jakub Dawidek #include <libutil.h> 88b8f92ce4SWarner Losh #include <paths.h> 89e1334f93SIan Lepore #include <poll.h> 905102ef84SWarner Losh #include <regex.h> 916d1014a3SAlan Somers #include <syslog.h> 923054f218SWarner Losh #include <unistd.h> 933054f218SWarner Losh 948b823408SWarner Losh #include <algorithm> 953054f218SWarner Losh #include <map> 963054f218SWarner Losh #include <string> 97251bedd1SWarner Losh #include <list> 983054f218SWarner Losh #include <vector> 993054f218SWarner Losh 1006aeeca8eSWarner Losh #include "devd.h" /* C compatible definitions */ 1016aeeca8eSWarner Losh #include "devd.hh" /* C++ class definitions */ 1023054f218SWarner Losh 103251bedd1SWarner Losh #define PIPE "/var/run/devd.pipe" 1043054f218SWarner Losh #define CF "/etc/devd.conf" 1059c180898SWarner Losh #define SYSCTL "hw.bus.devctl_disable" 1063054f218SWarner Losh 1073054f218SWarner Losh using namespace std; 1083054f218SWarner Losh 1093054f218SWarner Losh extern FILE *yyin; 1103054f218SWarner Losh extern int lineno; 1113054f218SWarner Losh 112842ccec5SWarner Losh static const char notify = '!'; 1138b823408SWarner Losh static const char nomatch = '?'; 1148b823408SWarner Losh static const char attach = '+'; 1158b823408SWarner Losh static const char detach = '-'; 1168b823408SWarner Losh 1171a0cc6b1SPawel Jakub Dawidek static struct pidfh *pfh; 1181a0cc6b1SPawel Jakub Dawidek 1193054f218SWarner Losh int dflag; 120812f53a7SWarner Losh int nflag; 121be685e01SAlan Somers static unsigned total_events = 0; 122be685e01SAlan Somers static volatile sig_atomic_t got_siginfo = 0; 123fcdcaa88SEitan Adler static volatile sig_atomic_t romeo_must_die = 0; 1243054f218SWarner Losh 1258334958aSJoseph Koshy static const char *configfile = CF; 1268334958aSJoseph Koshy 1276d1014a3SAlan Somers static void devdlog(int priority, const char* message, ...); 1283054f218SWarner Losh static void event_loop(void); 1293054f218SWarner Losh static void usage(void); 1303054f218SWarner Losh 1318b823408SWarner Losh template <class T> void 1328b823408SWarner Losh delete_and_clear(vector<T *> &v) 1338b823408SWarner Losh { 1348b823408SWarner Losh typename vector<T *>::const_iterator i; 1353054f218SWarner Losh 136edee691dSEitan Adler for (i = v.begin(); i != v.end(); ++i) 1378b823408SWarner Losh delete *i; 1388b823408SWarner Losh v.clear(); 1398b823408SWarner Losh } 1408b823408SWarner Losh 1413054f218SWarner Losh config cfg; 1423054f218SWarner Losh 1433054f218SWarner Losh event_proc::event_proc() : _prio(-1) 1443054f218SWarner Losh { 1455dfc0f6cSIan Lepore _epsvec.reserve(4); 1463054f218SWarner Losh } 1473054f218SWarner Losh 1483054f218SWarner Losh event_proc::~event_proc() 1493054f218SWarner Losh { 15013f8bb71SAlexander Nedotsukov delete_and_clear(_epsvec); 1513054f218SWarner Losh } 1523054f218SWarner Losh 1533054f218SWarner Losh void 1543054f218SWarner Losh event_proc::add(eps *eps) 1553054f218SWarner Losh { 1563054f218SWarner Losh _epsvec.push_back(eps); 1573054f218SWarner Losh } 1583054f218SWarner Losh 1593054f218SWarner Losh bool 160ef370346SEitan Adler event_proc::matches(config &c) const 1613054f218SWarner Losh { 1623054f218SWarner Losh vector<eps *>::const_iterator i; 1633054f218SWarner Losh 164edee691dSEitan Adler for (i = _epsvec.begin(); i != _epsvec.end(); ++i) 1653054f218SWarner Losh if (!(*i)->do_match(c)) 1663054f218SWarner Losh return (false); 1673054f218SWarner Losh return (true); 1683054f218SWarner Losh } 1693054f218SWarner Losh 1703054f218SWarner Losh bool 171ef370346SEitan Adler event_proc::run(config &c) const 1723054f218SWarner Losh { 1733054f218SWarner Losh vector<eps *>::const_iterator i; 1743054f218SWarner Losh 175edee691dSEitan Adler for (i = _epsvec.begin(); i != _epsvec.end(); ++i) 1763054f218SWarner Losh if (!(*i)->do_action(c)) 1773054f218SWarner Losh return (false); 1783054f218SWarner Losh return (true); 1793054f218SWarner Losh } 1803054f218SWarner Losh 1813054f218SWarner Losh action::action(const char *cmd) 1823054f218SWarner Losh : _cmd(cmd) 1833054f218SWarner Losh { 1843054f218SWarner Losh // nothing 1853054f218SWarner Losh } 1863054f218SWarner Losh 1873054f218SWarner Losh action::~action() 1883054f218SWarner Losh { 1893054f218SWarner Losh // nothing 1903054f218SWarner Losh } 1913054f218SWarner Losh 192b8f92ce4SWarner Losh static int 193b8f92ce4SWarner Losh my_system(const char *command) 194b8f92ce4SWarner Losh { 195b8f92ce4SWarner Losh pid_t pid, savedpid; 196b8f92ce4SWarner Losh int pstat; 197b8f92ce4SWarner Losh struct sigaction ign, intact, quitact; 198b8f92ce4SWarner Losh sigset_t newsigblock, oldsigblock; 199b8f92ce4SWarner Losh 200b8f92ce4SWarner Losh if (!command) /* just checking... */ 201b8f92ce4SWarner Losh return (1); 202b8f92ce4SWarner Losh 203b8f92ce4SWarner Losh /* 204b8f92ce4SWarner Losh * Ignore SIGINT and SIGQUIT, block SIGCHLD. Remember to save 205b8f92ce4SWarner Losh * existing signal dispositions. 206b8f92ce4SWarner Losh */ 207b8f92ce4SWarner Losh ign.sa_handler = SIG_IGN; 208b8f92ce4SWarner Losh ::sigemptyset(&ign.sa_mask); 209b8f92ce4SWarner Losh ign.sa_flags = 0; 210b8f92ce4SWarner Losh ::sigaction(SIGINT, &ign, &intact); 211b8f92ce4SWarner Losh ::sigaction(SIGQUIT, &ign, &quitact); 212b8f92ce4SWarner Losh ::sigemptyset(&newsigblock); 213b8f92ce4SWarner Losh ::sigaddset(&newsigblock, SIGCHLD); 214b8f92ce4SWarner Losh ::sigprocmask(SIG_BLOCK, &newsigblock, &oldsigblock); 215b8f92ce4SWarner Losh switch (pid = ::fork()) { 216b8f92ce4SWarner Losh case -1: /* error */ 217b8f92ce4SWarner Losh break; 218b8f92ce4SWarner Losh case 0: /* child */ 219b8f92ce4SWarner Losh /* 220b8f92ce4SWarner Losh * Restore original signal dispositions and exec the command. 221b8f92ce4SWarner Losh */ 222b8f92ce4SWarner Losh ::sigaction(SIGINT, &intact, NULL); 223b8f92ce4SWarner Losh ::sigaction(SIGQUIT, &quitact, NULL); 224b8f92ce4SWarner Losh ::sigprocmask(SIG_SETMASK, &oldsigblock, NULL); 225b8f92ce4SWarner Losh /* 226b8f92ce4SWarner Losh * Close the PID file, and all other open descriptors. 227b8f92ce4SWarner Losh * Inherit std{in,out,err} only. 228b8f92ce4SWarner Losh */ 229b8f92ce4SWarner Losh cfg.close_pidfile(); 230b8f92ce4SWarner Losh ::closefrom(3); 231b8f92ce4SWarner Losh ::execl(_PATH_BSHELL, "sh", "-c", command, (char *)NULL); 232b8f92ce4SWarner Losh ::_exit(127); 233b8f92ce4SWarner Losh default: /* parent */ 234b8f92ce4SWarner Losh savedpid = pid; 235b8f92ce4SWarner Losh do { 236b8f92ce4SWarner Losh pid = ::wait4(savedpid, &pstat, 0, (struct rusage *)0); 237b8f92ce4SWarner Losh } while (pid == -1 && errno == EINTR); 238b8f92ce4SWarner Losh break; 239b8f92ce4SWarner Losh } 240b8f92ce4SWarner Losh ::sigaction(SIGINT, &intact, NULL); 241b8f92ce4SWarner Losh ::sigaction(SIGQUIT, &quitact, NULL); 242b8f92ce4SWarner Losh ::sigprocmask(SIG_SETMASK, &oldsigblock, NULL); 243b8f92ce4SWarner Losh return (pid == -1 ? -1 : pstat); 244b8f92ce4SWarner Losh } 245b8f92ce4SWarner Losh 2463054f218SWarner Losh bool 2475102ef84SWarner Losh action::do_action(config &c) 2483054f218SWarner Losh { 2495dfc0f6cSIan Lepore string s = c.expand_string(_cmd.c_str()); 2506d1014a3SAlan Somers devdlog(LOG_NOTICE, "Executing '%s'\n", s.c_str()); 251b8f92ce4SWarner Losh my_system(s.c_str()); 2523054f218SWarner Losh return (true); 2533054f218SWarner Losh } 2543054f218SWarner Losh 2555dfc0f6cSIan Lepore match::match(config &c, const char *var, const char *re) : 2565dfc0f6cSIan Lepore _inv(re[0] == '!'), 2575dfc0f6cSIan Lepore _var(var), 2585dfc0f6cSIan Lepore _re(c.expand_string(_inv ? re + 1 : re, "^", "$")) 2593054f218SWarner Losh { 2609cb4a7bdSWarner Losh regcomp(&_regex, _re.c_str(), REG_EXTENDED | REG_NOSUB | REG_ICASE); 2613054f218SWarner Losh } 2623054f218SWarner Losh 2633054f218SWarner Losh match::~match() 2643054f218SWarner Losh { 2655102ef84SWarner Losh regfree(&_regex); 2663054f218SWarner Losh } 2673054f218SWarner Losh 2683054f218SWarner Losh bool 2695102ef84SWarner Losh match::do_match(config &c) 2703054f218SWarner Losh { 2715bf08422SUlf Lilleengen const string &value = c.get_variable(_var); 2725102ef84SWarner Losh bool retval; 2735102ef84SWarner Losh 2746d1014a3SAlan Somers /* 2756d1014a3SAlan Somers * This function gets called WAY too often to justify calling syslog() 2766d1014a3SAlan Somers * each time, even at LOG_DEBUG. Because if syslogd isn't running, it 2776d1014a3SAlan Somers * can consume excessive amounts of systime inside of connect(). Only 2786d1014a3SAlan Somers * log when we're in -d mode. 2796d1014a3SAlan Somers */ 2806d1014a3SAlan Somers if (dflag) { 2816d1014a3SAlan Somers devdlog(LOG_DEBUG, "Testing %s=%s against %s, invert=%d\n", 2820321b694SHiroki Sato _var.c_str(), value.c_str(), _re.c_str(), _inv); 2836d1014a3SAlan Somers } 2848b823408SWarner Losh 2855102ef84SWarner Losh retval = (regexec(&_regex, value.c_str(), 0, NULL, 0) == 0); 2860321b694SHiroki Sato if (_inv == 1) 2870321b694SHiroki Sato retval = (retval == 0) ? 1 : 0; 2880321b694SHiroki Sato 2890285e9b1SAlan Somers return (retval); 2903054f218SWarner Losh } 2913054f218SWarner Losh 292cd70782bSWarner Losh #include <sys/sockio.h> 293cd70782bSWarner Losh #include <net/if.h> 294cd70782bSWarner Losh #include <net/if_media.h> 295cd70782bSWarner Losh 29694d2d4ebSBrooks Davis media::media(config &, const char *var, const char *type) 297cd70782bSWarner Losh : _var(var), _type(-1) 298cd70782bSWarner Losh { 299cd70782bSWarner Losh static struct ifmedia_description media_types[] = { 300cd70782bSWarner Losh { IFM_ETHER, "Ethernet" }, 301cd70782bSWarner Losh { IFM_TOKEN, "Tokenring" }, 302cd70782bSWarner Losh { IFM_FDDI, "FDDI" }, 303cd70782bSWarner Losh { IFM_IEEE80211, "802.11" }, 304cd70782bSWarner Losh { IFM_ATM, "ATM" }, 305cd70782bSWarner Losh { -1, "unknown" }, 306cd70782bSWarner Losh { 0, NULL }, 307cd70782bSWarner Losh }; 308edee691dSEitan Adler for (int i = 0; media_types[i].ifmt_string != NULL; ++i) 309cd70782bSWarner Losh if (strcasecmp(type, media_types[i].ifmt_string) == 0) { 310cd70782bSWarner Losh _type = media_types[i].ifmt_word; 311cd70782bSWarner Losh break; 312cd70782bSWarner Losh } 313cd70782bSWarner Losh } 314cd70782bSWarner Losh 315cd70782bSWarner Losh media::~media() 316cd70782bSWarner Losh { 317cd70782bSWarner Losh } 318cd70782bSWarner Losh 319cd70782bSWarner Losh bool 320cd70782bSWarner Losh media::do_match(config &c) 321cd70782bSWarner Losh { 32209c47e76SWarner Losh string value; 323cd70782bSWarner Losh struct ifmediareq ifmr; 324cd70782bSWarner Losh bool retval; 325cd70782bSWarner Losh int s; 326cd70782bSWarner Losh 32709c47e76SWarner Losh // Since we can be called from both a device attach/detach 32809c47e76SWarner Losh // context where device-name is defined and what we want, 32909c47e76SWarner Losh // as well as from a link status context, where subsystem is 33009c47e76SWarner Losh // the name of interest, first try device-name and fall back 33109c47e76SWarner Losh // to subsystem if none exists. 33209c47e76SWarner Losh value = c.get_variable("device-name"); 333a6393aa1SEitan Adler if (value.empty()) 334de02eec8SWarner Losh value = c.get_variable("subsystem"); 3356d1014a3SAlan Somers devdlog(LOG_DEBUG, "Testing media type of %s against 0x%x\n", 336cd70782bSWarner Losh value.c_str(), _type); 337cd70782bSWarner Losh 338cd70782bSWarner Losh retval = false; 339cd70782bSWarner Losh 340cd70782bSWarner Losh s = socket(PF_INET, SOCK_DGRAM, 0); 341cd70782bSWarner Losh if (s >= 0) { 342cd70782bSWarner Losh memset(&ifmr, 0, sizeof(ifmr)); 343cd70782bSWarner Losh strncpy(ifmr.ifm_name, value.c_str(), sizeof(ifmr.ifm_name)); 344cd70782bSWarner Losh 345cd70782bSWarner Losh if (ioctl(s, SIOCGIFMEDIA, (caddr_t)&ifmr) >= 0 && 346cd70782bSWarner Losh ifmr.ifm_status & IFM_AVALID) { 3476d1014a3SAlan Somers devdlog(LOG_DEBUG, "%s has media type 0x%x\n", 348cd70782bSWarner Losh value.c_str(), IFM_TYPE(ifmr.ifm_active)); 349cd70782bSWarner Losh retval = (IFM_TYPE(ifmr.ifm_active) == _type); 350cd70782bSWarner Losh } else if (_type == -1) { 3516d1014a3SAlan Somers devdlog(LOG_DEBUG, "%s has unknown media type\n", 352cd70782bSWarner Losh value.c_str()); 353cd70782bSWarner Losh retval = true; 354cd70782bSWarner Losh } 355cd70782bSWarner Losh close(s); 356cd70782bSWarner Losh } 357cd70782bSWarner Losh 3580285e9b1SAlan Somers return (retval); 359cd70782bSWarner Losh } 360cd70782bSWarner Losh 3613054f218SWarner Losh const string var_list::bogus = "_$_$_$_$_B_O_G_U_S_$_$_$_$_"; 3623054f218SWarner Losh const string var_list::nothing = ""; 3633054f218SWarner Losh 3643054f218SWarner Losh const string & 3653054f218SWarner Losh var_list::get_variable(const string &var) const 3663054f218SWarner Losh { 3673054f218SWarner Losh map<string, string>::const_iterator i; 3683054f218SWarner Losh 3693054f218SWarner Losh i = _vars.find(var); 3703054f218SWarner Losh if (i == _vars.end()) 3718b823408SWarner Losh return (var_list::bogus); 3723054f218SWarner Losh return (i->second); 3733054f218SWarner Losh } 3743054f218SWarner Losh 3753054f218SWarner Losh bool 3763054f218SWarner Losh var_list::is_set(const string &var) const 3773054f218SWarner Losh { 3783054f218SWarner Losh return (_vars.find(var) != _vars.end()); 3793054f218SWarner Losh } 3803054f218SWarner Losh 3813054f218SWarner Losh void 3823054f218SWarner Losh var_list::set_variable(const string &var, const string &val) 3833054f218SWarner Losh { 3846d1014a3SAlan Somers /* 3856d1014a3SAlan Somers * This function gets called WAY too often to justify calling syslog() 3866d1014a3SAlan Somers * each time, even at LOG_DEBUG. Because if syslogd isn't running, it 3876d1014a3SAlan Somers * can consume excessive amounts of systime inside of connect(). Only 3886d1014a3SAlan Somers * log when we're in -d mode. 3896d1014a3SAlan Somers */ 3906d1014a3SAlan Somers if (dflag) 3916d1014a3SAlan Somers devdlog(LOG_DEBUG, "setting %s=%s\n", var.c_str(), val.c_str()); 3923054f218SWarner Losh _vars[var] = val; 3933054f218SWarner Losh } 3943054f218SWarner Losh 3953054f218SWarner Losh void 3963054f218SWarner Losh config::reset(void) 3973054f218SWarner Losh { 3983054f218SWarner Losh _dir_list.clear(); 3998b823408SWarner Losh delete_and_clear(_var_list_table); 4008b823408SWarner Losh delete_and_clear(_attach_list); 4018b823408SWarner Losh delete_and_clear(_detach_list); 4028b823408SWarner Losh delete_and_clear(_nomatch_list); 403842ccec5SWarner Losh delete_and_clear(_notify_list); 4043054f218SWarner Losh } 4053054f218SWarner Losh 4063054f218SWarner Losh void 4073054f218SWarner Losh config::parse_one_file(const char *fn) 4083054f218SWarner Losh { 4096d1014a3SAlan Somers devdlog(LOG_DEBUG, "Parsing %s\n", fn); 4103054f218SWarner Losh yyin = fopen(fn, "r"); 4113054f218SWarner Losh if (yyin == NULL) 4123054f218SWarner Losh err(1, "Cannot open config file %s", fn); 4134a3050fcSMaxim Konovalov lineno = 1; 4143054f218SWarner Losh if (yyparse() != 0) 4153054f218SWarner Losh errx(1, "Cannot parse %s at line %d", fn, lineno); 4163054f218SWarner Losh fclose(yyin); 4173054f218SWarner Losh } 4183054f218SWarner Losh 4193054f218SWarner Losh void 4203054f218SWarner Losh config::parse_files_in_dir(const char *dirname) 4213054f218SWarner Losh { 4223054f218SWarner Losh DIR *dirp; 4233054f218SWarner Losh struct dirent *dp; 4243054f218SWarner Losh char path[PATH_MAX]; 4253054f218SWarner Losh 4266d1014a3SAlan Somers devdlog(LOG_DEBUG, "Parsing files in %s\n", dirname); 4273054f218SWarner Losh dirp = opendir(dirname); 4283054f218SWarner Losh if (dirp == NULL) 4293054f218SWarner Losh return; 4303054f218SWarner Losh readdir(dirp); /* Skip . */ 4313054f218SWarner Losh readdir(dirp); /* Skip .. */ 4323054f218SWarner Losh while ((dp = readdir(dirp)) != NULL) { 4333054f218SWarner Losh if (strcmp(dp->d_name + dp->d_namlen - 5, ".conf") == 0) { 4343054f218SWarner Losh snprintf(path, sizeof(path), "%s/%s", 4353054f218SWarner Losh dirname, dp->d_name); 4363054f218SWarner Losh parse_one_file(path); 4373054f218SWarner Losh } 4383054f218SWarner Losh } 4395983e891SKevin Lo closedir(dirp); 4403054f218SWarner Losh } 4413054f218SWarner Losh 4428b823408SWarner Losh class epv_greater { 4438b823408SWarner Losh public: 444ef370346SEitan Adler int operator()(event_proc *const&l1, event_proc *const&l2) const 4458b823408SWarner Losh { 4468b823408SWarner Losh return (l1->get_priority() > l2->get_priority()); 4478b823408SWarner Losh } 4488b823408SWarner Losh }; 4498b823408SWarner Losh 4508b823408SWarner Losh void 4518b823408SWarner Losh config::sort_vector(vector<event_proc *> &v) 4528b823408SWarner Losh { 4530c04d185SDimitry Andric stable_sort(v.begin(), v.end(), epv_greater()); 4548b823408SWarner Losh } 4558b823408SWarner Losh 4563054f218SWarner Losh void 4573054f218SWarner Losh config::parse(void) 4583054f218SWarner Losh { 4593054f218SWarner Losh vector<string>::const_iterator i; 4603054f218SWarner Losh 4618334958aSJoseph Koshy parse_one_file(configfile); 462edee691dSEitan Adler for (i = _dir_list.begin(); i != _dir_list.end(); ++i) 4633054f218SWarner Losh parse_files_in_dir((*i).c_str()); 4648b823408SWarner Losh sort_vector(_attach_list); 4658b823408SWarner Losh sort_vector(_detach_list); 4668b823408SWarner Losh sort_vector(_nomatch_list); 467842ccec5SWarner Losh sort_vector(_notify_list); 4683054f218SWarner Losh } 4693054f218SWarner Losh 4703054f218SWarner Losh void 4711a0cc6b1SPawel Jakub Dawidek config::open_pidfile() 4723054f218SWarner Losh { 4731a0cc6b1SPawel Jakub Dawidek pid_t otherpid; 4743054f218SWarner Losh 4755a3b1a3dSEitan Adler if (_pidfile.empty()) 4763054f218SWarner Losh return; 4771a0cc6b1SPawel Jakub Dawidek pfh = pidfile_open(_pidfile.c_str(), 0600, &otherpid); 4781a0cc6b1SPawel Jakub Dawidek if (pfh == NULL) { 4791a0cc6b1SPawel Jakub Dawidek if (errno == EEXIST) 4801a0cc6b1SPawel Jakub Dawidek errx(1, "devd already running, pid: %d", (int)otherpid); 4811a0cc6b1SPawel Jakub Dawidek warn("cannot open pid file"); 4821a0cc6b1SPawel Jakub Dawidek } 4831a0cc6b1SPawel Jakub Dawidek } 4841a0cc6b1SPawel Jakub Dawidek 4851a0cc6b1SPawel Jakub Dawidek void 4861a0cc6b1SPawel Jakub Dawidek config::write_pidfile() 4871a0cc6b1SPawel Jakub Dawidek { 4881a0cc6b1SPawel Jakub Dawidek 4891a0cc6b1SPawel Jakub Dawidek pidfile_write(pfh); 4901a0cc6b1SPawel Jakub Dawidek } 4911a0cc6b1SPawel Jakub Dawidek 4921a0cc6b1SPawel Jakub Dawidek void 493b8f92ce4SWarner Losh config::close_pidfile() 494b8f92ce4SWarner Losh { 495b8f92ce4SWarner Losh 496b8f92ce4SWarner Losh pidfile_close(pfh); 497b8f92ce4SWarner Losh } 498b8f92ce4SWarner Losh 499b8f92ce4SWarner Losh void 5001a0cc6b1SPawel Jakub Dawidek config::remove_pidfile() 5011a0cc6b1SPawel Jakub Dawidek { 5021a0cc6b1SPawel Jakub Dawidek 5031a0cc6b1SPawel Jakub Dawidek pidfile_remove(pfh); 5043054f218SWarner Losh } 5053054f218SWarner Losh 5063054f218SWarner Losh void 5073054f218SWarner Losh config::add_attach(int prio, event_proc *p) 5083054f218SWarner Losh { 5093054f218SWarner Losh p->set_priority(prio); 5103054f218SWarner Losh _attach_list.push_back(p); 5113054f218SWarner Losh } 5123054f218SWarner Losh 5133054f218SWarner Losh void 5143054f218SWarner Losh config::add_detach(int prio, event_proc *p) 5153054f218SWarner Losh { 5163054f218SWarner Losh p->set_priority(prio); 5173054f218SWarner Losh _detach_list.push_back(p); 5183054f218SWarner Losh } 5193054f218SWarner Losh 5203054f218SWarner Losh void 5213054f218SWarner Losh config::add_directory(const char *dir) 5223054f218SWarner Losh { 5233054f218SWarner Losh _dir_list.push_back(string(dir)); 5243054f218SWarner Losh } 5253054f218SWarner Losh 5263054f218SWarner Losh void 5273054f218SWarner Losh config::add_nomatch(int prio, event_proc *p) 5283054f218SWarner Losh { 5293054f218SWarner Losh p->set_priority(prio); 5303054f218SWarner Losh _nomatch_list.push_back(p); 5313054f218SWarner Losh } 5323054f218SWarner Losh 5333054f218SWarner Losh void 534842ccec5SWarner Losh config::add_notify(int prio, event_proc *p) 535842ccec5SWarner Losh { 536842ccec5SWarner Losh p->set_priority(prio); 537842ccec5SWarner Losh _notify_list.push_back(p); 538842ccec5SWarner Losh } 539842ccec5SWarner Losh 540842ccec5SWarner Losh void 5413054f218SWarner Losh config::set_pidfile(const char *fn) 5423054f218SWarner Losh { 5435a3b1a3dSEitan Adler _pidfile = fn; 5443054f218SWarner Losh } 5453054f218SWarner Losh 5463054f218SWarner Losh void 5473054f218SWarner Losh config::push_var_table() 5483054f218SWarner Losh { 5493054f218SWarner Losh var_list *vl; 5503054f218SWarner Losh 5513054f218SWarner Losh vl = new var_list(); 5523054f218SWarner Losh _var_list_table.push_back(vl); 5536d1014a3SAlan Somers devdlog(LOG_DEBUG, "Pushing table\n"); 5543054f218SWarner Losh } 5553054f218SWarner Losh 5563054f218SWarner Losh void 5573054f218SWarner Losh config::pop_var_table() 5583054f218SWarner Losh { 5593054f218SWarner Losh delete _var_list_table.back(); 5603054f218SWarner Losh _var_list_table.pop_back(); 5616d1014a3SAlan Somers devdlog(LOG_DEBUG, "Popping table\n"); 5623054f218SWarner Losh } 5633054f218SWarner Losh 5643054f218SWarner Losh void 5653054f218SWarner Losh config::set_variable(const char *var, const char *val) 5663054f218SWarner Losh { 5673054f218SWarner Losh _var_list_table.back()->set_variable(var, val); 5683054f218SWarner Losh } 5693054f218SWarner Losh 5703054f218SWarner Losh const string & 5713054f218SWarner Losh config::get_variable(const string &var) 5723054f218SWarner Losh { 5733054f218SWarner Losh vector<var_list *>::reverse_iterator i; 5743054f218SWarner Losh 575edee691dSEitan Adler for (i = _var_list_table.rbegin(); i != _var_list_table.rend(); ++i) { 5763054f218SWarner Losh if ((*i)->is_set(var)) 5778b823408SWarner Losh return ((*i)->get_variable(var)); 5783054f218SWarner Losh } 5793054f218SWarner Losh return (var_list::nothing); 5803054f218SWarner Losh } 5813054f218SWarner Losh 5828b823408SWarner Losh bool 583ef370346SEitan Adler config::is_id_char(char ch) const 5848b823408SWarner Losh { 5858b823408SWarner Losh return (ch != '\0' && (isalpha(ch) || isdigit(ch) || ch == '_' || 5868b823408SWarner Losh ch == '-')); 5878b823408SWarner Losh } 5888b823408SWarner Losh 5895102ef84SWarner Losh void 5906cabd675SWarner Losh config::expand_one(const char *&src, string &dst) 5913054f218SWarner Losh { 5925102ef84SWarner Losh int count; 5935bf08422SUlf Lilleengen string buffer; 5945102ef84SWarner Losh 5958b823408SWarner Losh src++; 5965102ef84SWarner Losh // $$ -> $ 5975102ef84SWarner Losh if (*src == '$') { 5987d9e9c60SEitan Adler dst += *src++; 5995102ef84SWarner Losh return; 6005102ef84SWarner Losh } 6015102ef84SWarner Losh 6025102ef84SWarner Losh // $(foo) -> $(foo) 6038b823408SWarner Losh // Not sure if I want to support this or not, so for now we just pass 6048b823408SWarner Losh // it through. 6055102ef84SWarner Losh if (*src == '(') { 6067d9e9c60SEitan Adler dst += '$'; 6075102ef84SWarner Losh count = 1; 6086cabd675SWarner Losh /* If the string ends before ) is matched , return. */ 6096cabd675SWarner Losh while (count > 0 && *src) { 6105102ef84SWarner Losh if (*src == ')') 6115102ef84SWarner Losh count--; 6125102ef84SWarner Losh else if (*src == '(') 6135102ef84SWarner Losh count++; 6147d9e9c60SEitan Adler dst += *src++; 6155102ef84SWarner Losh } 6165102ef84SWarner Losh return; 6175102ef84SWarner Losh } 6185102ef84SWarner Losh 6194cb9d1beSEitan Adler // $[^A-Za-z] -> $\1 6205102ef84SWarner Losh if (!isalpha(*src)) { 6217d9e9c60SEitan Adler dst += '$'; 6227d9e9c60SEitan Adler dst += *src++; 6235102ef84SWarner Losh return; 6245102ef84SWarner Losh } 6255102ef84SWarner Losh 6265102ef84SWarner Losh // $var -> replace with value 6276cabd675SWarner Losh do { 6287d9e9c60SEitan Adler buffer += *src++; 62966ff6821SWarner Losh } while (is_id_char(*src)); 6305a3b1a3dSEitan Adler dst.append(get_variable(buffer)); 6315102ef84SWarner Losh } 6325102ef84SWarner Losh 6335102ef84SWarner Losh const string 6345dfc0f6cSIan Lepore config::expand_string(const char *src, const char *prepend, const char *append) 6355102ef84SWarner Losh { 6365dfc0f6cSIan Lepore const char *var_at; 6376cabd675SWarner Losh string dst; 6385102ef84SWarner Losh 6395dfc0f6cSIan Lepore /* 6405dfc0f6cSIan Lepore * 128 bytes is enough for 2427 of 2438 expansions that happen 6415dfc0f6cSIan Lepore * while parsing config files, as tested on 2013-01-30. 6425dfc0f6cSIan Lepore */ 6435dfc0f6cSIan Lepore dst.reserve(128); 6445dfc0f6cSIan Lepore 6455dfc0f6cSIan Lepore if (prepend != NULL) 6465dfc0f6cSIan Lepore dst = prepend; 6475dfc0f6cSIan Lepore 6485dfc0f6cSIan Lepore for (;;) { 6495dfc0f6cSIan Lepore var_at = strchr(src, '$'); 6505dfc0f6cSIan Lepore if (var_at == NULL) { 6515dfc0f6cSIan Lepore dst.append(src); 6525dfc0f6cSIan Lepore break; 6535102ef84SWarner Losh } 6545dfc0f6cSIan Lepore dst.append(src, var_at - src); 6555dfc0f6cSIan Lepore src = var_at; 6565dfc0f6cSIan Lepore expand_one(src, dst); 6575dfc0f6cSIan Lepore } 6585dfc0f6cSIan Lepore 6595dfc0f6cSIan Lepore if (append != NULL) 6605dfc0f6cSIan Lepore dst.append(append); 6615102ef84SWarner Losh 6626cabd675SWarner Losh return (dst); 6633054f218SWarner Losh } 6643054f218SWarner Losh 6658b823408SWarner Losh bool 66611fd1366SEitan Adler config::chop_var(char *&buffer, char *&lhs, char *&rhs) const 6678b823408SWarner Losh { 6688b823408SWarner Losh char *walker; 6698b823408SWarner Losh 6708b823408SWarner Losh if (*buffer == '\0') 6718b823408SWarner Losh return (false); 6728b823408SWarner Losh walker = lhs = buffer; 6738b823408SWarner Losh while (is_id_char(*walker)) 6748b823408SWarner Losh walker++; 6758b823408SWarner Losh if (*walker != '=') 6768b823408SWarner Losh return (false); 6778b823408SWarner Losh walker++; // skip = 6788b823408SWarner Losh if (*walker == '"') { 6798b823408SWarner Losh walker++; // skip " 6808b823408SWarner Losh rhs = walker; 6818b823408SWarner Losh while (*walker && *walker != '"') 6828b823408SWarner Losh walker++; 6838b823408SWarner Losh if (*walker != '"') 6848b823408SWarner Losh return (false); 6858b823408SWarner Losh rhs[-2] = '\0'; 6868b823408SWarner Losh *walker++ = '\0'; 6878b823408SWarner Losh } else { 6888b823408SWarner Losh rhs = walker; 6898b823408SWarner Losh while (*walker && !isspace(*walker)) 6908b823408SWarner Losh walker++; 6918b823408SWarner Losh if (*walker != '\0') 6928b823408SWarner Losh *walker++ = '\0'; 6938b823408SWarner Losh rhs[-1] = '\0'; 6948b823408SWarner Losh } 695b1ee04b1SWarner Losh while (isspace(*walker)) 696b1ee04b1SWarner Losh walker++; 6978b823408SWarner Losh buffer = walker; 6988b823408SWarner Losh return (true); 6998b823408SWarner Losh } 7008b823408SWarner Losh 7018b823408SWarner Losh 7028b823408SWarner Losh char * 7038b823408SWarner Losh config::set_vars(char *buffer) 7048b823408SWarner Losh { 7058b823408SWarner Losh char *lhs; 7068b823408SWarner Losh char *rhs; 7078b823408SWarner Losh 7088b823408SWarner Losh while (1) { 7098b823408SWarner Losh if (!chop_var(buffer, lhs, rhs)) 7108b823408SWarner Losh break; 7118b823408SWarner Losh set_variable(lhs, rhs); 7128b823408SWarner Losh } 7138b823408SWarner Losh return (buffer); 7148b823408SWarner Losh } 7158b823408SWarner Losh 7168b823408SWarner Losh void 7178b823408SWarner Losh config::find_and_execute(char type) 7188b823408SWarner Losh { 7198b823408SWarner Losh vector<event_proc *> *l; 7208b823408SWarner Losh vector<event_proc *>::const_iterator i; 72194d2d4ebSBrooks Davis const char *s; 7228b823408SWarner Losh 7238b823408SWarner Losh switch (type) { 7248b823408SWarner Losh default: 7258b823408SWarner Losh return; 726842ccec5SWarner Losh case notify: 727842ccec5SWarner Losh l = &_notify_list; 728842ccec5SWarner Losh s = "notify"; 729842ccec5SWarner Losh break; 7308b823408SWarner Losh case nomatch: 7318b823408SWarner Losh l = &_nomatch_list; 7328b823408SWarner Losh s = "nomatch"; 7338b823408SWarner Losh break; 7348b823408SWarner Losh case attach: 7358b823408SWarner Losh l = &_attach_list; 7368b823408SWarner Losh s = "attach"; 7378b823408SWarner Losh break; 7388b823408SWarner Losh case detach: 7398b823408SWarner Losh l = &_detach_list; 7408b823408SWarner Losh s = "detach"; 7418b823408SWarner Losh break; 7428b823408SWarner Losh } 7436d1014a3SAlan Somers devdlog(LOG_DEBUG, "Processing %s event\n", s); 744edee691dSEitan Adler for (i = l->begin(); i != l->end(); ++i) { 7458b823408SWarner Losh if ((*i)->matches(*this)) { 7468b823408SWarner Losh (*i)->run(*this); 7478b823408SWarner Losh break; 7488b823408SWarner Losh } 7498b823408SWarner Losh } 7508b823408SWarner Losh 7518b823408SWarner Losh } 7528b823408SWarner Losh 7530285e9b1SAlan Somers 7543054f218SWarner Losh static void 7558b823408SWarner Losh process_event(char *buffer) 7563054f218SWarner Losh { 7573054f218SWarner Losh char type; 7583054f218SWarner Losh char *sp; 7593054f218SWarner Losh 7608b823408SWarner Losh sp = buffer + 1; 7616d1014a3SAlan Somers devdlog(LOG_DEBUG, "Processing event '%s'\n", buffer); 7623054f218SWarner Losh type = *buffer++; 7638b823408SWarner Losh cfg.push_var_table(); 7648b823408SWarner Losh // No match doesn't have a device, and the format is a little 7658b823408SWarner Losh // different, so handle it separately. 766842ccec5SWarner Losh switch (type) { 767842ccec5SWarner Losh case notify: 768842ccec5SWarner Losh sp = cfg.set_vars(sp); 769842ccec5SWarner Losh break; 770842ccec5SWarner Losh case nomatch: 771528d3f79SWarner Losh //? at location pnp-info on bus 772528d3f79SWarner Losh sp = strchr(sp, ' '); 773528d3f79SWarner Losh if (sp == NULL) 774528d3f79SWarner Losh return; /* Can't happen? */ 775528d3f79SWarner Losh *sp++ = '\0'; 7769b6cb520SWarner Losh while (isspace(*sp)) 7779b6cb520SWarner Losh sp++; 778842ccec5SWarner Losh if (strncmp(sp, "at ", 3) == 0) 779842ccec5SWarner Losh sp += 3; 780842ccec5SWarner Losh sp = cfg.set_vars(sp); 7819b6cb520SWarner Losh while (isspace(*sp)) 7829b6cb520SWarner Losh sp++; 783842ccec5SWarner Losh if (strncmp(sp, "on ", 3) == 0) 784842ccec5SWarner Losh cfg.set_variable("bus", sp + 3); 785842ccec5SWarner Losh break; 786842ccec5SWarner Losh case attach: /*FALLTHROUGH*/ 787842ccec5SWarner Losh case detach: 7888b823408SWarner Losh sp = strchr(sp, ' '); 7893054f218SWarner Losh if (sp == NULL) 7903054f218SWarner Losh return; /* Can't happen? */ 7918b823408SWarner Losh *sp++ = '\0'; 7928b823408SWarner Losh cfg.set_variable("device-name", buffer); 7939b6cb520SWarner Losh while (isspace(*sp)) 7949b6cb520SWarner Losh sp++; 7958b823408SWarner Losh if (strncmp(sp, "at ", 3) == 0) 7968b823408SWarner Losh sp += 3; 7978b823408SWarner Losh sp = cfg.set_vars(sp); 7989b6cb520SWarner Losh while (isspace(*sp)) 7999b6cb520SWarner Losh sp++; 8008b823408SWarner Losh if (strncmp(sp, "on ", 3) == 0) 8018b823408SWarner Losh cfg.set_variable("bus", sp + 3); 802842ccec5SWarner Losh break; 803b1ee04b1SWarner Losh } 804b1ee04b1SWarner Losh 8058b823408SWarner Losh cfg.find_and_execute(type); 8068b823408SWarner Losh cfg.pop_var_table(); 8073054f218SWarner Losh } 8083054f218SWarner Losh 809251bedd1SWarner Losh int 810251bedd1SWarner Losh create_socket(const char *name) 811251bedd1SWarner Losh { 812251bedd1SWarner Losh int fd, slen; 813251bedd1SWarner Losh struct sockaddr_un sun; 814251bedd1SWarner Losh 815251bedd1SWarner Losh if ((fd = socket(PF_LOCAL, SOCK_STREAM, 0)) < 0) 816251bedd1SWarner Losh err(1, "socket"); 817251bedd1SWarner Losh bzero(&sun, sizeof(sun)); 818251bedd1SWarner Losh sun.sun_family = AF_UNIX; 819251bedd1SWarner Losh strlcpy(sun.sun_path, name, sizeof(sun.sun_path)); 820251bedd1SWarner Losh slen = SUN_LEN(&sun); 821251bedd1SWarner Losh unlink(name); 822e60fa014SJoe Marcus Clarke if (fcntl(fd, F_SETFL, O_NONBLOCK) < 0) 823e60fa014SJoe Marcus Clarke err(1, "fcntl"); 8245d586d29SDimitry Andric if (::bind(fd, (struct sockaddr *) & sun, slen) < 0) 825251bedd1SWarner Losh err(1, "bind"); 826251bedd1SWarner Losh listen(fd, 4); 82783818929SJoe Marcus Clarke chown(name, 0, 0); /* XXX - root.wheel */ 828e60fa014SJoe Marcus Clarke chmod(name, 0666); 829251bedd1SWarner Losh return (fd); 830251bedd1SWarner Losh } 831251bedd1SWarner Losh 832e1334f93SIan Lepore unsigned int max_clients = 10; /* Default, can be overriden on cmdline. */ 833e1334f93SIan Lepore unsigned int num_clients; 834251bedd1SWarner Losh list<int> clients; 835251bedd1SWarner Losh 836251bedd1SWarner Losh void 837251bedd1SWarner Losh notify_clients(const char *data, int len) 838251bedd1SWarner Losh { 839e1334f93SIan Lepore list<int>::iterator i; 840251bedd1SWarner Losh 841e1334f93SIan Lepore /* 842e1334f93SIan Lepore * Deliver the data to all clients. Throw clients overboard at the 843e1334f93SIan Lepore * first sign of trouble. This reaps clients who've died or closed 844e1334f93SIan Lepore * their sockets, and also clients who are alive but failing to keep up 845e1334f93SIan Lepore * (or who are maliciously not reading, to consume buffer space in 846e1334f93SIan Lepore * kernel memory or tie up the limited number of available connections). 847e1334f93SIan Lepore */ 848e1334f93SIan Lepore for (i = clients.begin(); i != clients.end(); ) { 849e1334f93SIan Lepore if (write(*i, data, len) != len) { 850e1334f93SIan Lepore --num_clients; 851251bedd1SWarner Losh close(*i); 852e1334f93SIan Lepore i = clients.erase(i); 8536d1014a3SAlan Somers devdlog(LOG_WARNING, "notify_clients: write() failed; " 8546d1014a3SAlan Somers "dropping unresponsive client\n"); 855e1334f93SIan Lepore } else 856e1334f93SIan Lepore ++i; 857251bedd1SWarner Losh } 858251bedd1SWarner Losh } 859251bedd1SWarner Losh 860e1334f93SIan Lepore void 861e1334f93SIan Lepore check_clients(void) 862e1334f93SIan Lepore { 863e1334f93SIan Lepore int s; 864e1334f93SIan Lepore struct pollfd pfd; 865e1334f93SIan Lepore list<int>::iterator i; 866e1334f93SIan Lepore 867e1334f93SIan Lepore /* 868e1334f93SIan Lepore * Check all existing clients to see if any of them have disappeared. 869e1334f93SIan Lepore * Normally we reap clients when we get an error trying to send them an 870e1334f93SIan Lepore * event. This check eliminates the problem of an ever-growing list of 871e1334f93SIan Lepore * zombie clients because we're never writing to them on a system 872e1334f93SIan Lepore * without frequent device-change activity. 873e1334f93SIan Lepore */ 874e1334f93SIan Lepore pfd.events = 0; 875e1334f93SIan Lepore for (i = clients.begin(); i != clients.end(); ) { 876e1334f93SIan Lepore pfd.fd = *i; 877e1334f93SIan Lepore s = poll(&pfd, 1, 0); 878e1334f93SIan Lepore if ((s < 0 && s != EINTR ) || 879e1334f93SIan Lepore (s > 0 && (pfd.revents & POLLHUP))) { 880e1334f93SIan Lepore --num_clients; 881e1334f93SIan Lepore close(*i); 882e1334f93SIan Lepore i = clients.erase(i); 8836d1014a3SAlan Somers devdlog(LOG_NOTICE, "check_clients: " 8846d1014a3SAlan Somers "dropping disconnected client\n"); 885e1334f93SIan Lepore } else 886e1334f93SIan Lepore ++i; 887e1334f93SIan Lepore } 888251bedd1SWarner Losh } 889251bedd1SWarner Losh 890251bedd1SWarner Losh void 891251bedd1SWarner Losh new_client(int fd) 892251bedd1SWarner Losh { 893251bedd1SWarner Losh int s; 894251bedd1SWarner Losh 895e1334f93SIan Lepore /* 896e1334f93SIan Lepore * First go reap any zombie clients, then accept the connection, and 897e1334f93SIan Lepore * shut down the read side to stop clients from consuming kernel memory 898e1334f93SIan Lepore * by sending large buffers full of data we'll never read. 899e1334f93SIan Lepore */ 900e1334f93SIan Lepore check_clients(); 901251bedd1SWarner Losh s = accept(fd, NULL, NULL); 902e1334f93SIan Lepore if (s != -1) { 903e1334f93SIan Lepore shutdown(s, SHUT_RD); 904251bedd1SWarner Losh clients.push_back(s); 905e1334f93SIan Lepore ++num_clients; 906e1334f93SIan Lepore } 907251bedd1SWarner Losh } 908251bedd1SWarner Losh 9093054f218SWarner Losh static void 9103054f218SWarner Losh event_loop(void) 9113054f218SWarner Losh { 9123054f218SWarner Losh int rv; 9133054f218SWarner Losh int fd; 9143054f218SWarner Losh char buffer[DEVCTL_MAXBUF]; 9155a882d77SWarner Losh int once = 0; 916251bedd1SWarner Losh int server_fd, max_fd; 917e1334f93SIan Lepore int accepting; 9185a882d77SWarner Losh timeval tv; 9195a882d77SWarner Losh fd_set fds; 9203054f218SWarner Losh 921901610f4SPawel Jakub Dawidek fd = open(PATH_DEVCTL, O_RDONLY | O_CLOEXEC); 9223054f218SWarner Losh if (fd == -1) 923251bedd1SWarner Losh err(1, "Can't open devctl device %s", PATH_DEVCTL); 924251bedd1SWarner Losh server_fd = create_socket(PIPE); 925e1334f93SIan Lepore accepting = 1; 926251bedd1SWarner Losh max_fd = max(fd, server_fd) + 1; 92767ae3bf1SEitan Adler while (!romeo_must_die) { 9285a882d77SWarner Losh if (!once && !dflag && !nflag) { 9295a882d77SWarner Losh // Check to see if we have any events pending. 9305a882d77SWarner Losh tv.tv_sec = 0; 9315a882d77SWarner Losh tv.tv_usec = 0; 9325a882d77SWarner Losh FD_ZERO(&fds); 9335a882d77SWarner Losh FD_SET(fd, &fds); 9345a882d77SWarner Losh rv = select(fd + 1, &fds, &fds, &fds, &tv); 9355a882d77SWarner Losh // No events -> we've processed all pending events 936f3059505SWarner Losh if (rv == 0) { 9376d1014a3SAlan Somers devdlog(LOG_DEBUG, "Calling daemon\n"); 9381a0cc6b1SPawel Jakub Dawidek cfg.remove_pidfile(); 9391a0cc6b1SPawel Jakub Dawidek cfg.open_pidfile(); 9405a882d77SWarner Losh daemon(0, 0); 9411a0cc6b1SPawel Jakub Dawidek cfg.write_pidfile(); 9425a882d77SWarner Losh once++; 9435a882d77SWarner Losh } 9445a882d77SWarner Losh } 945e1334f93SIan Lepore /* 946e1334f93SIan Lepore * When we've already got the max number of clients, stop 947e1334f93SIan Lepore * accepting new connections (don't put server_fd in the set), 948e1334f93SIan Lepore * shrink the accept() queue to reject connections quickly, and 949e1334f93SIan Lepore * poll the existing clients more often, so that we notice more 950e1334f93SIan Lepore * quickly when any of them disappear to free up client slots. 951e1334f93SIan Lepore */ 952251bedd1SWarner Losh FD_ZERO(&fds); 953251bedd1SWarner Losh FD_SET(fd, &fds); 954e1334f93SIan Lepore if (num_clients < max_clients) { 955e1334f93SIan Lepore if (!accepting) { 956e1334f93SIan Lepore listen(server_fd, max_clients); 957e1334f93SIan Lepore accepting = 1; 958e1334f93SIan Lepore } 959251bedd1SWarner Losh FD_SET(server_fd, &fds); 960e1334f93SIan Lepore tv.tv_sec = 60; 961e1334f93SIan Lepore tv.tv_usec = 0; 962e1334f93SIan Lepore } else { 963e1334f93SIan Lepore if (accepting) { 964e1334f93SIan Lepore listen(server_fd, 0); 965e1334f93SIan Lepore accepting = 0; 966e1334f93SIan Lepore } 967e1334f93SIan Lepore tv.tv_sec = 2; 968e1334f93SIan Lepore tv.tv_usec = 0; 969e1334f93SIan Lepore } 970e1334f93SIan Lepore rv = select(max_fd, &fds, NULL, NULL, &tv); 971be685e01SAlan Somers if (got_siginfo) { 972be685e01SAlan Somers devdlog(LOG_INFO, "Events received so far=%ld\n", 973be685e01SAlan Somers total_events); 974be685e01SAlan Somers got_siginfo = 0; 975be685e01SAlan Somers } 976251bedd1SWarner Losh if (rv == -1) { 977251bedd1SWarner Losh if (errno == EINTR) 978251bedd1SWarner Losh continue; 979251bedd1SWarner Losh err(1, "select"); 980e1334f93SIan Lepore } else if (rv == 0) 981e1334f93SIan Lepore check_clients(); 982251bedd1SWarner Losh if (FD_ISSET(fd, &fds)) { 9833054f218SWarner Losh rv = read(fd, buffer, sizeof(buffer) - 1); 9843054f218SWarner Losh if (rv > 0) { 985be685e01SAlan Somers total_events++; 9866d1014a3SAlan Somers if (rv == sizeof(buffer) - 1) { 9876d1014a3SAlan Somers devdlog(LOG_WARNING, "Warning: " 9886d1014a3SAlan Somers "available event data exceeded " 9896d1014a3SAlan Somers "buffer space\n"); 9906d1014a3SAlan Somers } 991251bedd1SWarner Losh notify_clients(buffer, rv); 9923054f218SWarner Losh buffer[rv] = '\0'; 9933054f218SWarner Losh while (buffer[--rv] == '\n') 9943054f218SWarner Losh buffer[rv] = '\0'; 9953054f218SWarner Losh process_event(buffer); 9963054f218SWarner Losh } else if (rv < 0) { 9973054f218SWarner Losh if (errno != EINTR) 9983054f218SWarner Losh break; 9993054f218SWarner Losh } else { 10003054f218SWarner Losh /* EOF */ 10013054f218SWarner Losh break; 10023054f218SWarner Losh } 10033054f218SWarner Losh } 1004251bedd1SWarner Losh if (FD_ISSET(server_fd, &fds)) 1005251bedd1SWarner Losh new_client(server_fd); 1006251bedd1SWarner Losh } 10073054f218SWarner Losh close(fd); 10083054f218SWarner Losh } 10090285e9b1SAlan Somers 10103054f218SWarner Losh /* 10113054f218SWarner Losh * functions that the parser uses. 10123054f218SWarner Losh */ 10133054f218SWarner Losh void 10143054f218SWarner Losh add_attach(int prio, event_proc *p) 10153054f218SWarner Losh { 10163054f218SWarner Losh cfg.add_attach(prio, p); 10173054f218SWarner Losh } 10183054f218SWarner Losh 10193054f218SWarner Losh void 10203054f218SWarner Losh add_detach(int prio, event_proc *p) 10213054f218SWarner Losh { 10223054f218SWarner Losh cfg.add_detach(prio, p); 10233054f218SWarner Losh } 10243054f218SWarner Losh 10253054f218SWarner Losh void 10263054f218SWarner Losh add_directory(const char *dir) 10273054f218SWarner Losh { 10283054f218SWarner Losh cfg.add_directory(dir); 10293054f218SWarner Losh free(const_cast<char *>(dir)); 10303054f218SWarner Losh } 10313054f218SWarner Losh 10323054f218SWarner Losh void 10333054f218SWarner Losh add_nomatch(int prio, event_proc *p) 10343054f218SWarner Losh { 10353054f218SWarner Losh cfg.add_nomatch(prio, p); 10363054f218SWarner Losh } 10373054f218SWarner Losh 1038842ccec5SWarner Losh void 1039842ccec5SWarner Losh add_notify(int prio, event_proc *p) 1040842ccec5SWarner Losh { 1041842ccec5SWarner Losh cfg.add_notify(prio, p); 1042842ccec5SWarner Losh } 1043842ccec5SWarner Losh 10443054f218SWarner Losh event_proc * 10453054f218SWarner Losh add_to_event_proc(event_proc *ep, eps *eps) 10463054f218SWarner Losh { 10473054f218SWarner Losh if (ep == NULL) 10483054f218SWarner Losh ep = new event_proc(); 10493054f218SWarner Losh ep->add(eps); 10503054f218SWarner Losh return (ep); 10513054f218SWarner Losh } 10523054f218SWarner Losh 10533054f218SWarner Losh eps * 10543054f218SWarner Losh new_action(const char *cmd) 10553054f218SWarner Losh { 10563054f218SWarner Losh eps *e = new action(cmd); 10573054f218SWarner Losh free(const_cast<char *>(cmd)); 10583054f218SWarner Losh return (e); 10593054f218SWarner Losh } 10603054f218SWarner Losh 10613054f218SWarner Losh eps * 10623054f218SWarner Losh new_match(const char *var, const char *re) 10633054f218SWarner Losh { 10645102ef84SWarner Losh eps *e = new match(cfg, var, re); 10653054f218SWarner Losh free(const_cast<char *>(var)); 10663054f218SWarner Losh free(const_cast<char *>(re)); 10673054f218SWarner Losh return (e); 10683054f218SWarner Losh } 10693054f218SWarner Losh 1070cd70782bSWarner Losh eps * 1071cd70782bSWarner Losh new_media(const char *var, const char *re) 1072cd70782bSWarner Losh { 1073cd70782bSWarner Losh eps *e = new media(cfg, var, re); 1074cd70782bSWarner Losh free(const_cast<char *>(var)); 1075cd70782bSWarner Losh free(const_cast<char *>(re)); 1076cd70782bSWarner Losh return (e); 1077cd70782bSWarner Losh } 1078cd70782bSWarner Losh 10793054f218SWarner Losh void 10803054f218SWarner Losh set_pidfile(const char *name) 10813054f218SWarner Losh { 10823054f218SWarner Losh cfg.set_pidfile(name); 10833054f218SWarner Losh free(const_cast<char *>(name)); 10843054f218SWarner Losh } 10853054f218SWarner Losh 10863054f218SWarner Losh void 10873054f218SWarner Losh set_variable(const char *var, const char *val) 10883054f218SWarner Losh { 10893054f218SWarner Losh cfg.set_variable(var, val); 10903054f218SWarner Losh free(const_cast<char *>(var)); 10913054f218SWarner Losh free(const_cast<char *>(val)); 10923054f218SWarner Losh } 10933054f218SWarner Losh 10940285e9b1SAlan Somers 10953054f218SWarner Losh 10963054f218SWarner Losh static void 10973054f218SWarner Losh gensighand(int) 10983054f218SWarner Losh { 1099fcdcaa88SEitan Adler romeo_must_die = 1; 11003054f218SWarner Losh } 11013054f218SWarner Losh 11026d1014a3SAlan Somers /* 1103be685e01SAlan Somers * SIGINFO handler. Will print useful statistics to the syslog or stderr 1104be685e01SAlan Somers * as appropriate 1105be685e01SAlan Somers */ 1106be685e01SAlan Somers static void 1107be685e01SAlan Somers siginfohand(int) 1108be685e01SAlan Somers { 1109be685e01SAlan Somers got_siginfo = 1; 1110be685e01SAlan Somers } 1111be685e01SAlan Somers 1112be685e01SAlan Somers /* 11136d1014a3SAlan Somers * Local logging function. Prints to syslog if we're daemonized; syslog 11146d1014a3SAlan Somers * otherwise. 11156d1014a3SAlan Somers */ 11166d1014a3SAlan Somers static void 11176d1014a3SAlan Somers devdlog(int priority, const char* fmt, ...) 11186d1014a3SAlan Somers { 11196d1014a3SAlan Somers va_list argp; 11206d1014a3SAlan Somers 11216d1014a3SAlan Somers va_start(argp, fmt); 11226d1014a3SAlan Somers if (dflag) 11236d1014a3SAlan Somers vfprintf(stderr, fmt, argp); 11246d1014a3SAlan Somers else 11256d1014a3SAlan Somers vsyslog(priority, fmt, argp); 11266d1014a3SAlan Somers va_end(argp); 11276d1014a3SAlan Somers } 11286d1014a3SAlan Somers 11293054f218SWarner Losh static void 11303054f218SWarner Losh usage() 11313054f218SWarner Losh { 11326d1014a3SAlan Somers fprintf(stderr, "usage: %s [-dn] [-l connlimit] [-f file]\n", 1133e1334f93SIan Lepore getprogname()); 11343054f218SWarner Losh exit(1); 11353054f218SWarner Losh } 11363054f218SWarner Losh 11379c180898SWarner Losh static void 11389c180898SWarner Losh check_devd_enabled() 11399c180898SWarner Losh { 11409c180898SWarner Losh int val = 0; 11419c180898SWarner Losh size_t len; 11429c180898SWarner Losh 11439c180898SWarner Losh len = sizeof(val); 114479f92e25SWarner Losh if (sysctlbyname(SYSCTL, &val, &len, NULL, 0) != 0) 11459c180898SWarner Losh errx(1, "devctl sysctl missing from kernel!"); 11469c180898SWarner Losh if (val) { 11479c180898SWarner Losh warnx("Setting " SYSCTL " to 0"); 11489c180898SWarner Losh val = 0; 11499c180898SWarner Losh sysctlbyname(SYSCTL, NULL, NULL, &val, sizeof(val)); 11509c180898SWarner Losh } 11519c180898SWarner Losh } 11529c180898SWarner Losh 11533054f218SWarner Losh /* 11543054f218SWarner Losh * main 11553054f218SWarner Losh */ 11563054f218SWarner Losh int 11573054f218SWarner Losh main(int argc, char **argv) 11583054f218SWarner Losh { 11593054f218SWarner Losh int ch; 11603054f218SWarner Losh 11619c180898SWarner Losh check_devd_enabled(); 11626d1014a3SAlan Somers while ((ch = getopt(argc, argv, "df:l:n")) != -1) { 11633054f218SWarner Losh switch (ch) { 11643054f218SWarner Losh case 'd': 11653054f218SWarner Losh dflag++; 11663054f218SWarner Losh break; 11678334958aSJoseph Koshy case 'f': 11688334958aSJoseph Koshy configfile = optarg; 11698334958aSJoseph Koshy break; 1170e1334f93SIan Lepore case 'l': 1171e1334f93SIan Lepore max_clients = MAX(1, strtoul(optarg, NULL, 0)); 1172e1334f93SIan Lepore break; 11735a882d77SWarner Losh case 'n': 11745a882d77SWarner Losh nflag++; 11755a882d77SWarner Losh break; 11763054f218SWarner Losh default: 11773054f218SWarner Losh usage(); 11783054f218SWarner Losh } 11793054f218SWarner Losh } 11803054f218SWarner Losh 11813054f218SWarner Losh cfg.parse(); 1182fd6a8f23SWarner Losh if (!dflag && nflag) { 11831a0cc6b1SPawel Jakub Dawidek cfg.open_pidfile(); 11843054f218SWarner Losh daemon(0, 0); 11851a0cc6b1SPawel Jakub Dawidek cfg.write_pidfile(); 1186fd6a8f23SWarner Losh } 1187f32a7686SWarner Losh signal(SIGPIPE, SIG_IGN); 11883054f218SWarner Losh signal(SIGHUP, gensighand); 11893054f218SWarner Losh signal(SIGINT, gensighand); 11903054f218SWarner Losh signal(SIGTERM, gensighand); 1191be685e01SAlan Somers signal(SIGINFO, siginfohand); 11923054f218SWarner Losh event_loop(); 11933054f218SWarner Losh return (0); 11943054f218SWarner Losh } 1195