freebsd-dev/contrib/sendmail/libsm/smstdio.c
2005-02-14 02:30:00 +00:00

364 lines
6.1 KiB
C

/*
* Copyright (c) 2000-2002, 2004 Sendmail, Inc. and its suppliers.
* All rights reserved.
*
* By using this file, you agree to the terms and conditions set
* forth in the LICENSE file which can be found at the top level of
* the sendmail distribution.
*/
#include <sm/gen.h>
SM_IDSTR(id, "@(#)$Id: smstdio.c,v 1.34 2004/08/03 20:46:34 ca Exp $")
#include <unistd.h>
#include <stdio.h>
#include <fcntl.h>
#include <errno.h>
#include <sys/stat.h>
#include <sm/assert.h>
#include <sm/io.h>
#include <sm/string.h>
#include "local.h"
static void setup __P((SM_FILE_T *));
/*
** Overall:
** This is a file type which implements a layer on top of the system
** stdio. fp->f_cookie is the FILE* of stdio. The cookie may be
** "bound late" because of the manner which Linux implements stdio.
** When binding late (when fp->f_cookie==NULL) then the value of
** fp->f_ival is used (0, 1 or 2) to map to stdio's stdin, stdout or
** stderr.
*/
/*
** SM_STDIOOPEN -- open a file to system stdio implementation
**
** Parameters:
** fp -- file pointer assign for this open
** info -- info about file to open
** flags -- indicating method of opening
** rpool -- ignored
**
** Returns:
** Failure: -1
** Success: 0 (zero)
*/
/* ARGSUSED3 */
int
sm_stdioopen(fp, info, flags, rpool)
SM_FILE_T *fp;
const void *info;
int flags;
const void *rpool;
{
register FILE *s;
char *stdiomode;
switch (flags)
{
case SM_IO_RDONLY:
stdiomode = "r";
break;
case SM_IO_WRONLY:
stdiomode = "w";
break;
case SM_IO_APPEND:
stdiomode = "a";
break;
case SM_IO_APPENDRW:
stdiomode = "a+";
break;
#if SM_IO_BINARY != 0
case SM_IO_RDONLY_B:
stdiomode = "rb";
break;
case SM_IO_WRONLY_B:
stdiomode = "wb";
break;
case SM_IO_APPEND_B:
stdiomode = "ab";
break;
case SM_IO_APPENDRW_B:
stdiomode = "a+b";
break;
case SM_IO_RDWR_B:
stdiomode = "r+b";
break;
#endif /* SM_IO_BINARY != 0 */
case SM_IO_RDWR:
default:
stdiomode = "r+";
break;
}
if ((s = fopen((char *)info, stdiomode)) == NULL)
return -1;
fp->f_cookie = s;
return 0;
}
/*
** SETUP -- assign file type cookie when not already assigned
**
** Parameters:
** fp - the file pointer to get the cookie assigned
**
** Return:
** none.
*/
static void
setup(fp)
SM_FILE_T *fp;
{
if (fp->f_cookie == NULL)
{
switch (fp->f_ival)
{
case 0:
fp->f_cookie = stdin;
break;
case 1:
fp->f_cookie = stdout;
break;
case 2:
fp->f_cookie = stderr;
break;
default:
sm_abort("fp->f_ival=%d: out of range (0...2)", fp->f_ival);
break;
}
}
}
/*
** SM_STDIOREAD -- read from the file
**
** Parameters:
** fp -- the file pointer
** buf -- location to place the read data
** n - number of bytes to read
**
** Returns:
** result from fread().
*/
ssize_t
sm_stdioread(fp, buf, n)
SM_FILE_T *fp;
char *buf;
size_t n;
{
register FILE *s;
if (fp->f_cookie == NULL)
setup(fp);
s = fp->f_cookie;
return fread(buf, 1, n, s);
}
/*
** SM_STDIOWRITE -- write to the file
**
** Parameters:
** fp -- the file pointer
** buf -- location of data to write
** n - number of bytes to write
**
** Returns:
** result from fwrite().
*/
ssize_t
sm_stdiowrite(fp, buf, n)
SM_FILE_T *fp;
char const *buf;
size_t n;
{
register FILE *s;
if (fp->f_cookie == NULL)
setup(fp);
s = fp->f_cookie;
return fwrite(buf, 1, n, s);
}
/*
** SM_STDIOSEEK -- set position within file
**
** Parameters:
** fp -- the file pointer
** offset -- new location based on 'whence'
** whence -- indicates "base" for 'offset'
**
** Returns:
** result from fseek().
*/
off_t
sm_stdioseek(fp, offset, whence)
SM_FILE_T *fp;
off_t offset;
int whence;
{
register FILE *s;
if (fp->f_cookie == NULL)
setup(fp);
s = fp->f_cookie;
return fseek(s, offset, whence);
}
/*
** SM_STDIOCLOSE -- close the file
**
** Parameters:
** fp -- close file pointer
**
** Return:
** status from fclose()
*/
int
sm_stdioclose(fp)
SM_FILE_T *fp;
{
register FILE *s;
if (fp->f_cookie == NULL)
setup(fp);
s = fp->f_cookie;
return fclose(s);
}
/*
** SM_STDIOSETINFO -- set info for this open file
**
** Parameters:
** fp -- the file pointer
** what -- type of information setting
** valp -- memory location of info to set
**
** Return:
** Failure: -1 and sets errno
** Success: none (currently).
*/
/* ARGSUSED2 */
int
sm_stdiosetinfo(fp, what, valp)
SM_FILE_T *fp;
int what;
void *valp;
{
switch (what)
{
case SM_IO_WHAT_MODE:
default:
errno = EINVAL;
return -1;
}
}
/*
** SM_STDIOGETINFO -- get info for this open file
**
** Parameters:
** fp -- the file pointer
** what -- type of information request
** valp -- memory location to place info
**
** Return:
** Failure: -1 and sets errno
** Success: none (currently).
*/
/* ARGSUSED2 */
int
sm_stdiogetinfo(fp, what, valp)
SM_FILE_T *fp;
int what;
void *valp;
{
switch (what)
{
case SM_IO_WHAT_SIZE:
{
int fd;
struct stat st;
if (fp->f_cookie == NULL)
setup(fp);
fd = fileno((FILE *) fp->f_cookie);
if (fd < 0)
return -1;
if (fstat(fd, &st) == 0)
return st.st_size;
else
return -1;
}
case SM_IO_WHAT_MODE:
default:
errno = EINVAL;
return -1;
}
}
/*
** SM_IO_STDIOOPEN -- create an SM_FILE which interfaces to a stdio FILE
**
** Parameters:
** stream -- an open stdio stream, as returned by fopen()
** mode -- the mode argument to fopen() which describes stream
**
** Return:
** On success, return a pointer to an SM_FILE object which
** can be used for reading and writing 'stream'.
** Abort if mode is gibberish or stream is bad.
** Raise an exception if we can't allocate memory.
*/
SM_FILE_T *
sm_io_stdioopen(stream, mode)
FILE *stream;
char *mode;
{
int fd;
bool r, w;
int ioflags;
SM_FILE_T *fp;
fd = fileno(stream);
SM_REQUIRE(fd >= 0);
r = w = false;
switch (mode[0])
{
case 'r':
r = true;
break;
case 'w':
case 'a':
w = true;
break;
default:
sm_abort("sm_io_stdioopen: mode '%s' is bad", mode);
}
if (strchr(&mode[1], '+') != NULL)
r = w = true;
if (r && w)
ioflags = SMRW;
else if (r)
ioflags = SMRD;
else
ioflags = SMWR;
fp = sm_fp(SmFtRealStdio, ioflags, NULL);
fp->f_file = fd;
fp->f_cookie = stream;
return fp;
}