Head

GCC Code Coverage Report

Directory:	./		Exec	Total	Coverage
File:	usr.sbin/dhcpd/db.c	Lines:	0	81	0.0 %
Date:	2017-11-13	Branches:	0	78	0.0 %


/*	$OpenBSD: db.c,v 1.18 2017/02/13 23:04:05 krw Exp $	*/

/*
 * Persistent database management routines for DHCPD.
 */

/*
 * Copyright (c) 1995, 1996 The Internet Software Consortium.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the name of The Internet Software Consortium nor the names
 *    of its contributors may be used to endorse or promote products derived
 *    from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE INTERNET SOFTWARE CONSORTIUM AND
 * CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED.  IN NO EVENT SHALL THE INTERNET SOFTWARE CONSORTIUM OR
 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
 * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 *
 * This software has been written for the Internet Software Consortium
 * by Ted Lemon <mellon@fugue.com> in cooperation with Vixie
 * Enterprises.  To learn more about the Internet Software Consortium,
 * see ``http://www.vix.com/isc''.  To learn more about Vixie
 * Enterprises, see ``http://www.vix.com''.
 */

#include <sys/types.h>
#include <sys/socket.h>

#include <net/if.h>

#include <netinet/in.h>

#include <fcntl.h>
#include <stdio.h>
#include <time.h>
#include <unistd.h>

#include "dhcp.h"
#include "tree.h"
#include "dhcpd.h"
#include "log.h"

FILE *db_file;

static int counting = 0;
static int count = 0;
time_t write_time;

/*
 * Write the specified lease to the current lease database file.
 */
int
write_lease(struct lease *lease)
{
	char tbuf[26];	/* "w yyyy/mm/dd hh:mm:ss UTC" */
	size_t rsltsz;
	int errors = 0;
	int i;

	if (counting)
		++count;
	if (fprintf(db_file, "lease %s {\n", piaddr(lease->ip_addr)) == -1)
		++errors;

	rsltsz = strftime(tbuf, sizeof(tbuf), DB_TIMEFMT,
	    gmtime(&lease->starts));
	if (rsltsz == 0 || fprintf(db_file, "\tstarts %s;\n", tbuf) == -1)
		errors++;

	rsltsz = strftime(tbuf, sizeof(tbuf), DB_TIMEFMT,
	    gmtime(&lease->ends));
	if (rsltsz == 0 || fprintf(db_file, "\tends %s;\n", tbuf) == -1)
		errors++;

	if (lease->hardware_addr.hlen) {
		if (fprintf(db_file, "\thardware %s %s;",
		    hardware_types[lease->hardware_addr.htype],
		    print_hw_addr(lease->hardware_addr.htype,
		    lease->hardware_addr.hlen,
		    lease->hardware_addr.haddr)) == -1)
			++errors;
	}

	if (lease->uid_len) {
		int j;

		if (fprintf(db_file, "\n\tuid %2.2x", lease->uid[0]) == -1)
			++errors;

		for (j = 1; j < lease->uid_len; j++) {
			if (fprintf(db_file, ":%2.2x", lease->uid[j]) == -1)
				++errors;
		}
		if (fputc(';', db_file) == EOF)
			++errors;
	}

	if (lease->flags & BOOTP_LEASE) {
		if (fprintf(db_file, "\n\tdynamic-bootp;") == -1)
			++errors;
	}

	if (lease->flags & ABANDONED_LEASE) {
		if (fprintf(db_file, "\n\tabandoned;") == -1)
			++errors;
	}

	if (lease->client_hostname) {
		for (i = 0; lease->client_hostname[i]; i++)
			if (lease->client_hostname[i] < 33 ||
			    lease->client_hostname[i] > 126)
				goto bad_client_hostname;
		if (fprintf(db_file, "\n\tclient-hostname \"%s\";",
		    lease->client_hostname) == -1)
			++errors;
	}

bad_client_hostname:
	if (lease->hostname) {
		for (i = 0; lease->hostname[i]; i++)
			if (lease->hostname[i] < 33 ||
			    lease->hostname[i] > 126)
				goto bad_hostname;
		if (fprintf(db_file, "\n\thostname \"%s\";",
		    lease->hostname) == -1)
			++errors;
	}

bad_hostname:
	if (fputs("\n}\n", db_file) == EOF)
		++errors;

	if (errors)
		log_info("write_lease: unable to write lease %s",
		    piaddr(lease->ip_addr));

	return (!errors);
}

/*
 * Commit any leases that have been written out...
 */
int
commit_leases(void)
{
	/*
	 * Commit any outstanding writes to the lease database file. We need to
	 * do this even if we're rewriting the file below, just in case the
	 * rewrite fails.
	 */
	if (fflush(db_file) == EOF) {
		log_warn("commit_leases: unable to commit");
		return (0);
	}

	if (fsync(fileno(db_file)) == -1) {
		log_warn("commit_leases: unable to commit");
		return (0);
	}

	/*
	 * If we've written more than a thousand leases or if we haven't
	 * rewritten the lease database in over an hour, rewrite it now.
	 */
	if (count > 1000 || (count && cur_time - write_time > 3600)) {
		count = 0;
		write_time = cur_time;
		new_lease_file();
	}

	return (1);
}

void
db_startup(void)
{
	int db_fd;

	/* open lease file. once we dropped privs it has to stay open */
	db_fd = open(path_dhcpd_db, O_WRONLY|O_CREAT, 0640);
	if (db_fd == -1)
		fatal("Can't create new lease file");
	if ((db_file = fdopen(db_fd, "w")) == NULL)
		fatalx("Can't fdopen new lease file!");

	/* Read in the existing lease file... */
	read_leases();
	time(&write_time);

	new_lease_file();
}

void
new_lease_file(void)
{
	fflush(db_file);
	rewind(db_file);

	/* Write out all the leases that we know of... */
	counting = 0;
	write_leases();

	fflush(db_file);
	ftruncate(fileno(db_file), ftello(db_file));
	fsync(fileno(db_file));

	counting = 1;
}


Generated by: GCOVR (Version 3.3)

Line	Branch	Exec	Source
1			/* $OpenBSD: db.c,v 1.18 2017/02/13 23:04:05 krw Exp $ */
2
3			/*
4			* Persistent database management routines for DHCPD.
5			*/
6
7			/*
8			* Copyright (c) 1995, 1996 The Internet Software Consortium.
9			* All rights reserved.
10			*
11			* Redistribution and use in source and binary forms, with or without
12			* modification, are permitted provided that the following conditions
13			* are met:
14			*
15			* 1. Redistributions of source code must retain the above copyright
16			* notice, this list of conditions and the following disclaimer.
17			* 2. Redistributions in binary form must reproduce the above copyright
18			* notice, this list of conditions and the following disclaimer in the
19			* documentation and/or other materials provided with the distribution.
20			* 3. Neither the name of The Internet Software Consortium nor the names
21			* of its contributors may be used to endorse or promote products derived
22			* from this software without specific prior written permission.
23			*
24			* THIS SOFTWARE IS PROVIDED BY THE INTERNET SOFTWARE CONSORTIUM AND
25			* CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
26			* INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
27			* MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
28			* DISCLAIMED. IN NO EVENT SHALL THE INTERNET SOFTWARE CONSORTIUM OR
29			* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
30			* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
31			* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
32			* USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
33			* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
34			* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
35			* OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
36			* SUCH DAMAGE.
37			*
38			* This software has been written for the Internet Software Consortium
39			* by Ted Lemon <mellon@fugue.com> in cooperation with Vixie
40			* Enterprises. To learn more about the Internet Software Consortium,
41			* see ``http://www.vix.com/isc''. To learn more about Vixie
42			* Enterprises, see ``http://www.vix.com''.
43			*/
44
45			#include <sys/types.h>
46			#include <sys/socket.h>
47
48			#include <net/if.h>
49
50			#include <netinet/in.h>
51
52			#include <fcntl.h>
53			#include <stdio.h>
54			#include <time.h>
55			#include <unistd.h>
56
57			#include "dhcp.h"
58			#include "tree.h"
59			#include "dhcpd.h"
60			#include "log.h"
61
62			FILE *db_file;
63
64			static int counting = 0;
65			static int count = 0;
66			time_t write_time;
67
68			/*
69			* Write the specified lease to the current lease database file.
70			*/
71			int
72			write_lease(struct lease *lease)
73			{
74			char tbuf[26]; /* "w yyyy/mm/dd hh:mm:ss UTC" */
75			size_t rsltsz;
76			int errors = 0;
77			int i;
78
79			if (counting)
80			++count;
81			if (fprintf(db_file, "lease %s {\n", piaddr(lease->ip_addr)) == -1)
82			++errors;
83
84			rsltsz = strftime(tbuf, sizeof(tbuf), DB_TIMEFMT,
85			gmtime(&lease->starts));
86			if (rsltsz == 0 \|\| fprintf(db_file, "\tstarts %s;\n", tbuf) == -1)
87			errors++;
88
89			rsltsz = strftime(tbuf, sizeof(tbuf), DB_TIMEFMT,
90			gmtime(&lease->ends));
91			if (rsltsz == 0 \|\| fprintf(db_file, "\tends %s;\n", tbuf) == -1)
92			errors++;
93
94			if (lease->hardware_addr.hlen) {
95			if (fprintf(db_file, "\thardware %s %s;",
96			hardware_types[lease->hardware_addr.htype],
97			print_hw_addr(lease->hardware_addr.htype,
98			lease->hardware_addr.hlen,
99			lease->hardware_addr.haddr)) == -1)
100			++errors;
101			}
102
103			if (lease->uid_len) {
104			int j;
105
106			if (fprintf(db_file, "\n\tuid %2.2x", lease->uid[0]) == -1)
107			++errors;
108
109			for (j = 1; j < lease->uid_len; j++) {
110			if (fprintf(db_file, ":%2.2x", lease->uid[j]) == -1)
111			++errors;
112			}
113			if (fputc(';', db_file) == EOF)
114			++errors;
115			}
116
117			if (lease->flags & BOOTP_LEASE) {
118			if (fprintf(db_file, "\n\tdynamic-bootp;") == -1)
119			++errors;
120			}
121
122			if (lease->flags & ABANDONED_LEASE) {
123			if (fprintf(db_file, "\n\tabandoned;") == -1)
124			++errors;
125			}
126
127			if (lease->client_hostname) {
128			for (i = 0; lease->client_hostname[i]; i++)
129			if (lease->client_hostname[i] < 33 \|\|
130			lease->client_hostname[i] > 126)
131			goto bad_client_hostname;
132			if (fprintf(db_file, "\n\tclient-hostname \"%s\";",
133			lease->client_hostname) == -1)
134			++errors;
135			}
136
137			bad_client_hostname:
138			if (lease->hostname) {
139			for (i = 0; lease->hostname[i]; i++)
140			if (lease->hostname[i] < 33 \|\|
141			lease->hostname[i] > 126)
142			goto bad_hostname;
143			if (fprintf(db_file, "\n\thostname \"%s\";",
144			lease->hostname) == -1)
145			++errors;
146			}
147
148			bad_hostname:
149			if (fputs("\n}\n", db_file) == EOF)
150			++errors;
151
152			if (errors)
153			log_info("write_lease: unable to write lease %s",
154			piaddr(lease->ip_addr));
155
156			return (!errors);
157			}
158
159			/*
160			* Commit any leases that have been written out...
161			*/
162			int
163			commit_leases(void)
164			{
165			/*
166			* Commit any outstanding writes to the lease database file. We need to
167			* do this even if we're rewriting the file below, just in case the
168			* rewrite fails.
169			*/
170			if (fflush(db_file) == EOF) {
171			log_warn("commit_leases: unable to commit");
172			return (0);
173			}
174
175			if (fsync(fileno(db_file)) == -1) {
176			log_warn("commit_leases: unable to commit");
177			return (0);
178			}
179
180			/*
181			* If we've written more than a thousand leases or if we haven't
182			* rewritten the lease database in over an hour, rewrite it now.
183			*/
184			if (count > 1000 \|\| (count && cur_time - write_time > 3600)) {
185			count = 0;
186			write_time = cur_time;
187			new_lease_file();
188			}
189
190			return (1);
191			}
192
193			void
194			db_startup(void)
195			{
196			int db_fd;
197
198			/* open lease file. once we dropped privs it has to stay open */
199			db_fd = open(path_dhcpd_db, O_WRONLY\|O_CREAT, 0640);
200			if (db_fd == -1)
201			fatal("Can't create new lease file");
202			if ((db_file = fdopen(db_fd, "w")) == NULL)
203			fatalx("Can't fdopen new lease file!");
204
205			/* Read in the existing lease file... */
206			read_leases();
207			time(&write_time);
208
209			new_lease_file();
210			}
211
212			void
213			new_lease_file(void)
214			{
215			fflush(db_file);
216			rewind(db_file);
217
218			/* Write out all the leases that we know of... */
219			counting = 0;
220			write_leases();
221
222			fflush(db_file);
223			ftruncate(fileno(db_file), ftello(db_file));
224			fsync(fileno(db_file));
225
226			counting = 1;
227			}