windows-nt/Source/XPSP1/NT/base/crts/crtw32/lowio/write.c

280 lines
10 KiB
C
Raw Normal View History

2020-09-26 03:20:57 -05:00
/***
*write.c - write to a file handle
*
* Copyright (c) 1989-2001, Microsoft Corporation. All rights reserved.
*
*Purpose:
* defines _write() - write to a file handle
*
*Revision History:
* 06-14-89 PHG Module created, based on asm version
* 03-13-90 GJF Made calling type _CALLTYPE2 (for now), added #include
* <cruntime.h>, fixed compiler warnings and fixed the
* copyright. Also, cleaned up the formatting a bit.
* 04-03-90 GJF Now _CALLTYPE1.
* 07-24-90 SBM Removed '32' from API names
* 08-14-90 SBM Compiles cleanly with -W3
* 10-01-90 GJF New-style function declarators.
* 12-04-90 GJF Appended Win32 version onto source with #ifdef-s.
* Should come back latter and do a better merge.
* 12-04-90 SRW Changed to include <oscalls.h> instead of <doscalls.h>
* 12-06-90 SRW Changed to use _osfile and _osfhnd instead of _osfinfo
* 12-28-90 SRW Added _CRUISER_ conditional around check_stack pragma
* 12-28-90 SRW Added cast of void * to char * for Mips C Compiler
* 01-17-91 GJF ANSI naming.
* 02-25-91 MHL Adapt to ReadFile/WriteFile changes (_WIN32_)
* 04-09-91 PNT Added _MAC_ conditional
* 07-18-91 GJF Removed unreferenced local variable from _write_lk
* routine [_WIN32_].
* 10-24-91 GJF Added LPDWORD casts to make MIPS compiler happy.
* ASSUMES THAT sizeof(int) == sizeof(DWORD).
* 02-13-92 GJF Replaced _nfile by _nhandle for Win32.
* 02-15-92 GJF Increased BUF_SIZE and simplified LF translation code
* for Win32.
* 04-06-93 SKS Replace _CRTAPI* with __cdecl
* 09-06-94 CFW Remove Cruiser support.
* 09-06-94 CFW Replace MTHREAD with _MT.
* 12-03-94 SKS Clean up OS/2 references
* 01-04-95 GJF _WIN32_ -> _WIN32
* 02-15-95 GJF Appended Mac version of source file (somewhat cleaned
* up), with appropriate #ifdef-s.
* 06-12-95 GJF Changed _osfile[] and _osfhnd[] to _osfile() and
* _osfhnd(), which reference __pioinfo[].
* 06-27-95 GJF Added check that the file handle is open.
* 07-09-96 GJF Replaced defined(_WIN32) with !defined(_MAC) and
* defined(_M_M68K) || defined(_M_MPPC) with
* defined(_MAC). Also, detab-ed and cleaned up the
* format a bit.
* 12-30-97 GJF Exception-safe locking.
* 03-03-98 RKP Forced number of bytes written to always be an int
* 05-17-99 PML Remove all Macintosh support.
* 11-10-99 GB Replaced lseek for lseeki64 so as to able to append
* files longer than 4GB
*
*******************************************************************************/
#include <cruntime.h>
#include <oscalls.h>
#include <io.h>
#include <errno.h>
#include <msdos.h>
#include <mtdll.h>
#include <stdlib.h>
#include <string.h>
#include <internal.h>
#define BUF_SIZE 1025 /* size of LF translation buffer */
#define LF '\n' /* line feed */
#define CR '\r' /* carriage return */
#define CTRLZ 26 /* ctrl-z */
/***
*int _write(fh, buf, cnt) - write bytes to a file handle
*
*Purpose:
* Writes count bytes from the buffer to the handle specified.
* If the file was opened in text mode, each LF is translated to
* CR-LF. This does not affect the return value. In text
* mode ^Z indicates end of file.
*
* Multi-thread notes:
* (1) _write() - Locks/unlocks file handle
* _write_lk() - Does NOT lock/unlock file handle
*
*Entry:
* int fh - file handle to write to
* char *buf - buffer to write from
* unsigned int cnt - number of bytes to write
*
*Exit:
* returns number of bytes actually written.
* This may be less than cnt, for example, if out of disk space.
* returns -1 (and set errno) if fails.
*
*Exceptions:
*
*******************************************************************************/
#ifdef _MT
/* define normal version that locks/unlocks, validates fh */
int __cdecl _write (
int fh,
const void *buf,
unsigned cnt
)
{
int r; /* return value */
/* validate handle */
if ( ((unsigned)fh >= (unsigned)_nhandle) ||
!(_osfile(fh) & FOPEN) )
{
/* out of range -- return error */
errno = EBADF;
_doserrno = 0; /* not o.s. error */
return -1;
}
_lock_fh(fh); /* lock file */
__try {
if ( _osfile(fh) & FOPEN )
r = _write_lk(fh, buf, cnt); /* write bytes */
else {
errno = EBADF;
_doserrno = 0; /* not o.s. error */
r = -1;
}
}
__finally {
_unlock_fh(fh); /* unlock file */
}
return r;
}
/* now define version that doesn't lock/unlock, validate fh */
int __cdecl _write_lk (
int fh,
const void *buf,
unsigned cnt
)
{
int lfcount; /* count of line feeds */
int charcount; /* count of chars written so far */
int written; /* count of chars written on this write */
ULONG dosretval; /* o.s. return value */
char ch; /* current character */
char *p, *q; /* pointers into buf and lfbuf resp. */
char lfbuf[BUF_SIZE]; /* lf translation buffer */
#else
/* now define normal version */
int __cdecl _write (
int fh,
const void *buf,
unsigned cnt
)
{
int lfcount; /* count of line feeds */
int charcount; /* count of chars written so far */
int written; /* count of chars written on this write */
ULONG dosretval; /* o.s. return value */
char ch; /* current character */
char *p, *q; /* pointers into buf and lfbuf resp. */
char lfbuf[BUF_SIZE]; /* lf translation buffer */
/* validate handle */
if ( ((unsigned)fh >= (unsigned)_nhandle) ||
!(_osfile(fh) & FOPEN) )
{
/* out of range -- return error */
errno = EBADF;
_doserrno = 0; /* not o.s. error */
return -1;
}
#endif
lfcount = charcount = 0; /* nothing written yet */
if (cnt == 0)
return 0; /* nothing to do */
if (_osfile(fh) & FAPPEND) {
/* appending - seek to end of file; ignore error, because maybe
file doesn't allow seeking */
#if _INTEGRAL_MAX_BITS >= 64 /*IFSTRIP=IGN*/
(void)_lseeki64_lk(fh, 0, FILE_END);
#else
(void)_lseek_lk(fh, 0, FILE_END);
#endif
}
/* check for text mode with LF's in the buffer */
if ( _osfile(fh) & FTEXT ) {
/* text mode, translate LF's to CR/LF's on output */
p = (char *)buf; /* start at beginning of buffer */
dosretval = 0; /* no OS error yet */
while ( (unsigned)(p - (char *)buf) < cnt ) {
q = lfbuf; /* start at beginning of lfbuf */
/* fill the lf buf, except maybe last char */
while ( q - lfbuf < BUF_SIZE - 1 &&
(unsigned)(p - (char *)buf) < cnt ) {
ch = *p++;
if ( ch == LF ) {
++lfcount;
*q++ = CR;
}
*q++ = ch;
}
/* write the lf buf and update total */
if ( WriteFile( (HANDLE)_osfhnd(fh),
lfbuf,
(int)(q - lfbuf),
(LPDWORD)&written,
NULL) )
{
charcount += written;
if (written < q - lfbuf)
break;
}
else {
dosretval = GetLastError();
break;
}
}
}
else {
/* binary mode, no translation */
if ( WriteFile( (HANDLE)_osfhnd(fh),
(LPVOID)buf,
cnt,
(LPDWORD)&written,
NULL) )
{
dosretval = 0;
charcount = written;
}
else
dosretval = GetLastError();
}
if (charcount == 0) {
/* If nothing was written, first check if an o.s. error,
otherwise we return -1 and set errno to ENOSPC,
unless a device and first char was CTRL-Z */
if (dosretval != 0) {
/* o.s. error happened, map error */
if (dosretval == ERROR_ACCESS_DENIED) {
/* wrong read/write mode should return EBADF, not
EACCES */
errno = EBADF;
_doserrno = dosretval;
}
else
_dosmaperr(dosretval);
return -1;
}
else if ((_osfile(fh) & FDEV) && *(char *)buf == CTRLZ)
return 0;
else {
errno = ENOSPC;
_doserrno = 0; /* no o.s. error */
return -1;
}
}
else
/* return adjusted bytes written */
return charcount - lfcount;
}