Changeset 185 for trunk/lib/regfi.c

Timestamp:

03/22/10 22:22:07 (14 years ago)

Author:

tim

Message:

reworked logging API again to simplify interface

updated regfi-threadtest to work with more recent commits

File:

• trunk/lib/regfi.c (modified) (4 diffs)

trunk/lib/regfi.c

@@ -45 +45 @@
 
 
+/* Ensures regfi_init runs only once */
+static pthread_once_t regfi_init_once = PTHREAD_ONCE_INIT;
+
+
 
 /******************************************************************************
  ******************************************************************************/
-void regfi_log_start(uint16_t msg_mask)
-{
+void regfi_log_free(void* ptr)
+{
+  REGFI_LOG* log_info = (REGFI_LOG*)ptr;
+  
+  if(log_info->messages != NULL)
+    free(log_info->messages);
+
+  talloc_free(log_info);
+}
+
+
+/******************************************************************************
+ ******************************************************************************/
+void regfi_init()
+{
+  int err;
+  if((err = pthread_key_create(&regfi_log_key, regfi_log_free)) != 0)
+    fprintf(stderr, "ERROR: key_create: %s\n", strerror(err));
+  errno = err;
+}
+
+
+/******************************************************************************
+ ******************************************************************************/
+REGFI_LOG* regfi_log_new()
+{
+  int err;
   REGFI_LOG* log_info = talloc(NULL, REGFI_LOG);
   if(log_info == NULL)
-    return;
-
-  log_info->msg_mask = msg_mask;
+    return NULL;
+
+  log_info->msg_mask = REGFI_DEFAULT_LOG_MASK;
   log_info->messages = NULL;
 
-  /* XXX: Should we bother with a destructor here? */
-  if(pthread_key_create(&REGFI_LOG_KEY, NULL) != 0)
-  {
-    fprintf(stderr, "ERROR: key_create\n");
-    goto fail;
-  }
-
-  if(pthread_setspecific(REGFI_LOG_KEY, log_info) != 0)
-  {
-    fprintf(stderr, "ERROR: setspecific\n");
-    goto fail;
-  }
-
-  return;
+  pthread_once(&regfi_init_once, regfi_init);
+
+  if((err = pthread_setspecific(regfi_log_key, log_info)) != 0)
+  {
+    fprintf(stderr, "ERROR: setspecific: %s\n", strerror(err));
+    goto fail;
+  }
+
+  return log_info;
 
  fail:
   talloc_free(log_info);
+  errno = err;
+  return NULL;
 }
 
@@ -93 +119 @@
   va_list args;
 
-  log_info = (REGFI_LOG*)pthread_getspecific(REGFI_LOG_KEY);
-  if(log_info == NULL || (log_info->msg_mask & msg_type) == 0)
+  log_info = (REGFI_LOG*)pthread_getspecific(regfi_log_key);
+  if(log_info == NULL && (log_info = regfi_log_new()) == NULL)
+    return;
+
+  if((log_info->msg_mask & msg_type) == 0)
     return;
 
@@ -138 +167 @@
 {
   char* ret_val;
-  REGFI_LOG* log_info = (REGFI_LOG*)pthread_getspecific(REGFI_LOG_KEY);
-  if(log_info == NULL)
-    return NULL;
-
+  REGFI_LOG* log_info = (REGFI_LOG*)pthread_getspecific(regfi_log_key);
+  if(log_info == NULL && (log_info = regfi_log_new()) == NULL)
+    return NULL;
+  
   ret_val = log_info->messages;
   log_info->messages = NULL;
@@ -151 +180 @@
 /******************************************************************************
  ******************************************************************************/
-void regfi_log_set_mask(uint16_t msg_mask)
-{
-  REGFI_LOG* log_info = (REGFI_LOG*)pthread_getspecific(REGFI_LOG_KEY);
-  if(log_info == NULL)
-    return;
+bool regfi_log_set_mask(uint16_t msg_mask)
+{
+  REGFI_LOG* log_info = (REGFI_LOG*)pthread_getspecific(regfi_log_key);
+  if(log_info == NULL && (log_info = regfi_log_new()) == NULL)
+  {
+      return false;
+  }
 
   log_info->msg_mask = msg_mask;
-}
-
-
-/******************************************************************************
- ******************************************************************************/
-void regfi_log_stop()
-{
-  REGFI_LOG* log_info = (REGFI_LOG*)pthread_getspecific(REGFI_LOG_KEY);
-  if(log_info == NULL)
-    return;
-  
-  if(log_info->messages != NULL)
-    free(log_info->messages);
-
-  pthread_key_delete(REGFI_LOG_KEY);
-  talloc_free(log_info);
+  return true;
 }