sfputr.c revision 1
0N/A/***********************************************************************
77N/A* *
0N/A* This software is part of the ast package *
0N/A* Copyright (c) 1985-2011 AT&T Intellectual Property *
0N/A* and is licensed under the *
0N/A* Common Public License, Version 1.0 *
0N/A* by AT&T Intellectual Property *
0N/A* *
0N/A* A copy of the License is available at *
0N/A* http://www.opensource.org/licenses/cpl1.0.txt *
0N/A* (with md5 checksum 059e8cd6165cb4c31e351f2b69388fd9) *
0N/A* *
0N/A* Information and Software Systems Research *
0N/A* AT&T Research *
0N/A* Florham Park NJ *
0N/A* *
0N/A* Glenn Fowler <gsf@research.att.com> *
0N/A* David Korn <dgk@research.att.com> *
0N/A* Phong Vo <kpv@research.att.com> *
0N/A* *
0N/A***********************************************************************/
0N/A#include "sfhdr.h"
0N/A
0N/A/* Put out a null-terminated string
0N/A**
0N/A** Written by Kiem-Phong Vo.
0N/A*/
0N/A#if __STD_C
0N/Assize_t sfputr(Sfio_t* f, const char* s, int rc)
0N/A#else
0N/Assize_t sfputr(f,s,rc)
0N/ASfio_t* f; /* write to this stream */
0N/Achar* s; /* string to write */
0N/Aint rc; /* record separator. */
0N/A#endif
77N/A{
77N/A reg ssize_t p, n, w;
0N/A reg uchar* ps;
0N/A SFMTXDECL(f);
0N/A
0N/A SFMTXENTER(f,-1);
0N/A
0N/A if(f->mode != SF_WRITE && _sfmode(f,SF_WRITE,0) < 0)
0N/A SFMTXRETURN(f, -1);
0N/A
0N/A SFLOCK(f,0);
0N/A
0N/A for(w = 0; (*s || rc >= 0); )
0N/A { if(SFWPEEK(f,ps,p) < 0)
0N/A break;
0N/A
0N/A if(p == 0 || (f->flags&SF_WHOLE) )
0N/A { n = strlen(s);
0N/A if(p >= (n + (rc < 0 ? 0 : 1)) )
0N/A { /* buffer can hold everything */
0N/A if(n > 0)
0N/A { memcpy(ps, s, n);
0N/A ps += n;
0N/A w += n;
0N/A }
0N/A if(rc >= 0)
0N/A { *ps++ = rc;
0N/A w += 1;
0N/A }
0N/A f->next = ps;
0N/A }
0N/A else
0N/A { /* create a reserve buffer to hold data */
0N/A Sfrsrv_t* rsrv;
0N/A
0N/A p = n + (rc >= 0 ? 1 : 0);
0N/A if(!(rsrv = _sfrsrv(f, p)) )
0N/A n = 0;
0N/A else
0N/A { if(n > 0)
0N/A memcpy(rsrv->data, s, n);
0N/A if(rc >= 0)
0N/A rsrv->data[n] = rc;
0N/A if((n = SFWRITE(f,rsrv->data,p)) < 0 )
0N/A n = 0;
0N/A }
0N/A
0N/A w += n;
0N/A }
0N/A break;
0N/A }
0N/A
0N/A if(*s == 0)
0N/A { *ps++ = rc;
0N/A f->next = ps;
0N/A w += 1;
0N/A break;
0N/A }
0N/A
0N/A#if _lib_memccpy && !__ia64 /* these guys may never get it right */
0N/A if((ps = (uchar*)memccpy(ps,s,'\0',p)) != NIL(uchar*))
0N/A ps -= 1;
0N/A else ps = f->next+p;
0N/A s += ps - f->next;
0N/A#else
0N/A for(; p > 0; --p, ++ps, ++s)
0N/A if((*ps = *s) == 0)
0N/A break;
0N/A#endif
0N/A w += ps - f->next;
0N/A f->next = ps;
0N/A }
0N/A
0N/A /* sync unseekable shared streams */
0N/A if(f->extent < 0 && (f->flags&SF_SHARE) )
0N/A (void)SFFLSBUF(f,-1);
0N/A
0N/A /* check for line buffering */
0N/A else if((f->flags&SF_LINE) && !(f->flags&SF_STRING) && (n = f->next-f->data) > 0)
0N/A { if(n > w)
0N/A n = w;
0N/A f->next -= n;
0N/A (void)SFWRITE(f,(Void_t*)f->next,n);
0N/A }
0N/A
0N/A SFOPEN(f,0);
0N/A SFMTXRETURN(f, w);
0N/A}
0N/A