Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

395 lines
11 KiB

/***
*filebuf.cpp - core filebuf member functions
*
* Copyright (c) 1991-2001, Microsoft Corporation. All rights reserved.
*
*Purpose:
* Contains the core member functions for filebuf class.
*
*Revision History:
* 08-08-91 KRS Created.
* 08-20-91 KRS Added virtual xsgetn()/xsputn() functions.
* 08-21-91 KRS Fix circular reference between sync() and seekoff().
* Close file in destructor only if we opened it!
* 09-06-91 KRS Fix ios::ate case in filebuf::open().
* 09-09-91 KRS Add support for ios::binary in filebuf::open().
* 09-10-91 KRS Remove virtual xsputn()/xsgetn().
* 09-11-91 KRS Fix filebuf::seekoff() for ios::cur and in_avail().
* 09-12-91 KRS Make sure close() always closes even if sync() fails.
* Fix seekoff call in filebuf::sync() and pbackfail().
* 09-16-91 KRS Make virtual filebuf::setbuf() more robust.
* 09-19-91 KRS Add calls to delbuf(1) in constructors.
* 09-20-91 KRS C700 #4453: Improve efficiency in overflow().
* 09-29-91 KRS Granularity split. Move fstream into separate file.
* 10-24-91 KRS Avoid virtual calls from virtual functions.
* 11-13-91 KRS Use allocate() properly in overflow() and underflow().
* Fix constructor.
* 01-03-92 KRS Remove virtual keyword. Add function headers and PCH.
* 01-20-92 KRS In text mode, account for CR/LF pairs in sync().
* 02-03-92 KRS Change for new compiler destructor behavior.
* 08-19-92 KRS Remove sh_compat for NT.
* 08-27-92 KRS Fix bug in close() introduced in MTHREAD work.
* 02-23-95 CFW Fix bug: buffer overwritten when disk full.
* 06-14-95 CFW Comment cleanup.
* 06-19-95 GJF Replaced _osfile[] with _osfile() (which references
* a field in the ioinfo struct).
* 05-17-99 PML Remove all Macintosh support.
*
*******************************************************************************/
#include <cruntime.h>
#include <internal.h>
#include <string.h>
#include <stdio.h>
#include <fcntl.h>
#include <crtdbg.h>
#include <sys\types.h>
#include <io.h>
#include <fstream.h>
#pragma hdrstop
#include <msdos.h>
#include <sys\stat.h>
const int filebuf::openprot = 0644;
const int filebuf::sh_none = 04000; // deny rw
const int filebuf::sh_read = 05000; // deny wr
const int filebuf::sh_write = 06000; // deny rd
const int filebuf::binary = O_BINARY;
const int filebuf::text = O_TEXT;
/***
*filebuf::filebuf() - filebuf default constructor
*
*Purpose:
* Default constructor.
*
*Entry:
*
*******************************************************************************/
filebuf::filebuf()
: streambuf()
{
x_fOpened = 0;
x_fd = -1;
}
/***
*filebuf::filebuf(filedesc fd) - filebuf constructor
*
*Purpose:
* Constructor. Initialize filebuf and attach to file descriptor.
*
*Entry:
* fd = file descriptor to attach to filebuf
*
*******************************************************************************/
filebuf::filebuf(filedesc fd)
: streambuf()
{
x_fOpened = 0;
x_fd=fd;
}
/***
*filebuf::filebuf(filedesc fd, char* p, int len) - filebuf constructor
*
*Purpose:
* Constructor. Initialize filebuf and attach to file descriptor.
*
*Entry:
* fd = file descriptor to attach to filebuf
* p = user-supplied buffer
* len = length of buffer
*
*******************************************************************************/
filebuf::filebuf(filedesc fd, char* p, int len)
: streambuf()
{
filebuf::setbuf(p,len);
x_fOpened = 0;
x_fd=fd;
}
/***
*filebuf::~filebuf() - filebuf destructor
*
*Purpose:
* Destructor. Close attached file only if we opened it.
*
*Entry:
* None.
*
*******************************************************************************/
filebuf::~filebuf()
{
lock(); // no need to unlock...
if (x_fOpened)
close(); // calls filebuf::sync()
else
filebuf::sync();
}
/***
*filebuf* filebuf::close() - close an attached file
*
*Purpose:
* Close attached file.
*
*Entry:
* None.
*Exit:
* Returns NULL if error, otherwise returns "this" pointer.
*
*******************************************************************************/
filebuf* filebuf::close()
{
int retval;
if (x_fd==-1)
return NULL;
lock();
retval = sync();
if ((_close(x_fd)==-1) || (retval==EOF))
{
unlock();
return NULL;
}
x_fd = -1;
unlock();
return this;
}
/***
*virtual int filebuf::overflow(int c) - overflow virtual function
*
*Purpose:
* flush any characters in the reserve area and handle 'c'.
*
*Entry:
* c = character to output (if not EOF)
*
*Exit:
* Returns EOF if error, otherwise returns something else.
*
*Exceptions:
* Returns EOF if error.
*
*******************************************************************************/
int filebuf::overflow(int c)
{
if (allocate()==EOF) // make sure there is a reserve area
return EOF;
if (filebuf::sync()==EOF) // sync before new buffer created below
return EOF;
if (!unbuffered())
setp(base(),ebuf());
if (c!=EOF)
{
if ((!unbuffered()) && (pptr() < epptr())) // guard against recursion
sputc(c);
else
{
if (_write(x_fd,&c,1)!=1)
return(EOF);
}
}
return(1); // return something other than EOF if successful
}
/***
*virtual int filebuf::underflow() - underflow virtual function
*
*Purpose:
* return next character in get area, or get more characters from source.
*
*Entry:
* None.
*
*Exit:
* Returns current character in file. Does not advance get pointer.
*
*Exceptions:
* Returns EOF if error.
*
*******************************************************************************/
int filebuf::underflow()
{
int count;
unsigned char tbuf;
if (in_avail())
return (int)(unsigned char) *gptr();
if (allocate()==EOF) // make sure there is a reserve area
return EOF;
if (filebuf::sync()==EOF)
return EOF;
if (unbuffered())
{
if (_read(x_fd,(void *)&tbuf,1)<=0)
return EOF;
return (int)tbuf;
}
if ((count=_read(x_fd,(void *)base(),blen())) <= 0)
return EOF; // reached EOF
setg(base(),base(),base()+count);
return (int)(unsigned char) *gptr();
}
/***
*virtual streampos filebuf::seekoff() - seekoff virtual function
*
*Purpose:
* Seeks to given absolute or relative file offset.
*
*Entry:
* off = offset to seek to relative to beginning, end or current
* position in the file.
* dir = one of ios::beg, ios::cur, or ios::end
*
*Exit:
* Returns current file position after seek.
*
*Exceptions:
* Returns EOF if error.
*
*******************************************************************************/
streampos filebuf::seekoff(streamoff off, ios::seek_dir dir, int)
{
int fdir;
long retpos;
switch (dir) {
case ios::beg :
fdir = SEEK_SET;
break;
case ios::cur :
fdir = SEEK_CUR;
break;
case ios::end :
fdir = SEEK_END;
break;
default:
// error
return(EOF);
}
if (filebuf::sync()==EOF)
return EOF;
if ((retpos=_lseek(x_fd, off, fdir))==-1L)
return (EOF);
return((streampos)retpos);
}
/***
*virtual int filebuf::sync() - synchronize buffers with external file postion.
*
*Purpose:
* Synchronizes buffer with external file, by flushing any output and/or
* discarding any unread input data. Discards any get or put area(s).
*
*Entry:
* None.
*
*Exit:
* Returns EOF if error, else 0.
*
*Exceptions:
* Returns EOF if error.
*
*******************************************************************************/
int filebuf::sync()
{
long count, nout;
char * p;
if (x_fd==-1)
return(EOF);
if (!unbuffered())
{
if ((count=out_waiting())!=0)
{
if ((nout =_write(x_fd,(void *) pbase(),(unsigned int)count)) != count)
{
if (nout > 0) {
// should set _pptr -= nout
pbump(-(int)nout);
memmove(pbase(), pbase()+nout, (int)(count-nout));
}
return(EOF);
}
}
setp(0,0); // empty put area
if ((count=in_avail()) > 0)
{
// can't use seekoff here!!
if (_osfile(x_fd) & FTEXT)
{
// If text mode, need to account for CR/LF etc.
for (p = gptr(); p < egptr(); p++)
if (*p == '\n')
count++;
// account for EOF if read, not counted by _read
if ((_osfile(x_fd) & FEOFLAG))
count++;
}
if (_lseek(x_fd, -count, SEEK_CUR)==-1L)
{
// unlock();
return (EOF);
}
}
setg(0,0,0); // empty get area
}
// unlock();
return(0);
}
/***
*virtual streambuf* filebuf::setbuf(char* ptr, int len) - set reserve area.
*
*Purpose:
* Synchronizes buffer with external file, by flushing any output and/or
* discarding any unread input data. Discards any get or put area(s).
*
*Entry:
* ptr = requested reserve area. If NULL, request is for unbuffered.
* len = size of reserve area. If <= 0, request is for unbuffered.
*
*Exit:
* Returns this pointer if request is honored, else NULL.
*
*Exceptions:
* Returns NULL if request is not honored.
*
*******************************************************************************/
streambuf * filebuf::setbuf(char * ptr, int len)
{
if (is_open() && (ebuf()))
return NULL;
if ((!ptr) || (len <= 0))
unbuffered(1);
else
{
lock();
setb(ptr, ptr+len, 0);
unlock();
}
return this;
}