xref: /freebsd/usr.sbin/bluetooth/sdpd/main.c (revision a0ee8cc6)
1 /*
2  * main.c
3  *
4  * Copyright (c) 2004 Maksim Yevmenkin <m_evmenkin@yahoo.com>
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19  * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26  * SUCH DAMAGE.
27  *
28  * $Id: main.c,v 1.8 2004/01/13 19:31:54 max Exp $
29  * $FreeBSD$
30  */
31 
32 #include <sys/select.h>
33 #define L2CAP_SOCKET_CHECKED
34 #include <bluetooth.h>
35 #include <errno.h>
36 #include <grp.h>
37 #include <pwd.h>
38 #include <signal.h>
39 #include <sdp.h>
40 #include <stdio.h>
41 #include <stdlib.h>
42 #include <string.h>
43 #include <unistd.h>
44 #include "log.h"
45 #include "server.h"
46 
47 #include <netinet/in.h>
48 #include <arpa/inet.h>
49 #include <sys/queue.h>
50 #include "profile.h"
51 #include "provider.h"
52 
53 #define	SDPD			"sdpd"
54 
55 static int32_t 	drop_root	(char const *user, char const *group);
56 static void	sighandler	(int32_t s);
57 static void	usage		(void);
58 
59 static int32_t	done;
60 
61 /*
62  * Bluetooth Service Discovery Procotol (SDP) daemon
63  */
64 
65 int
66 main(int argc, char *argv[])
67 {
68 	server_t		 server;
69 	char const		*control = SDP_LOCAL_PATH;
70 	char const		*user = "nobody", *group = "nobody";
71 	int32_t			 detach = 1, opt;
72 	struct sigaction	 sa;
73 
74 	while ((opt = getopt(argc, argv, "c:dg:hu:")) != -1) {
75 		switch (opt) {
76 		case 'c': /* control */
77 			control = optarg;
78 			break;
79 
80 		case 'd': /* do not detach */
81 			detach = 0;
82 			break;
83 
84 		case 'g': /* group */
85 			group = optarg;
86 			break;
87 
88 		case 'u': /* user */
89 			user = optarg;
90 			break;
91 
92 		case 'h':
93 		default:
94 			usage();
95 			/* NOT REACHED */
96 		}
97 	}
98 
99 	log_open(SDPD, !detach);
100 
101 	/* Become daemon if required */
102 	if (detach && daemon(0, 0) < 0) {
103 		log_crit("Could not become daemon. %s (%d)",
104 			strerror(errno), errno);
105 		exit(1);
106 	}
107 
108 	/* Set signal handlers */
109 	memset(&sa, 0, sizeof(sa));
110 	sa.sa_handler = sighandler;
111 
112 	if (sigaction(SIGTERM, &sa, NULL) < 0 ||
113 	    sigaction(SIGHUP,  &sa, NULL) < 0 ||
114 	    sigaction(SIGINT,  &sa, NULL) < 0) {
115 		log_crit("Could not install signal handlers. %s (%d)",
116 			strerror(errno), errno);
117 		exit(1);
118 	}
119 
120 	sa.sa_handler = SIG_IGN;
121 	if (sigaction(SIGPIPE, &sa, NULL) < 0) {
122 		log_crit("Could not install signal handlers. %s (%d)",
123 			strerror(errno), errno);
124 		exit(1);
125 	}
126 
127 	/* Initialize server */
128 	if (server_init(&server, control) < 0)
129 		exit(1);
130 
131 	if ((user != NULL || group != NULL) && drop_root(user, group) < 0)
132 		exit(1);
133 
134 	for (done = 0; !done; ) {
135 		if (server_do(&server) != 0)
136 			done ++;
137 	}
138 
139 	server_shutdown(&server);
140 	log_close();
141 
142 	return (0);
143 }
144 
145 /*
146  * Drop root
147  */
148 
149 static int32_t
150 drop_root(char const *user, char const *group)
151 {
152 	int	 uid, gid;
153 	char	*ep;
154 
155 	if ((uid = getuid()) != 0) {
156 		log_notice("Cannot set uid/gid. Not a superuser");
157 		return (0); /* dont do anything unless root */
158 	}
159 
160 	gid = getgid();
161 
162 	if (user != NULL) {
163 		uid = strtol(user, &ep, 10);
164 		if (*ep != '\0') {
165 			struct passwd	*pwd = getpwnam(user);
166 
167 			if (pwd == NULL) {
168 				log_err("Could not find passwd entry for " \
169 					"user %s", user);
170 				return (-1);
171 			}
172 
173 			uid = pwd->pw_uid;
174 		}
175 	}
176 
177 	if (group != NULL) {
178 		gid = strtol(group, &ep, 10);
179 		if (*ep != '\0') {
180 			struct group	*grp = getgrnam(group);
181 
182 			if (grp == NULL) {
183 				log_err("Could not find group entry for " \
184 					"group %s", group);
185 				return (-1);
186 			}
187 
188 			gid = grp->gr_gid;
189 		}
190 	}
191 
192 	if (setgid(gid) < 0) {
193 		log_err("Could not setgid(%s). %s (%d)",
194 			group, strerror(errno), errno);
195 		return (-1);
196 	}
197 
198 	if (setuid(uid) < 0) {
199 		log_err("Could not setuid(%s). %s (%d)",
200 			user, strerror(errno), errno);
201 		return (-1);
202 	}
203 
204 	return (0);
205 }
206 
207 /*
208  * Signal handler
209  */
210 
211 static void
212 sighandler(int32_t s)
213 {
214 	log_notice("Got signal %d. Total number of signals received %d",
215 		s, ++ done);
216 }
217 
218 /*
219  * Display usage information and quit
220  */
221 
222 static void
223 usage(void)
224 {
225 	fprintf(stderr,
226 "Usage: %s [options]\n" \
227 "Where options are:\n" \
228 "	-c	specify control socket name (default %s)\n" \
229 "	-d	do not detach (run in foreground)\n" \
230 "	-g grp	specify group\n" \
231 "	-h	display usage and exit\n" \
232 "	-u usr	specify user\n",
233 		SDPD, SDP_LOCAL_PATH);
234 	exit(255);
235 }
236 
237