Blame src/lib/libast/sfio/sfread.c

Packit 992a25
/***********************************************************************
Packit 992a25
*                                                                      *
Packit 992a25
*               This software is part of the ast package               *
Packit 992a25
*          Copyright (c) 1985-2011 AT&T Intellectual Property          *
Packit 992a25
*                      and is licensed under the                       *
Packit 992a25
*                 Eclipse Public License, Version 1.0                  *
Packit 992a25
*                    by AT&T Intellectual Property                     *
Packit 992a25
*                                                                      *
Packit 992a25
*                A copy of the License is available at                 *
Packit 992a25
*          http://www.eclipse.org/org/documents/epl-v10.html           *
Packit 992a25
*         (with md5 checksum b35adb5213ca9657e911e9befb180842)         *
Packit 992a25
*                                                                      *
Packit 992a25
*              Information and Software Systems Research               *
Packit 992a25
*                            AT&T Research                             *
Packit 992a25
*                           Florham Park NJ                            *
Packit 992a25
*                                                                      *
Packit 992a25
*                 Glenn Fowler <gsf@research.att.com>                  *
Packit 992a25
*                  David Korn <dgk@research.att.com>                   *
Packit 992a25
*                   Phong Vo <kpv@research.att.com>                    *
Packit 992a25
*                                                                      *
Packit 992a25
***********************************************************************/
Packit 992a25
#include	"sfhdr.h"
Packit 992a25
Packit 992a25
/*	Read n bytes from a stream into a buffer
Packit 992a25
**
Packit 992a25
**	Written by Kiem-Phong Vo.
Packit 992a25
*/
Packit 992a25
Packit 992a25
#if __STD_C
Packit 992a25
ssize_t sfread(Sfio_t* f, Void_t* buf, size_t n)
Packit 992a25
#else
Packit 992a25
ssize_t sfread(f,buf,n)
Packit 992a25
Sfio_t*		f;	/* read from this stream. 	*/
Packit 992a25
Void_t*		buf;	/* buffer to read into		*/
Packit 992a25
size_t		n;	/* number of bytes to be read. 	*/
Packit 992a25
#endif
Packit 992a25
{
Packit 992a25
	reg uchar	*s, *begs;
Packit 992a25
	reg ssize_t	r;
Packit 992a25
	reg int		local, justseek;
Packit 992a25
	SFMTXDECL(f);
Packit 992a25
Packit 992a25
	SFMTXENTER(f, (ssize_t)(-1));
Packit 992a25
Packit 992a25
	GETLOCAL(f,local);
Packit 992a25
	justseek = f->bits&SF_JUSTSEEK; f->bits &= ~SF_JUSTSEEK;
Packit 992a25
Packit 992a25
	if(!buf)
Packit 992a25
		SFMTXRETURN(f, (ssize_t)(n == 0 ? 0 : -1) );
Packit 992a25
Packit 992a25
	/* release peek lock */
Packit 992a25
	if(f->mode&SF_PEEK)
Packit 992a25
	{	if(!(f->mode&SF_READ) )
Packit 992a25
			SFMTXRETURN(f, (ssize_t)(-1));
Packit 992a25
Packit 992a25
		if(f->mode&SF_GETR)
Packit 992a25
		{	if(((uchar*)buf + f->val) != f->next &&
Packit 992a25
			   (!f->rsrv || f->rsrv->data != (uchar*)buf) )
Packit 992a25
				SFMTXRETURN(f, (ssize_t)(-1));
Packit 992a25
			f->mode &= ~SF_PEEK;
Packit 992a25
			SFMTXRETURN(f, 0);
Packit 992a25
		}
Packit 992a25
		else
Packit 992a25
		{	if((uchar*)buf != f->next)
Packit 992a25
				SFMTXRETURN(f, (ssize_t)(-1));
Packit 992a25
			f->mode &= ~SF_PEEK;
Packit 992a25
			if(f->mode&SF_PKRD)
Packit 992a25
			{	/* actually read the data now */
Packit 992a25
				f->mode &= ~SF_PKRD;
Packit 992a25
				if(n > 0)
Packit 992a25
					n = (r = sysreadf(f->file,f->data,n)) < 0 ? 0 : r;
Packit 992a25
				f->endb = f->data+n;
Packit 992a25
				f->here += n;
Packit 992a25
			}
Packit 992a25
			f->next += n;
Packit 992a25
			f->endr = f->endb;
Packit 992a25
			SFMTXRETURN(f, n);
Packit 992a25
		}
Packit 992a25
	}
Packit 992a25
Packit 992a25
	s = begs = (uchar*)buf;
Packit 992a25
	for(;; f->mode &= ~SF_LOCK)
Packit 992a25
	{	/* check stream mode */
Packit 992a25
		if(SFMODE(f,local) != SF_READ && _sfmode(f,SF_READ,local) < 0)
Packit 992a25
		{	n = s > begs ? s-begs : (size_t)(-1);
Packit 992a25
			SFMTXRETURN(f, (ssize_t)n);
Packit 992a25
		}
Packit 992a25
Packit 992a25
		SFLOCK(f,local);
Packit 992a25
Packit 992a25
		if((r = f->endb - f->next) > 0) /* has buffered data */
Packit 992a25
		{	if(r > (ssize_t)n)
Packit 992a25
				r = (ssize_t)n;
Packit 992a25
			if(s != f->next)
Packit 992a25
				memcpy(s, f->next, r);
Packit 992a25
			f->next += r;
Packit 992a25
			s += r;
Packit 992a25
			n -= r;
Packit 992a25
		}
Packit 992a25
Packit 992a25
		if(n <= 0)	/* all done */
Packit 992a25
			break;
Packit 992a25
Packit 992a25
		if(!(f->flags&SF_STRING) && !(f->bits&SF_MMAP) )
Packit 992a25
		{	f->next = f->endb = f->data;
Packit 992a25
Packit 992a25
			/* exact IO is desirable for these cases */
Packit 992a25
			if(SFDIRECT(f,n) ||
Packit 992a25
			   ((f->flags&SF_SHARE) && f->extent < 0) )
Packit 992a25
				r = (ssize_t)n;
Packit 992a25
			else if(justseek && n <= f->iosz && f->iosz <= f->size)
Packit 992a25
				r = f->iosz;	/* limit buffering */
Packit 992a25
			else	r = f->size;	/* full buffering */
Packit 992a25
Packit 992a25
			/* if read almost full size, then just do it direct */
Packit 992a25
			if(r > (ssize_t)n && (r - r/8) <= (ssize_t)n)
Packit 992a25
				r = (ssize_t)n;
Packit 992a25
Packit 992a25
			/* read directly to user's buffer */
Packit 992a25
			if(r == (ssize_t)n && (r = SFRD(f,s,r,f->disc)) >= 0)
Packit 992a25
			{	s += r;
Packit 992a25
				n -= r;
Packit 992a25
				if(r == 0 || n == 0) /* eof or eob */ 
Packit 992a25
					break;
Packit 992a25
			}
Packit 992a25
			else	goto do_filbuf;
Packit 992a25
		}
Packit 992a25
		else
Packit 992a25
		{ do_filbuf:
Packit 992a25
			if(justseek)
Packit 992a25
				f->bits |= SF_JUSTSEEK;
Packit 992a25
			if(SFFILBUF(f,-1) <= 0)
Packit 992a25
				break;
Packit 992a25
		}
Packit 992a25
	}
Packit 992a25
Packit 992a25
	SFOPEN(f,local);
Packit 992a25
	r = s-begs;
Packit 992a25
	SFMTXRETURN(f, r);
Packit 992a25
}