da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin/***********************************************************************
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* This software is part of the ast package *
3e14f97f673e8a630f076077de35afdd43dc1587Roger A. Faulkner* Copyright (c) 1985-2010 AT&T Intellectual Property *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* and is licensed under the *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* Common Public License, Version 1.0 *
7c2fbfb345896881c631598ee3852ce9ce33fb07April Chin* by AT&T Intellectual Property *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* A copy of the License is available at *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* http://www.opensource.org/licenses/cpl1.0.txt *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* (with md5 checksum 059e8cd6165cb4c31e351f2b69388fd9) *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* Information and Software Systems Research *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* AT&T Research *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* Florham Park NJ *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* Glenn Fowler <gsf@research.att.com> *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* David Korn <dgk@research.att.com> *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* Phong Vo <kpv@research.att.com> *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin***********************************************************************/
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin#include "sfhdr.h"
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin/* Read a record delineated by a character.
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin** The record length can be accessed via sfvalue(f).
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin**
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin** Written by Kiem-Phong Vo
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin*/
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin#if __STD_C
7c2fbfb345896881c631598ee3852ce9ce33fb07April Chinchar* sfgetr(Sfio_t *f, int rc, int type)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin#else
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chinchar* sfgetr(f,rc,type)
7c2fbfb345896881c631598ee3852ce9ce33fb07April ChinSfio_t* f; /* stream to read from */
7c2fbfb345896881c631598ee3852ce9ce33fb07April Chinint rc; /* record separator */
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chinint type;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin#endif
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin{
7c2fbfb345896881c631598ee3852ce9ce33fb07April Chin ssize_t n, un;
7c2fbfb345896881c631598ee3852ce9ce33fb07April Chin uchar *s, *ends, *us;
7c2fbfb345896881c631598ee3852ce9ce33fb07April Chin int found;
7c2fbfb345896881c631598ee3852ce9ce33fb07April Chin Sfrsrv_t* rsrv;
7c2fbfb345896881c631598ee3852ce9ce33fb07April Chin SFMTXDECL(f);
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
7c2fbfb345896881c631598ee3852ce9ce33fb07April Chin SFMTXENTER(f, NIL(char*));
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if(rc < 0 || (f->mode != SF_READ && _sfmode(f,SF_READ,0) < 0) )
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin SFMTXRETURN(f, NIL(char*));
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin SFLOCK(f,0);
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin /* buffer to be returned */
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin rsrv = NIL(Sfrsrv_t*);
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin us = NIL(uchar*);
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin un = 0;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin found = 0;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin /* compatibility mode */
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin type = type < 0 ? SF_LASTR : type == 1 ? SF_STRING : type;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if(type&SF_LASTR) /* return the broken record */
7c2fbfb345896881c631598ee3852ce9ce33fb07April Chin { if((f->flags&SF_STRING) && (un = f->endb - f->next))
7c2fbfb345896881c631598ee3852ce9ce33fb07April Chin { us = f->next;
7c2fbfb345896881c631598ee3852ce9ce33fb07April Chin f->next = f->endb;
7c2fbfb345896881c631598ee3852ce9ce33fb07April Chin found = 1;
7c2fbfb345896881c631598ee3852ce9ce33fb07April Chin }
7c2fbfb345896881c631598ee3852ce9ce33fb07April Chin else if((rsrv = f->rsrv) && (un = -rsrv->slen) > 0)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin { us = rsrv->data;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin found = 1;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin goto done;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin while(!found)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin { /* fill buffer if necessary */
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if((n = (ends = f->endb) - (s = f->next)) <= 0)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin { /* for unseekable devices, peek-read 1 record */
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin f->getr = rc;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin f->mode |= SF_RC;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin /* fill buffer the conventional way */
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if(SFRPEEK(f,s,n) <= 0)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin { us = NIL(uchar*);
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin goto done;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin else
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin { ends = s+n;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if(f->mode&SF_RC)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin { s = ends[-1] == rc ? ends-1 : ends;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin goto do_copy;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin#if _lib_memchr
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if(!(s = (uchar*)memchr((char*)s,rc,n)))
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin s = ends;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin#else
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin while(*s != rc)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if((s += 1) == ends)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin break;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin#endif
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin do_copy:
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if(s < ends) /* found separator */
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin { s += 1; /* include the separator */
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin found = 1;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if(!us &&
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin (!(type&SF_STRING) || !(f->flags&SF_STRING) ||
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin ((f->flags&SF_STRING) && (f->bits&SF_BOTH) ) ) )
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin { /* returning data in buffer */
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin us = f->next;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin un = s - f->next;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin f->next = s;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin goto done;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin /* amount to be read */
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin n = s - f->next;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
7c2fbfb345896881c631598ee3852ce9ce33fb07April Chin if(!found && (_Sfmaxr > 0 && un+n+1 >= _Sfmaxr || (f->flags&SF_STRING))) /* already exceed limit */
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin { us = NIL(uchar*);
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin goto done;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin /* get internal buffer */
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if(!rsrv || rsrv->size < un+n+1)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin { if(rsrv)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin rsrv->slen = un;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if((rsrv = _sfrsrv(f,un+n+1)) != NIL(Sfrsrv_t*))
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin us = rsrv->data;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin else
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin { us = NIL(uchar*);
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin goto done;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin /* now copy data */
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin s = us+un;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin un += n;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin ends = f->next;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin f->next += n;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin MEMCPY(s,ends,n);
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chindone:
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin _Sfi = f->val = un;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin f->getr = 0;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if(found && rc != 0 && (type&SF_STRING) )
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin { us[un-1] = '\0';
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if(us >= f->data && us < f->endb)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin { f->getr = rc;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin f->mode |= SF_GETR;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin /* prepare for a call to get the broken record */
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if(rsrv)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin rsrv->slen = found ? 0 : -un;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin SFOPEN(f,0);
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if(us && (type&SF_LOCKR) )
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin { f->mode |= SF_PEEK|SF_GETR;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin f->endr = f->data;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin SFMTXRETURN(f, (char*)us);
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin}