static bool
handle_sockets_set (const char *path, const char *value)
{
const char *guid = path + strlen (APTERYX_SOCKETS_PATH"/");
bool res = true;
DEBUG ("SOCKET %s:%s\n", guid, value);
if (value)
res = rpc_server_bind (rpc, guid, value);
else
res = rpc_server_release (rpc, guid);
return res;
}
bool
apteryx_init (bool debug_enabled)
{
/* Increment refcount */
pthread_mutex_lock (&lock);
ref_count++;
debug |= debug_enabled;
if (ref_count == 1)
{
char * uri = NULL;
/* Create RPC instance */
rpc = rpc_init ((ProtobufCService *)&apteryx_client_service, &apteryx__server__descriptor, RPC_TIMEOUT_US);
if (rpc == NULL)
{
ERROR ("Init: Failed to initialise RPC service\n");
ref_count--;
pthread_mutex_unlock (&lock);
return false;
}
/* Bind to the default uri for this client */
if (asprintf((char **)&uri, APTERYX_SERVER".%"PRIu64, (uint64_t)getpid ()) <= 0)
return false;
if (!rpc_server_bind (rpc, uri, uri))
{
ERROR ("Failed to bind to default rpc service\n");
ref_count--;
free ((void*) uri);
pthread_mutex_unlock (&lock);
return false;
}
free ((void*) uri);
}
pthread_mutex_unlock (&lock);
/* Ready to go */
if (ref_count > 1)
DEBUG ("Init: Initialised\n");
return true;
}
int
main (int argc, char **argv)
{
const char *pid_file = NULL;
const char *run_file = NULL;
const char *url = APTERYX_SERVER;
bool background = false;
pthread_mutexattr_t callback_recursive;
FILE *fp;
int i;
/* Parse options */
while ((i = getopt (argc, argv, "hdmbp:r:l:")) != -1)
{
switch (i)
{
case 'd':
apteryx_debug = true;
background = false;
break;
case 'b':
background = true;
break;
case 'p':
pid_file = optarg;
break;
case 'r':
run_file = optarg;
break;
case 'l':
url = optarg;
break;
case 'm':
g_mem_set_vtable (glib_mem_profiler_table);
break;
case '?':
case 'h':
default:
help ();
return 0;
}
}
/* Handle SIGTERM/SIGINT/SIGPIPE gracefully */
signal (SIGTERM, (__sighandler_t) termination_handler);
signal (SIGINT, (__sighandler_t) termination_handler);
signal (SIGPIPE, SIG_IGN);
/* Daemonize */
if (background && fork () != 0)
{
/* Parent */
return 0;
}
/* Create pid file */
if (background && pid_file)
{
fp = fopen (pid_file, "w");
if (!fp)
{
ERROR ("Failed to create PID file %s\n", pid_file);
goto exit;
}
fprintf (fp, "%d\n", getpid ());
fclose (fp);
}
/* Initialise the database */
db_init ();
/* Initialise callbacks to clients */
cb_init ();
/* Configuration Set/Get */
config_init ();
/* Create a lock for currently-validating */
pthread_mutexattr_init (&callback_recursive);
pthread_mutexattr_settype (&callback_recursive, PTHREAD_MUTEX_RECURSIVE);
pthread_mutex_init (&validating, &callback_recursive);
/* Init the RPC for the server instance */
rpc = rpc_init ((ProtobufCService *)&apteryx_server_service, &apteryx__client__descriptor, RPC_TIMEOUT_US);
if (rpc == NULL)
{
ERROR ("Failed to initialise RPC service\n");
goto exit;
}
/* Create server and process requests */
if (!rpc_server_bind (rpc, url, url))
{
ERROR ("Failed to start rpc service\n");
goto exit;
}
/* Init the RPC for the proxy client */
proxy_rpc = rpc_init (NULL, &apteryx__server__descriptor, RPC_TIMEOUT_US);
if (proxy_rpc == NULL)
{
ERROR ("Failed to initialise proxy RPC service\n");
goto exit;
}
/* Create run file */
if (run_file)
{
fp = fopen (run_file, "w");
if (!fp)
{
ERROR ("Failed to create RUN file %s\n", run_file);
goto exit;
}
fclose (fp);
}
/* Loop while running */
while (running)
{
pause ();
}
exit:
DEBUG ("Exiting\n");
/* Cleanup callbacks */
cb_shutdown ();
db_shutdown ();
if (proxy_rpc)
{
rpc_shutdown (proxy_rpc);
}
if (rpc)
{
rpc_server_release (rpc, url);
rpc_shutdown (rpc);
}
/* Remove the pid file */
if (background && pid_file)
unlink (pid_file);
/* Remove the run file */
if (run_file)
unlink (run_file);
/* Memory profiling */
g_mem_profile ();
return 0;
}
