467 lines
8.1 KiB
C
467 lines
8.1 KiB
C
/*
|
|
* $Id: stdio.c 463 2008-12-09 06:52:03Z baconevi $
|
|
*
|
|
* {License}
|
|
*/
|
|
|
|
#include <ase/utl/stdio.h>
|
|
#include "../cmn/chr.h"
|
|
|
|
#include <wchar.h>
|
|
#include <stdlib.h>
|
|
#include <limits.h>
|
|
|
|
#ifndef PATH_MAX
|
|
#define PATH_MAX 2048
|
|
#endif
|
|
|
|
#if defined(_WIN32)
|
|
|
|
int ase_vsprintf (ase_char_t* buf, size_t size, const ase_char_t* fmt, va_list ap)
|
|
{
|
|
int n;
|
|
|
|
#ifdef ASE_CHAR_IS_MCHAR
|
|
n = _vsnprintf (buf, size, fmt, ap);
|
|
#else
|
|
n = _vsnwprintf (buf, size, fmt, ap);
|
|
#endif
|
|
if (n < 0 || (size_t)n >= size)
|
|
{
|
|
if (size > 0) buf[size-1] = ASE_T('\0');
|
|
n = -1;
|
|
}
|
|
|
|
return n;
|
|
}
|
|
|
|
int ase_sprintf (ase_char_t* buf, size_t size, const ase_char_t* fmt, ...)
|
|
{
|
|
int n;
|
|
va_list ap;
|
|
|
|
va_start (ap, fmt);
|
|
n = ase_vsprintf (buf, size, fmt, ap);
|
|
va_end (ap);
|
|
return n;
|
|
}
|
|
|
|
#else
|
|
|
|
static ase_char_t* __adjust_format (const ase_char_t* format);
|
|
|
|
int ase_vfprintf (ASE_FILE *stream, const ase_char_t* fmt, va_list ap)
|
|
{
|
|
int n;
|
|
ase_char_t* nf = __adjust_format (fmt);
|
|
if (nf == NULL) return -1;
|
|
|
|
#ifdef ASE_CHAR_IS_MCHAR
|
|
n = vfprintf (stream, nf, ap);
|
|
#else
|
|
n = vfwprintf (stream, nf, ap);
|
|
#endif
|
|
free (nf);
|
|
return n;
|
|
}
|
|
|
|
int ase_vprintf (const ase_char_t* fmt, va_list ap)
|
|
{
|
|
return ase_vfprintf (stdout, fmt, ap);
|
|
}
|
|
|
|
int ase_fprintf (ASE_FILE* file, const ase_char_t* fmt, ...)
|
|
{
|
|
int n;
|
|
va_list ap;
|
|
|
|
va_start (ap, fmt);
|
|
n = ase_vfprintf (file, fmt, ap);
|
|
va_end (ap);
|
|
return n;
|
|
}
|
|
|
|
int ase_printf (const ase_char_t* fmt, ...)
|
|
{
|
|
int n;
|
|
va_list ap;
|
|
|
|
va_start (ap, fmt);
|
|
n = ase_vprintf (fmt, ap);
|
|
va_end (ap);
|
|
return n;
|
|
}
|
|
|
|
int ase_vsprintf (ase_char_t* buf, size_t size, const ase_char_t* fmt, va_list ap)
|
|
{
|
|
int n;
|
|
ase_char_t* nf = __adjust_format (fmt);
|
|
if (nf == NULL) return -1;
|
|
|
|
#if defined(ASE_CHAR_IS_MCHAR)
|
|
n = vsnprintf (buf, size, nf, ap);
|
|
#elif defined(_WIN32)
|
|
n = _vsnwprintf (buf, size, nf, ap);
|
|
#else
|
|
n = vswprintf (buf, size, nf, ap);
|
|
#endif
|
|
if (n < 0 || (size_t)n >= size)
|
|
{
|
|
if (size > 0) buf[size-1] = ASE_T('\0');
|
|
n = -1;
|
|
}
|
|
|
|
free (nf);
|
|
return n;
|
|
}
|
|
|
|
int ase_sprintf (ase_char_t* buf, size_t size, const ase_char_t* fmt, ...)
|
|
{
|
|
int n;
|
|
va_list ap;
|
|
|
|
va_start (ap, fmt);
|
|
n = ase_vsprintf (buf, size, fmt, ap);
|
|
va_end (ap);
|
|
return n;
|
|
}
|
|
|
|
#define MOD_SHORT 1
|
|
#define MOD_LONG 2
|
|
#define MOD_LONGLONG 3
|
|
|
|
#define ADDC(buf,c) \
|
|
do { \
|
|
if (buf.len >= buf.cap) \
|
|
{ \
|
|
ase_char_t* tmp; \
|
|
tmp = (ase_char_t*)realloc ( \
|
|
buf.ptr, sizeof(ase_char_t)*(buf.cap+256+1)); \
|
|
if (tmp == NULL) \
|
|
{ \
|
|
free (buf.ptr); \
|
|
return NULL; \
|
|
} \
|
|
buf.ptr = tmp; \
|
|
buf.cap = buf.cap + 256; \
|
|
} \
|
|
buf.ptr[buf.len++] = c; \
|
|
} while (0)
|
|
|
|
static ase_char_t* __adjust_format (const ase_char_t* format)
|
|
{
|
|
const ase_char_t* fp = format;
|
|
int modifier;
|
|
ase_char_t ch;
|
|
|
|
struct
|
|
{
|
|
ase_char_t* ptr;
|
|
ase_size_t len;
|
|
ase_size_t cap;
|
|
} buf;
|
|
|
|
buf.len = 0;
|
|
buf.cap = 256;
|
|
#if (defined(vms) || defined(__vms)) && (ASE_SIZEOF_VOID_P >= 8)
|
|
buf.ptr = (ase_char_t*) _malloc32 (sizeof(ase_char_t)*(buf.cap+1));
|
|
#else
|
|
buf.ptr = (ase_char_t*) malloc (sizeof(ase_char_t)*(buf.cap+1));
|
|
#endif
|
|
if (buf.ptr == NULL) return NULL;
|
|
|
|
while (*fp != ASE_T('\0'))
|
|
{
|
|
while (*fp != ASE_T('\0') && *fp != ASE_T('%'))
|
|
{
|
|
ADDC (buf, *fp++);
|
|
}
|
|
|
|
if (*fp == ASE_T('\0')) break;
|
|
|
|
ch = *fp++;
|
|
ADDC (buf, ch); /* add % */
|
|
|
|
ch = *fp++;
|
|
|
|
/* flags */
|
|
while (1)
|
|
{
|
|
if (ch == ASE_T(' ') || ch == ASE_T('+') ||
|
|
ch == ASE_T('-') || ch == ASE_T('#'))
|
|
{
|
|
ADDC (buf, ch);
|
|
ch = *fp++;
|
|
}
|
|
else
|
|
{
|
|
if (ch == ASE_T('0'))
|
|
{
|
|
ADDC (buf, ch);
|
|
ch = *fp++;
|
|
}
|
|
|
|
break;
|
|
}
|
|
}
|
|
|
|
/* check the width */
|
|
if (ch == ASE_T('*'))
|
|
{
|
|
ADDC (buf, ch);
|
|
ch = *fp++;
|
|
}
|
|
else
|
|
{
|
|
while (ASE_ISDIGIT(ch))
|
|
{
|
|
ADDC (buf, ch);
|
|
ch = *fp++;
|
|
}
|
|
}
|
|
|
|
/* precision */
|
|
if (ch == ASE_T('.'))
|
|
{
|
|
ADDC (buf, ch);
|
|
ch = *fp++;
|
|
|
|
if (ch == ASE_T('*'))
|
|
{
|
|
ADDC (buf, ch);
|
|
ch = *fp++;
|
|
}
|
|
else
|
|
{
|
|
while (ASE_ISDIGIT(ch))
|
|
{
|
|
ADDC (buf, ch);
|
|
ch = *fp++;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* modifier */
|
|
for (modifier = 0;;)
|
|
{
|
|
if (ch == ASE_T('h')) modifier = MOD_SHORT;
|
|
else if (ch == ASE_T('l'))
|
|
{
|
|
modifier = (modifier == MOD_LONG)? MOD_LONGLONG: MOD_LONG;
|
|
}
|
|
else break;
|
|
ch = *fp++;
|
|
}
|
|
|
|
|
|
/* type */
|
|
if (ch == ASE_T('%')) ADDC (buf, ch);
|
|
else if (ch == ASE_T('c') || ch == ASE_T('s'))
|
|
{
|
|
#if !defined(ASE_CHAR_IS_MCHAR) && !defined(_WIN32)
|
|
ADDC (buf, 'l');
|
|
#endif
|
|
ADDC (buf, ch);
|
|
}
|
|
else if (ch == ASE_T('C') || ch == ASE_T('S'))
|
|
{
|
|
#if defined(_WIN32)
|
|
ADDC (buf, ch);
|
|
#else
|
|
#ifdef ASE_CHAR_IS_MCHAR
|
|
ADDC (buf, 'l');
|
|
#endif
|
|
ADDC (buf, ASE_TOLOWER(ch));
|
|
#endif
|
|
}
|
|
else if (ch == ASE_T('d') || ch == ASE_T('i') ||
|
|
ch == ASE_T('o') || ch == ASE_T('u') ||
|
|
ch == ASE_T('x') || ch == ASE_T('X'))
|
|
{
|
|
if (modifier == MOD_SHORT)
|
|
{
|
|
ADDC (buf, 'h');
|
|
}
|
|
else if (modifier == MOD_LONG)
|
|
{
|
|
ADDC (buf, 'l');
|
|
}
|
|
else if (modifier == MOD_LONGLONG)
|
|
{
|
|
#if defined(_WIN32) && !defined(__LCC__)
|
|
ADDC (buf, 'I');
|
|
ADDC (buf, '6');
|
|
ADDC (buf, '4');
|
|
#else
|
|
ADDC (buf, 'l');
|
|
ADDC (buf, 'l');
|
|
#endif
|
|
}
|
|
ADDC (buf, ch);
|
|
}
|
|
else if (ch == ASE_T('\0')) break;
|
|
else ADDC (buf, ch);
|
|
}
|
|
|
|
buf.ptr[buf.len] = ASE_T('\0');
|
|
|
|
return buf.ptr;
|
|
}
|
|
|
|
#endif
|
|
|
|
int ase_dprintf (const ase_char_t* fmt, ...)
|
|
{
|
|
int n;
|
|
va_list ap;
|
|
|
|
va_start (ap, fmt);
|
|
n = ase_vfprintf (stderr, fmt, ap);
|
|
va_end (ap);
|
|
return n;
|
|
}
|
|
|
|
ASE_FILE* ase_fopen (const ase_char_t* path, const ase_char_t* mode)
|
|
{
|
|
#if defined(ASE_CHAR_IS_MCHAR)
|
|
return fopen (path, mode);
|
|
#elif defined(_WIN32)
|
|
return _wfopen (path, mode);
|
|
#else
|
|
|
|
char path_mb[PATH_MAX + 1];
|
|
char mode_mb[32];
|
|
size_t n;
|
|
|
|
n = wcstombs (path_mb, path, ASE_COUNTOF(path_mb));
|
|
if (n == (size_t)-1) return NULL;
|
|
if (n == ASE_COUNTOF(path_mb)) path_mb[ASE_COUNTOF(path_mb)-1] = '\0';
|
|
|
|
n = wcstombs (mode_mb, mode, ASE_COUNTOF(mode_mb));
|
|
if (n == (size_t)-1) return NULL;
|
|
if (n == ASE_COUNTOF(mode_mb)) path_mb[ASE_COUNTOF(mode_mb)-1] = '\0';
|
|
|
|
return fopen (path_mb, mode_mb);
|
|
#endif
|
|
}
|
|
|
|
ASE_FILE* ase_popen (const ase_char_t* cmd, const ase_char_t* mode)
|
|
{
|
|
#if defined(ASE_CHAR_IS_MCHAR)
|
|
return popen (cmd, mode);
|
|
#elif defined(_WIN32)
|
|
return _wpopen (cmd, mode);
|
|
#else
|
|
char cmd_mb[PATH_MAX + 1];
|
|
char mode_mb[32];
|
|
size_t n;
|
|
|
|
n = wcstombs (cmd_mb, cmd, ASE_COUNTOF(cmd_mb));
|
|
if (n == (size_t)-1) return NULL;
|
|
if (n == ASE_COUNTOF(cmd_mb)) cmd_mb[ASE_COUNTOF(cmd_mb)-1] = '\0';
|
|
|
|
n = wcstombs (mode_mb, mode, ASE_COUNTOF(mode_mb));
|
|
if (n == (size_t)-1) return NULL;
|
|
if (n == ASE_COUNTOF(mode_mb)) cmd_mb[ASE_COUNTOF(mode_mb)-1] = '\0';
|
|
|
|
return popen (cmd_mb, mode_mb);
|
|
#endif
|
|
}
|
|
|
|
static int isnl (const ase_char_t* ptr, ase_size_t len, void* delim)
|
|
{
|
|
return (ptr[len-1] == *(ase_char_t*)delim)? 1: 0;
|
|
}
|
|
|
|
ase_ssize_t ase_getline (ase_char_t **buf, ase_size_t *n, ASE_FILE *fp)
|
|
{
|
|
ase_char_t nl = ASE_T('\n');
|
|
return ase_getdelim (buf, n, isnl, &nl, fp);
|
|
}
|
|
|
|
ase_ssize_t ase_getdelim (
|
|
ase_char_t **buf, ase_size_t *n,
|
|
ase_getdelim_t fn, void* fnarg, ASE_FILE *fp)
|
|
{
|
|
ase_char_t* b;
|
|
ase_size_t capa;
|
|
ase_size_t len = 0;
|
|
int x;
|
|
|
|
ASE_ASSERT (buf != ASE_NULL);
|
|
ASE_ASSERT (n != ASE_NULL);
|
|
|
|
b = *buf;
|
|
capa = *n;
|
|
|
|
if (b == ASE_NULL)
|
|
{
|
|
capa = 256;
|
|
#if (defined(vms) || defined(__vms)) && (ASE_SIZEOF_VOID_P >= 8)
|
|
b = (ase_char_t*) _malloc32 (sizeof(ase_char_t)*(capa+1));
|
|
#else
|
|
b = (ase_char_t*) malloc (sizeof(ase_char_t)*(capa+1));
|
|
#endif
|
|
if (b == ASE_NULL) return -2;
|
|
}
|
|
|
|
if (ase_feof(fp))
|
|
{
|
|
len = (ase_size_t)-1;
|
|
goto exit_task;
|
|
}
|
|
|
|
while (1)
|
|
{
|
|
ase_cint_t c = ase_fgetc(fp);
|
|
if (c == ASE_CHAR_EOF)
|
|
{
|
|
if (ase_ferror(fp))
|
|
{
|
|
len = (ase_size_t)-2;
|
|
goto exit_task;
|
|
}
|
|
if (len == 0)
|
|
{
|
|
len = (ase_size_t)-1;
|
|
goto exit_task;
|
|
}
|
|
|
|
break;
|
|
}
|
|
|
|
if (len+1 >= capa)
|
|
{
|
|
ase_size_t ncapa = capa + 256;
|
|
ase_char_t* nb;
|
|
|
|
nb = realloc (b, ncapa*sizeof(ase_char_t));
|
|
if (nb == ASE_NULL)
|
|
{
|
|
len = (ase_size_t)-2;
|
|
goto exit_task;
|
|
}
|
|
|
|
b = nb;
|
|
capa = ncapa;
|
|
}
|
|
|
|
b[len++] = c;
|
|
|
|
x = fn (b, len, fnarg);
|
|
if (x < 0)
|
|
{
|
|
len = (ase_size_t)-3;
|
|
goto exit_task;
|
|
}
|
|
if (x > 0) break;
|
|
}
|
|
b[len] = ASE_T('\0');
|
|
|
|
exit_task:
|
|
*buf = b;
|
|
*n = capa;
|
|
|
|
return (ase_ssize_t)len;
|
|
}
|