libsent/src/util/j_printf.c

Go to the documentation of this file.

/*
 * Copyright (c) 1991-2006 Kawahara Lab., Kyoto University
 * Copyright (c) 2000-2005 Shikano Lab., Nara Institute of Science and Technology
 * Copyright (c) 2005-2006 Julius project team, Nagoya Institute of Technology
 * All rights reserved
 */

#include <sent/stddefs.h>
#include <sent/tcpip.h>
#include <stdarg.h>

/* pointer to functions of general text output */
static void (*j_print_func)(char *) = NULL;
static void (*j_printerr_func)(char *) = NULL; 
static void (*j_flush_func)(void) = NULL;

#define MAX_PRINTF_LEN 4096     
static char inbuf[MAX_PRINTF_LEN]; 
#ifdef CHARACTER_CONVERSION
static char outbuf[MAX_PRINTF_LEN]; 
#endif

void
set_print_func(void (*print)(char *), void (*printerr)(char *), void (*flush)(void))
{
  j_print_func = print;
  j_printerr_func = printerr;
  j_flush_func = flush;
  return;
}

int
j_printf(char *fmt, ...)
{
  va_list ap;
  int ret;
  char *buf;
  
  va_start(ap,fmt);
  ret = vsnprintf(inbuf, MAX_PRINTF_LEN, fmt, ap);
  va_end(ap);
#ifdef CHARACTER_CONVERSION
  buf = charconv(inbuf, outbuf, MAX_PRINTF_LEN);
#else
  buf = inbuf;
#endif
  if (j_print_func != NULL) {
    (*j_print_func)(buf);
  } else {
    ret = printf("%s", buf);
  }

  return(ret);
}

int
j_printerr(char *fmt, ...)
{
  va_list ap;
  int ret;
  char *buf;
  
  va_start(ap,fmt);
  ret = vsnprintf(inbuf, MAX_PRINTF_LEN, fmt, ap);
  va_end(ap);
#ifdef CHARACTER_CONVERSION
  buf = charconv(inbuf, outbuf, MAX_PRINTF_LEN);
#else
  buf = inbuf;
#endif
  if (j_printerr_func != NULL) {
    (*j_printerr_func)(buf);
  } else {
    ret = fprintf(stderr, "%s", buf);
  }

  return(ret);
}

int
j_flushprint()
{
  if (j_flush_func != NULL) {
    (*j_flush_func)();
    return 0;
  } else {
    return(fflush(stdout));
  }
  return(fflush(stdout));
}

int
module_send(int sd, char *fmt, ...)
{
  va_list ap;
  int ret;
  char *buf;
  
  va_start(ap,fmt);
  ret = vsnprintf(inbuf, MAX_PRINTF_LEN, fmt, ap);
  va_end(ap);
  if (ret > 0) {                /* success */
    
#ifdef CHARACTER_CONVERSION
    buf = charconv(inbuf, outbuf, MAX_PRINTF_LEN);
#else
    buf = inbuf;
#endif
    if (
#ifdef WINSOCK
        send(sd, buf, strlen(buf), 0)
#else
        write(sd, buf, strlen(buf))
#endif
        < 0) {
      perror("Error: module_send:");
    }
  }
  return(ret);
}

static void (*error_func)() = NULL; 

void
j_error_register_exitfunc(void (*f)())
{
  error_func = f;
}

void
j_error(char *fmt, ...)
{
  va_list ap;
  int ret;
  va_start(ap,fmt);
  ret = vfprintf(stderr, fmt, ap);
  va_end(ap);
  /* clean up socket if already opened */
  cleanup_socket();
  if (error_func != NULL) (*error_func)(); /* call external error function */
  else exit(1);
  return;
}

/* normal exit end  */
static void (*exit_func)() = NULL; 

void
j_exit_register_exitfunc(void (*f)())
{
  exit_func = f;
}

void
j_exit()
{
  /* clean up socket if already opened */
  cleanup_socket();
  if (exit_func != NULL) (*exit_func)(); /* call external error function */
  else exit(0);
}

---