1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
|
/***********************************************************************
* *
* This software is part of the ast package *
* Copyright (c) 1985-2011 AT&T Intellectual Property *
* and is licensed under the *
* Eclipse Public License, Version 1.0 *
* by AT&T Intellectual Property *
* *
* A copy of the License is available at *
* http://www.eclipse.org/org/documents/epl-v10.html *
* (with md5 checksum b35adb5213ca9657e911e9befb180842) *
* *
* Information and Software Systems Research *
* AT&T Research *
* Florham Park NJ *
* *
* Glenn Fowler <gsf@research.att.com> *
* David Korn <dgk@research.att.com> *
* Phong Vo <kpv@research.att.com> *
* *
***********************************************************************/
#include "sfhdr.h"
/* Put out a null-terminated string
**
** Written by Kiem-Phong Vo.
*/
#if __STD_C
ssize_t sfputr(Sfio_t* f, const char* s, int rc)
#else
ssize_t sfputr(f,s,rc)
Sfio_t* f; /* write to this stream */
char* s; /* string to write */
int rc; /* record separator. */
#endif
{
ssize_t p, n, w, sn;
uchar *ps;
char *ss;
SFMTXDECL(f);
SFMTXENTER(f,-1);
if(f->mode != SF_WRITE && _sfmode(f,SF_WRITE,0) < 0)
SFMTXRETURN(f, -1);
SFLOCK(f,0);
f->val = sn = -1; ss = (char*)s;
for(w = 0; (*s || rc >= 0); )
{ /* need to communicate string size to exception handler */
if((f->flags&SF_STRING) && f->next >= f->endb )
{ sn = sn < 0 ? strlen(s) : (sn - (s-ss));
ss = (char*)s; /* save current checkpoint */
f->val = sn + (rc >= 0 ? 1 : 0); /* space requirement */
f->bits |= SF_PUTR; /* tell sfflsbuf to use f->val */
}
SFWPEEK(f,ps,p);
f->bits &= ~SF_PUTR; /* remove any trace of this */
if(p < 0 ) /* something not right about buffering */
break;
if(p == 0 || (f->flags&SF_WHOLE) )
{ n = sn < 0 ? strlen(s) : sn - (s-ss);
if(p >= (n + (rc < 0 ? 0 : 1)) )
{ /* buffer can hold everything */
if(n > 0)
{ memcpy(ps, s, n);
ps += n;
w += n;
}
if(rc >= 0)
{ *ps++ = rc;
w += 1;
}
f->next = ps;
}
else
{ /* create a reserve buffer to hold data */
Sfrsrv_t* rsrv;
p = n + (rc >= 0 ? 1 : 0);
if(!(rsrv = _sfrsrv(f, p)) )
n = 0;
else
{ if(n > 0)
memcpy(rsrv->data, s, n);
if(rc >= 0)
rsrv->data[n] = rc;
if((n = SFWRITE(f,rsrv->data,p)) < 0 )
n = 0;
}
w += n;
}
break;
}
if(*s == 0)
{ *ps++ = rc;
f->next = ps;
w += 1;
break;
}
#if _lib_memccpy && !__ia64 /* these guys may never get it right */
if((ps = (uchar*)memccpy(ps,s,'\0',p)) != NIL(uchar*))
ps -= 1;
else ps = f->next+p;
s += ps - f->next;
#else
for(; p > 0; --p, ++ps, ++s)
if((*ps = *s) == 0)
break;
#endif
w += ps - f->next;
f->next = ps;
}
/* sync unseekable shared streams */
if(f->extent < 0 && (f->flags&SF_SHARE) )
(void)SFFLSBUF(f,-1);
/* check for line buffering */
else if((f->flags&SF_LINE) && !(f->flags&SF_STRING) && (n = f->next-f->data) > 0)
{ if(n > w)
n = w;
f->next -= n;
(void)SFWRITE(f,(Void_t*)f->next,n);
}
SFOPEN(f,0);
SFMTXRETURN(f, w);
}
|
