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