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
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
|
/***********************************************************************
* *
* This software is part of the ast package *
* Copyright (c) 1985-2010 AT&T Intellectual Property *
* and is licensed under the *
* Common Public License, Version 1.0 *
* by AT&T Intellectual Property *
* *
* A copy of the License is available at *
* http://www.opensource.org/licenses/cpl1.0.txt *
* (with md5 checksum 059e8cd6165cb4c31e351f2b69388fd9) *
* *
* 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 "sfdchdr.h"
/*
* a discipline that prepends a prefix string to each output line
*
* Glenn Fowler
* AT&T Research
*
* @(#)$Id: sfdcprefix (AT&T Research) 1998-06-25 $
*/
typedef struct
{
Sfdisc_t disc; /* sfio discipline */
size_t length; /* prefix length */
size_t empty; /* empty line prefix length */
int skip; /* this line already prefixed */
char prefix[1]; /* prefix string */
} Prefix_t;
/*
* prefix write
*/
#if __STD_C
static ssize_t pfxwrite(Sfio_t* f, const Void_t* buf, register size_t n, Sfdisc_t* dp)
#else
static ssize_t pfxwrite(f, buf, n, dp)
Sfio_t* f;
Void_t* buf;
register size_t n;
Sfdisc_t* dp;
#endif
{
register Prefix_t* pfx = (Prefix_t*)dp;
register char* b;
register char* s;
register char* e;
register char* t;
register ssize_t w;
int skip;
skip = 0;
w = 0;
b = (char*)buf;
s = b;
e = s + n;
do
{
if (!(t = memchr(s, '\n', e - s)))
{
skip = 1;
t = e - 1;
}
n = t - s + 1;
if (pfx->skip)
pfx->skip = 0;
else
sfwr(f, pfx->prefix, n > 1 ? pfx->length : pfx->empty, dp);
w += sfwr(f, s, n, dp);
if ((s = t + 1) >= e)
return w;
} while ((s = t + 1) < e);
pfx->skip = skip;
return w;
}
/*
* remove the discipline on close
*/
#if __STD_C
static int pfxexcept(Sfio_t* f, int type, Void_t* data, Sfdisc_t* dp)
#else
static int pfxexcept(f, type, data, dp)
Sfio_t* f;
int type;
Void_t* data;
Sfdisc_t* dp;
#endif
{
if (type == SF_FINAL || type == SF_DPOP)
free(dp);
return 0;
}
/*
* push the prefix discipline on f
*/
#if __STD_C
int sfdcprefix(Sfio_t* f, const char* prefix)
#else
int sfdcprefix(f, prefix)
Sfio_t* f;
char* prefix;
#endif
{
register Prefix_t* pfx;
register char* s;
size_t n;
/*
* this is a writeonly discipline
*/
if (!prefix || !(n = strlen(prefix)) || !(sfset(f, 0, 0) & SF_WRITE))
return -1;
if (!(pfx = (Prefix_t*)malloc(sizeof(Prefix_t) + n)))
return -1;
memset(pfx, 0, sizeof(*pfx));
pfx->disc.writef = pfxwrite;
pfx->disc.exceptf = pfxexcept;
pfx->length = n;
memcpy(pfx->prefix, prefix, n);
s = (char*)prefix + n;
while (--s > (char*)prefix && (*s == ' ' || *s == '\t'));
n = s - (char*)prefix;
if (*s != ' ' || *s != '\t')
n++;
pfx->empty = n;
if (sfdisc(f, &pfx->disc) != &pfx->disc)
{
free(pfx);
return -1;
}
return 0;
}
|
