source: roaraudio/roard/roard.c @ 576:89baebafdc4f

Last change on this file since 576:89baebafdc4f was 576:89baebafdc4f, checked in by phi, 16 years ago

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