source: roaraudio/roard/roard.c @ 523:0c640b607a7f

Last change on this file since 523:0c640b607a7f was 523:0c640b607a7f, checked in by phi, 16 years ago

it should now be possible to use --setuid/--setgid on non unix sockets
File size: 12.6 KB
Line 
1//roard.c:
2
3#include "roard.h"
4
5char * server = ROAR_DEFAULT_SOCK_GLOBAL; // global server address
6
7void usage (void) {
8 printf("Usage: roard [OPTIONS]...\n\n");
9
10 printf("Misc Options:\n\n");
11 printf(
12        " --demon               - Bring the server into background after init\n"
13        " --terminate           - Terminate after last client quited\n"
14        " --restart             - Trys to stop an old instance and start a new with new settings\n"
15        " --realtime            - Trys to get realtime priority,\n"
16        "                         give multible times for being more realtime\n"
17        " --chroot DIR          - chroots to the given dir\n"
18        " --setgid              - GroupID to the audio group as specified via -G\n"
19        " --setuid              - UserID to the audio user as specified via -U\n"
20       );
21
22 printf("\nAudio Options:\n\n");
23 printf(
24        " -R  --rate   RATE     - Set server rate\n"
25        " -B  --bits   BITS     - Set server bits\n"
26        " -C  --chans  CHANNELS - Set server channels\n"
27       );
28
29 printf("\nDriver Options:\n\n");
30 printf(" -d  --driver DRV      - Set the driver, use '-d list' to get a list (default: %s)\n", ROAR_DRIVER_DEFAULT);
31 printf(" -D  --device DEV      - Set the device\n");
32 printf(" -dO OPTS              - Set output options\n");
33
34 printf("\nSource Options:\n\n");
35 printf(" -s  --source DRV      - Use DRV as input driver\n"
36        " -S           DEV      - Use DEV as input device\n"
37        " -sO          OPTS     - Use OPTS as input options\n"
38        " -sP                   - Make souce as primary\n"
39       );
40
41 printf("\nCodec Filter Options:\n\n");
42 printf(" --list-cf             - List all codec filter\n"
43       );
44
45 printf("\nServer Options:\n\n");
46 printf(" -t  --tcp             - Use TCP listen socket\n"
47        " -u  --unix            - Use UNIX Domain listen socket (default)\n"
48#ifdef ROAR_HAVE_LIBDNET
49        " -n  --decnet          - use DECnet listen socket\n"
50#endif
51        " -4                    - Use IPv4 connections (implies -t)\n"
52#ifdef PF_INET6
53        " -6                    - Use IPv6 connections (implies -t)\n"
54#endif
55#ifdef IPV6_ADDRFORM
56        " -64                   - Try to downgrade sockets from IPv6 into IPv4,\n"
57        "                         this is normaly not usefull.\n"
58#endif
59        " -p  --port            - TCP Port to bind to\n"
60        " -b  --bind            - IP/Hostname to bind to\n"
61        " -s  --sock            - Filename for UNIX Domain Socket\n"
62        " -G  GROUP             - Sets the group for the UNIX Domain Socket, (default: %s)\n"
63        "                         You need the permittions to change the GID\n"
64        " -U  USER              - Sets the user for the UNIX Domain Socket, (default: do not set)\n"
65        "                         You need the permittions to change the UID (normaly only root has)\n"
66        " --no-listen           - Do not listen for new clients (only usefull for relaing)\n"
67        " --client-fh           - Comunicate with a client over this handle\n"
68        "                         (only usefull for relaing)\n"
69        " --close-fh            - Closes the given fh\n",
70        ROAR_DEFAULT_SOCKGRP
71       );
72// printf("\n Options:\n\n");
73 printf("\n");
74}
75
76#define R_SETUID 1
77#define R_SETGID 2
78
79int main (int argc, char * argv[]) {
80 int i;
81 char * k;
82 char user_sock[80] = {0};
83 struct roar_audio_info sa;
84 int    demon    = 0;
85 int    realtime = 0;
86 char * driver = getenv("ROAR_DRIVER");
87 char * device = getenv("ROAR_DEVICE");
88 char * opts   = NULL;
89// char * server = ROAR_DEFAULT_SOCK_GLOBAL;
90 int      port = ROAR_DEFAULT_PORT;
91 int               drvid;
92 char * s_dev     = NULL;
93 char * s_con     = NULL;
94 char * s_opt     = NULL;
95 int    s_prim    = 0;
96 char * sock_grp  = ROAR_DEFAULT_SOCKGRP;
97 char * sock_user = NULL;
98 int    sock_type = ROAR_SOCKET_TYPE_UNKNOWN;
99 char * chrootdir = NULL;
100 int    setids    = 0;
101 struct group   * grp  = NULL;
102 struct passwd  * pwd  = NULL;
103 struct servent * serv = NULL;
104 DRIVER_USERDATA_T drvinst;
105 struct roar_client * self = NULL;
106#ifdef ROAR_HAVE_LIBDNET
107 char decnethost[80];
108#endif
109
110 g_listen_socket = -1;
111 g_standby       =  0;
112
113 sa.bits     = ROAR_BITS_DEFAULT;
114 sa.channels = ROAR_CHANNELS_DEFAULT;
115 sa.rate     = ROAR_RATE_DEFAULT;
116 sa.codec    = ROAR_CODEC_DEFAULT;
117
118 g_sa = &sa;
119
120
121 if ( getuid() != 0 && getenv("HOME") ) {
122  snprintf(user_sock, 79, "%s/%s", getenv("HOME"), ROAR_DEFAULT_SOCK_USER);
123  server = user_sock;
124 }
125
126 if ( getenv("ROAR_SERVER") != NULL )
127  server = getenv("ROAR_SERVER");
128
129 if ( clients_init() == -1 ) {
130  ROAR_ERR("Can not init clients!");
131  return 1;
132 }
133
134 if ( streams_init() == -1 ) {
135  ROAR_ERR("Can not init streams!");
136  return 1;
137 }
138
139 if ( (g_self_client = clients_new()) == -1 ) {
140  ROAR_ERR("Can not create self client!");
141  return 1;
142 }
143
144 if ( sources_init() == -1 ) {
145  ROAR_ERR("Can not init sources!");
146  return 1;
147 }
148
149 if ( (sources_set_client(g_self_client)) == -1 ) {
150  ROAR_ERR("Can not init set source client!");
151  return 1;
152 }
153
154 for (i = 1; i < argc; i++) {
155  k = argv[i];
156
157  if ( strcmp(k, "-h") == 0 || strcmp(k, "--help") == 0 ) {
158   usage();
159   return 0;
160
161  } else if ( strcmp(k, "--demon") == 0 ) {
162   demon = 1;
163  } else if ( strcmp(k, "--terminate") == 0 ) {
164   g_terminate = 1;
165  } else if ( strcmp(k, "--realtime") == 0 ) {
166   realtime++;
167  } else if ( strcmp(k, "--chroot") == 0 ) {
168   chrootdir = argv[++i];
169  } else if ( strcmp(k, "--setgid") == 0 ) {
170   setids |= R_SETGID;
171  } else if ( strcmp(k, "--setuid") == 0 ) {
172   setids |= R_SETUID;
173
174  } else if ( strcmp(k, "--list-cf") == 0 ) {
175   print_codecfilterlist();
176   return 0;
177
178  } else if ( strcmp(k, "-R") == 0 || strcmp(k, "--rate") == 0 ) {
179   sa.rate = atoi(argv[++i]);
180  } else if ( strcmp(k, "-B") == 0 || strcmp(k, "--bits") == 0 ) {
181   sa.bits = atoi(argv[++i]);
182  } else if ( strcmp(k, "-C") == 0 || strcmp(k, "--chans") == 0 ) {
183   sa.channels = atoi(argv[++i]);
184
185  } else if ( strcmp(k, "-d") == 0 || strcmp(k, "--driver") == 0 ) {
186   driver = argv[++i];
187   if ( strcmp(driver, "list") == 0 ) {
188    print_driverlist();
189   }
190  } else if ( strcmp(k, "-D") == 0 || strcmp(k, "--device") == 0 ) {
191   device = argv[++i];
192  } else if ( strcmp(k, "-dO") == 0 ) {
193   opts = argv[++i];
194
195  } else if ( strcmp(k, "-s") == 0 || strcmp(k, "--source") == 0 ) {
196   k = argv[++i];
197   if ( sources_add(k, s_dev, s_con, s_opt, s_prim) == -1 ) {
198    ROAR_ERR("main(*): adding source '%s' via '%s' failed!", s_dev, k);
199   }
200   s_opt = s_dev = s_con = NULL;
201   s_prim = 0;
202  } else if ( strcmp(k, "-S") == 0 ) {
203   s_dev = argv[++i];
204  } else if ( strcmp(k, "-sO") == 0 ) {
205   s_opt = argv[++i];
206  } else if ( strcmp(k, "-sC") == 0 ) {
207   s_con = argv[++i];
208  } else if ( strcmp(k, "-sP") == 0 ) {
209   s_prim = 1;
210
211  } else if ( strcmp(k, "-p") == 0 || strcmp(k, "--port") == 0 ) {
212   // This is only usefull in INET not UNIX mode.
213   if ( *server == '/' )
214    server = ROAR_DEFAULT_HOST;
215
216   errno = 0;
217   if ( (port = atoi(argv[++i])) < 1 ) {
218    if ( (serv = getservbyname(argv[i], "tcp")) == NULL ) {
219     ROAR_ERR("Unknown service: %s: %s", argv[i], strerror(errno));
220     return 1;
221    }
222    // NOTE: we need to use ROAR_NET2HOST16() here even if s_port is of type int!
223    ROAR_DBG("main(*): serv = {s_name='%s', s_aliases={...}, s_port=%i, s_proto='%s'}",
224            serv->s_name, ROAR_NET2HOST16(serv->s_port), serv->s_proto);
225    port = ROAR_NET2HOST16(serv->s_port);
226   }
227  } else if ( strcmp(k, "-b") == 0 || strcmp(k, "--bind") == 0 || strcmp(k, "-s") == 0 || strcmp(k, "--sock") == 0 ) {
228   server = argv[++i];
229
230  } else if ( strcmp(k, "-t") == 0 ) {
231   if ( sock_type != ROAR_SOCKET_TYPE_TCP && sock_type != ROAR_SOCKET_TYPE_TCP6 )
232    sock_type = ROAR_SOCKET_TYPE_TCP;
233
234   if ( *server == '/' )
235    server = ROAR_DEFAULT_HOST;
236
237  } else if ( strcmp(k, "-4") == 0 ) {
238   sock_type = ROAR_SOCKET_TYPE_TCP;
239   if ( *server == '/' )
240    server = ROAR_DEFAULT_HOST;
241  } else if ( strcmp(k, "-6") == 0 ) {
242#ifdef PF_INET6
243   sock_type = ROAR_SOCKET_TYPE_TCP6;
244   if ( *server == '/' )
245    server = ROAR_DEFAULT_HOST;
246#else
247    ROAR_ERR("No IPv6 support compiled in!");
248    return 1;
249#endif
250
251  } else if ( strcmp(k, "-u") == 0 ) {
252   // ignore this case as it is the default behavor.
253   sock_type = ROAR_SOCKET_TYPE_UNIX;
254
255  } else if ( strcmp(k, "-n") == 0 ) {
256#ifdef ROAR_HAVE_LIBDNET
257    port   = ROAR_DEFAULT_NUM;
258    strcpy(decnethost, ROAR_DEFAULT_LISTEN_OBJECT);
259    server = decnethost;
260    sock_type = ROAR_SOCKET_TYPE_DECNET;
261#else
262    ROAR_ERR("No DECnet support compiled in!");
263    return 1;
264#endif
265
266  } else if ( strcmp(k, "-G") == 0 ) {
267   sock_grp  = argv[++i];
268  } else if ( strcmp(k, "-U") == 0 ) {
269   sock_user = argv[++i];
270
271  } else if ( strcmp(k, "--no-listen") == 0 ) {
272   *server = 0;
273  } else if ( strcmp(k, "--client-fh") == 0 ) {
274   if ( clients_set_fh(clients_new(), atoi(argv[++i])) == -1 ) {
275    ROAR_ERR("main(*): Can not set client's fh");
276    return 1;
277   }
278  } else if ( strcmp(k, "--close-fh") == 0 ) {
279   close(atoi(argv[++i]));
280
281  } else {
282   usage();
283   return 1;
284  }
285
286 }
287
288 ROAR_DBG("Server config: rate=%i, bits=%i, chans=%i", sa.rate, sa.bits, sa.channels);
289
290 if ( midi_init() == -1 )
291  ROAR_ERR("Can not initialize MIDI subsystem");
292
293 if ( *server != 0 ) {
294  if ( (g_listen_socket = roar_socket_listen(sock_type, server, port)) == -1 ) {
295   if ( *server == '/' ) {
296    if ( (i = roar_socket_connect(server, port)) != -1 ) {
297     close(i);
298     ROAR_ERR("Can not open listen socket!");
299     return 1;
300    } else {
301     unlink(server);
302     if ( (g_listen_socket = roar_socket_listen(sock_type, server, port)) == -1 ) {
303      ROAR_ERR("Can not open listen socket!");
304      return 1;
305     }
306    }
307   } else {
308    ROAR_ERR("Can not open listen socket!");
309    return 1;
310   }
311  }
312
313  if ( (grp = getgrnam(sock_grp)) == NULL ) {
314   ROAR_ERR("Can not get GID for group %s: %s", sock_grp, strerror(errno));
315  }
316  if ( sock_user || (setids & R_SETUID) ) {
317   if ( (pwd = getpwnam(sock_user)) == NULL ) {
318    ROAR_ERR("Can not get UID for user %s: %s", sock_user, strerror(errno));
319   }
320  }
321
322  if ( *server == '/' ) {
323   if ( grp ) {
324    if ( pwd ) {
325     chown(server, pwd->pw_uid, grp->gr_gid);
326    } else {
327     chown(server, -1, grp->gr_gid);
328    }
329    if ( getuid() == 0 )
330     chmod(server, S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP);
331   }
332  }
333 }
334
335 if ( output_buffer_init(&sa) == -1 ) {
336  ROAR_ERR("Can not init output buffer!");
337  return 1;
338 }
339
340 if ( driver_open(&drvinst, &drvid, driver, device, &sa) == -1 ) {
341  ROAR_ERR("Can not open output driver!");
342  return 1;
343 }
344
345 if ( samples_init() == -1 ) {
346  ROAR_ERR("Can not init samples!");
347  return 1;
348 }
349
350
351 signal(SIGINT,  on_sig_int);
352 signal(SIGCHLD, on_sig_chld);
353 signal(SIGPIPE, SIG_IGN);  // ignore broken pipes
354
355 if ( realtime ) {
356#ifdef DEBUG
357  ROAR_WARN("compiled with -DDEBUG but realtime is enabled: for real realtime support compiel without -DDEBUG");
358#endif
359
360  errno = 0;
361  nice(-5*realtime); // -5 for each --realtime
362  if ( errno )
363   ROAR_WARN("Can not decrease nice value by 5: %s", strerror(errno));
364/*
365#ifdef __linux__
366  if ( ioprio_set(IOPRIO_WHO_PROCESS, getpid(), IOPRIO_PRIO_VALUE(IOPRIO_CLASS_BE, 0)) == -1 )
367   ROAR_WARN("Can not set io priority: %s", strerror(errno));
368#endif
369*/
370 }
371
372 if ( setids & R_SETGID ) {
373  if ( setgroups(0, (const gid_t *) NULL) == -1 ) {
374   ROAR_ERR("Can not clear supplementary group IDs: %s", strerror(errno));
375  }
376  if ( !grp || setgid(grp->gr_gid) == -1 ) {
377   ROAR_ERR("Can not set GroupID: %s", strerror(errno));
378  }
379 }
380
381
382 clients_set_pid(g_self_client, getpid());
383 clients_set_uid(g_self_client, getuid());
384 clients_set_gid(g_self_client, getgid());
385 clients_get(g_self_client, &self);
386
387 if ( self == NULL ) {
388  ROAR_ERR("Can not get self client!");
389  return 1;
390 }
391
392 strcpy(self->name, "RoarAudio demon internal");
393
394 if ( demon ) {
395  close(ROAR_STDIN );
396  close(ROAR_STDOUT);
397  close(ROAR_STDERR);
398  setsid();
399  if ( fork() )
400   _exit(0);
401 }
402
403 if (chrootdir) {
404  if ( chroot(chrootdir) == -1 ) {
405   ROAR_ERR("Can not chroot to %s: %s", chrootdir, strerror(errno));
406   return 2;
407  }
408  if ( chdir("/") == -1 ) {
409   ROAR_ERR("Can not chdir to /: %s", strerror(errno));
410   return 2;
411  }
412 }
413
414 if ( setids & R_SETUID ) {
415  if ( !pwd || setuid(pwd->pw_uid) == -1 ) {
416   ROAR_ERR("Can not set UserID: %s", strerror(errno));
417   return 3;
418  }
419  clients_set_uid(g_self_client, getuid());
420 }
421
422 // start main loop...
423 main_loop(drvid, drvinst, &sa);
424
425 // clean up.
426 clean_quit_prep();
427 driver_close(drvinst, drvid);
428 output_buffer_free();
429
430 return 0;
431}
432
433void clean_quit_prep (void) {
434 close(g_listen_socket);
435
436 if ( *server == '/' )
437  unlink(server);
438
439
440 sources_free();
441 streams_free();
442 clients_free();
443 midi_cb_stop(); // stop console beep
444 midi_free();
445}
446
447void clean_quit (void) {
448 clean_quit_prep();
449// driver_close(drvinst, drvid);
450// output_buffer_free();
451 exit(0);
452}
453
454//ll
Note: See TracBrowser for help on using the repository browser.