xref: /inferno-os/utils/test/test-Nt.c (revision 74a4d8c26dd3c1e9febcb717cfd6cb6512991a7a)
1 /*
2  * POSIX standard
3  *	test expression
4  *	[ expression ]
5  */
6 
7 #include <lib9.h>
8 #include <windows.h>
9 
10 #define EQ(a,b)	((tmp=a)==0?0:(strcmp(tmp,b)==0))
11 
12 int	ap;
13 int	ac;
14 char	**av;
15 char	*tmp;
16 
17 void	synbad(char *, char *);
18 int	length(char *);
19 int	fsizep(char *);
20 int	isdir(char *);
21 int	isreg(char *);
22 int	Ntisatty(int);
23 int	isint(char *, int *);
24 int	tio(char *, int);
25 int	e(void), e1(void), e2(void), e3(void);
26 
27 void
main(int argc,char * argv[])28 main(int argc, char *argv[])
29 {
30 
31 	ac = argc; av = argv; ap = 1;
32 	if(EQ(argv[0],"[")) {
33 		if(!EQ(argv[--ac],"]"))
34 			synbad("] missing","");
35 	}
36 	argv[ac] = 0;
37 	if (ac<=1) exits("usage");
38 	exits(e()?0:"false");
39 }
40 
41 char *
nxtarg(int mt)42 nxtarg(int mt)
43 {
44 	if(ap>=ac){
45 		if(mt){
46 			ap++;
47 			return(0);
48 		}
49 		synbad("argument expected","");
50 	}
51 	return(av[ap++]);
52 }
53 
54 int
nxtintarg(int * pans)55 nxtintarg(int *pans)
56 {
57 	if(ap<ac && isint(av[ap], pans)){
58 		ap++;
59 		return 1;
60 	}
61 	return 0;
62 }
63 
64 int
e(void)65 e(void) {
66 	int p1;
67 
68 	p1 = e1();
69 	if (EQ(nxtarg(1), "-o")) return(p1 || e());
70 	ap--;
71 	return(p1);
72 }
73 
74 int
e1(void)75 e1(void) {
76 	int p1;
77 
78 	p1 = e2();
79 	if (EQ(nxtarg(1), "-a")) return (p1 && e1());
80 	ap--;
81 	return(p1);
82 }
83 
84 int
e2(void)85 e2(void) {
86 	if (EQ(nxtarg(0), "!"))
87 		return(!e2());
88 	ap--;
89 	return(e3());
90 }
91 
92 int
e3(void)93 e3(void) {
94 	int p1;
95 	char *a;
96 	char *p2;
97 	int int1, int2;
98 
99 	a = nxtarg(0);
100 	if(EQ(a, "(")) {
101 		p1 = e();
102 		if(!EQ(nxtarg(0), ")")) synbad(") expected","");
103 		return(p1);
104 	}
105 
106 	if(EQ(a, "-f"))
107 		return(isreg(nxtarg(0)));
108 
109 	if(EQ(a, "-d"))
110 		return(isdir(nxtarg(0)));
111 
112 	if(EQ(a, "-r"))
113 		return(tio(nxtarg(0), 4));
114 
115 	if(EQ(a, "-w"))
116 		return(tio(nxtarg(0), 2));
117 
118 	if(EQ(a, "-x"))
119 		return(tio(nxtarg(0), 1));
120 
121 	if(EQ(a, "-e"))
122 		return(tio(nxtarg(0), 0));
123 
124 	if(EQ(a, "-c"))
125 		return(0);
126 
127 	if(EQ(a, "-b"))
128 		return(0);
129 
130 	if(EQ(a, "-u"))
131 		return(0);
132 
133 	if(EQ(a, "-g"))
134 		return(0);
135 
136 	if(EQ(a, "-s"))
137 		return(fsizep(nxtarg(0)));
138 
139 	if(EQ(a, "-t"))
140 		if(ap>=ac || !nxtintarg(&int1))
141 			return(Ntisatty(1));
142 		else
143 			return(Ntisatty(int1));
144 
145 	if(EQ(a, "-n"))
146 		return(!EQ(nxtarg(0), ""));
147 	if(EQ(a, "-z"))
148 		return(EQ(nxtarg(0), ""));
149 
150 	p2 = nxtarg(1);
151 	if (p2==0)
152 		return(!EQ(a,""));
153 	if(EQ(p2, "="))
154 		return(EQ(nxtarg(0), a));
155 
156 	if(EQ(p2, "!="))
157 		return(!EQ(nxtarg(0), a));
158 
159 	if(!isint(a, &int1))
160 		return(!EQ(a,""));
161 
162 	if(nxtintarg(&int2)){
163 		if(EQ(p2, "-eq"))
164 			return(int1==int2);
165 		if(EQ(p2, "-ne"))
166 			return(int1!=int2);
167 		if(EQ(p2, "-gt"))
168 			return(int1>int2);
169 		if(EQ(p2, "-lt"))
170 			return(int1<int2);
171 		if(EQ(p2, "-ge"))
172 			return(int1>=int2);
173 		if(EQ(p2, "-le"))
174 			return(int1<=int2);
175 	}
176 
177 	synbad("unknown operator ",p2);
178 	return 0;		/* to shut ken up */
179 }
180 
181 int
tio(char * a,int f)182 tio(char *a, int f)
183 {
184 	return access (a, f) >= 0;
185 }
186 
187 /*
188  * dirstat fails for:
189  * [drivename]:
190  * [drivename]:/
191  * <dirname>/
192  * [drivename]:<dirname>/
193  */
194 int
isdir(char * f)195 isdir(char *f)
196 {
197 	Dir *dir;
198 	int	len;
199 
200 	len = strlen(f);
201 	if (f[(len-1)] == '/') {
202 		/* zap trailing '/' */
203 		f[(len-1)] = '\0';
204 	}
205 	if (len == 2) {
206 		if (f[1] == ':') {
207 			return DMDIR;
208 		}
209 	}
210 	if((dir = dirstat(f))==nil)
211 		return(0);
212 	return(dir->mode&DMDIR);
213 }
214 
215 int
isreg(char * f)216 isreg(char *f)
217 {
218 	Dir *dir;
219 
220 	if((dir = dirstat(f))==nil)
221 		return(0);
222 	return(!(dir->mode&DMDIR));
223 }
224 
225 int
Ntisatty(int fd)226 Ntisatty(int fd)
227 {
228 	HANDLE h;
229 
230 	if(fd < 0)
231 		return 0;
232 
233 	h = (HANDLE)_get_osfhandle(fd);
234 	if(h < 0)
235 		return 0;
236 	return _isatty((int)h);
237 }
238 
239 int
fsizep(char * f)240 fsizep(char *f)
241 {
242 	Dir *dir;
243 
244 	if((dir = dirstat(f))==nil)
245 		return(0);
246 	return(dir->length>0);
247 }
248 
249 void
synbad(char * s1,char * s2)250 synbad(char *s1, char *s2)
251 {
252 	int len;
253 
254 	write(2, "test: ", 6);
255 	if ((len = strlen(s1)) != 0)
256 		write(2, s1, len);
257 	if ((len = strlen(s2)) != 0)
258 		write(2, s2, len);
259 	write(2, "\n", 1);
260 	exits("bad syntax");
261 }
262 
263 int
length(char * s)264 length(char *s)
265 {
266 	char *es=s;
267 	while(*es++);
268 	return(es-s-1);
269 }
270 
271 int
isint(char * s,int * pans)272 isint(char *s, int *pans)
273 {
274 	char *ep;
275 
276 	*pans = strtol(s, &ep, 0);
277 	return (*ep == 0);
278 }
279