1/*-
2 * Copyright (c) 2002-2003 M. Warner Losh.
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24 * SUCH DAMAGE.
25 *
26 * $FreeBSD$
27 */
28
29#ifndef DEVD_HH
30#define DEVD_HH
31
32class config;
33
34/**
35 * var_list is a collection of variables.  These collections of variables
36 * are stacked up and popped down for each event that we have to process.
37 * We have multiple levels so that we can push variables that are unique
38 * to the event in question, in addition to having global variables.  This
39 * allows for future flexibility.
40 */
41class var_list
42{
43public:
44	/** Set a variable in this var list.
45	 */
46	void set_variable(const std::string &var, const std::string &val);
47	/** Get the variable out of this, and no other, var_list.  If
48	 * no variable of %var is set, then %bogus will be returned.
49	 */
50	const std::string &get_variable(const std::string &var) const;
51	/** Is there a variable of %var set in thi stable?
52	 */
53	bool is_set(const std::string &var) const;
54	/** A completely bogus string.
55	 */
56	static const std::string bogus;
57	static const std::string nothing;
58private:
59	std::map<std::string, std::string> _vars;
60};
61
62/**
63 * eps is short for event_proc_single.  It is a single entry in an
64 * event_proc.  Each keyword needs its own subclass from eps.
65 */
66struct eps
67{
68public:
69	virtual ~eps() {}
70	/** Does this eps match the current config?
71	 */
72	virtual bool do_match(config &) = 0;
73	/** Perform some action for this eps.
74	 */
75	virtual bool do_action(config &) = 0;
76};
77
78/**
79 * match is the subclass used to match an individual variable.  Its
80 * actions are nops.
81 */
82class match : public eps
83{
84public:
85	match(config &, const char *var, const char *re);
86	virtual ~match();
87	virtual bool do_match(config &);
88	virtual bool do_action(config &) { return true; }
89private:
90	bool _inv;
91	std::string _var;
92	std::string _re;
93	regex_t _regex;
94};
95
96/**
97 * media is the subclass used to match an individual variable.  Its
98 * actions are nops.
99 */
100class media : public eps
101{
102public:
103	media(config &, const char *var, const char *type);
104	virtual ~media();
105	virtual bool do_match(config &);
106	virtual bool do_action(config &) { return true; }
107private:
108	std::string _var;
109	int _type;
110};
111
112/**
113 * action is used to fork a process.  It matches everything.
114 */
115class action : public eps
116{
117public:
118	action(const char *cmd);
119	virtual ~action();
120	virtual bool do_match(config &) { return true; }
121	virtual bool do_action(config &);
122private:
123	std::string _cmd;
124};
125
126struct event_proc
127{
128public:
129	event_proc();
130	virtual ~event_proc();
131	int get_priority() const { return (_prio); }
132	void set_priority(int prio) { _prio = prio; }
133	void add(eps *);
134	bool matches(config &) const;
135	bool run(config &) const;
136private:
137	int _prio;
138	std::vector<eps *> _epsvec;
139};
140
141class config
142{
143public:
144	config() { push_var_table(); }
145	virtual ~config() { reset(); }
146	void add_attach(int, event_proc *);
147	void add_detach(int, event_proc *);
148	void add_directory(const char *);
149	void add_nomatch(int, event_proc *);
150	void add_notify(int, event_proc *);
151	void set_pidfile(const char *);
152	void reset();
153	void parse();
154	void close_pidfile();
155	void open_pidfile();
156	void write_pidfile();
157	void remove_pidfile();
158	void push_var_table();
159	void pop_var_table();
160	void set_variable(const char *var, const char *val);
161	const std::string &get_variable(const std::string &var);
162	const std::string expand_string(const char * var,
163	    const char * prepend = NULL, const char * append = NULL);
164	char *set_vars(char *);
165	void find_and_execute(char);
166protected:
167	void sort_vector(std::vector<event_proc *> &);
168	void parse_one_file(const char *fn);
169	void parse_files_in_dir(const char *dirname);
170	void expand_one(const char *&src, std::string &dst);
171	bool is_id_char(char) const;
172	bool chop_var(char *&buffer, char *&lhs, char *&rhs) const;
173private:
174	std::vector<std::string> _dir_list;
175	std::string _pidfile;
176	std::vector<var_list *> _var_list_table;
177	std::vector<event_proc *> _attach_list;
178	std::vector<event_proc *> _detach_list;
179	std::vector<event_proc *> _nomatch_list;
180	std::vector<event_proc *> _notify_list;
181};
182
183#endif /* DEVD_HH */
184