xref: /freebsd/contrib/less/lsystem.c (revision 0b3105a37d7adcadcb720112fed4dc4e8040be99)
1 /*
2  * Copyright (C) 1984-2015  Mark Nudelman
3  *
4  * You may distribute under the terms of either the GNU General Public
5  * License or the Less License, as specified in the README file.
6  *
7  * For more information, see the README file.
8  */
9 
10 
11 /*
12  * Routines to execute other programs.
13  * Necessarily very OS dependent.
14  */
15 
16 #include "less.h"
17 #include <signal.h>
18 #include "position.h"
19 
20 #if MSDOS_COMPILER
21 #include <dos.h>
22 #ifdef _MSC_VER
23 #include <direct.h>
24 #define setdisk(n) _chdrive((n)+1)
25 #else
26 #include <dir.h>
27 #endif
28 #endif
29 
30 extern int screen_trashed;
31 extern IFILE curr_ifile;
32 
33 
34 #if HAVE_SYSTEM
35 
36 /*
37  * Pass the specified command to a shell to be executed.
38  * Like plain "system()", but handles resetting terminal modes, etc.
39  */
40 	public void
41 lsystem(cmd, donemsg)
42 	char *cmd;
43 	char *donemsg;
44 {
45 	register int inp;
46 #if HAVE_SHELL
47 	register char *shell;
48 	register char *p;
49 #endif
50 	IFILE save_ifile;
51 #if MSDOS_COMPILER && MSDOS_COMPILER!=WIN32C
52 	char cwd[FILENAME_MAX+1];
53 #endif
54 
55 	/*
56 	 * Print the command which is to be executed,
57 	 * unless the command starts with a "-".
58 	 */
59 	if (cmd[0] == '-')
60 		cmd++;
61 	else
62 	{
63 		clear_bot();
64 		putstr("!");
65 		putstr(cmd);
66 		putstr("\n");
67 	}
68 
69 #if MSDOS_COMPILER
70 #if MSDOS_COMPILER==WIN32C
71 	if (*cmd == '\0')
72 		cmd = getenv("COMSPEC");
73 #else
74 	/*
75 	 * Working directory is global on MSDOS.
76 	 * The child might change the working directory, so we
77 	 * must save and restore CWD across calls to "system",
78 	 * or else we won't find our file when we return and
79 	 * try to "reedit_ifile" it.
80 	 */
81 	getcwd(cwd, FILENAME_MAX);
82 #endif
83 #endif
84 
85 	/*
86 	 * Close the current input file.
87 	 */
88 	save_ifile = save_curr_ifile();
89 	(void) edit_ifile(NULL_IFILE);
90 
91 	/*
92 	 * De-initialize the terminal and take out of raw mode.
93 	 */
94 	deinit();
95 	flush();	/* Make sure the deinit chars get out */
96 	raw_mode(0);
97 #if MSDOS_COMPILER==WIN32C
98 	close_getchr();
99 #endif
100 
101 	/*
102 	 * Restore signals to their defaults.
103 	 */
104 	init_signals(0);
105 
106 #if HAVE_DUP
107 	/*
108 	 * Force standard input to be the user's terminal
109 	 * (the normal standard input), even if less's standard input
110 	 * is coming from a pipe.
111 	 */
112 	inp = dup(0);
113 	close(0);
114 #if OS2
115 	/* The __open() system call translates "/dev/tty" to "con". */
116 	if (__open("/dev/tty", OPEN_READ) < 0)
117 #else
118 	if (open("/dev/tty", OPEN_READ) < 0)
119 #endif
120 		dup(inp);
121 #endif
122 
123 	/*
124 	 * Pass the command to the system to be executed.
125 	 * If we have a SHELL environment variable, use
126 	 * <$SHELL -c "command"> instead of just <command>.
127 	 * If the command is empty, just invoke a shell.
128 	 */
129 #if HAVE_SHELL
130 	p = NULL;
131 	if ((shell = lgetenv("SHELL")) != NULL && *shell != '\0')
132 	{
133 		if (*cmd == '\0')
134 			p = save(shell);
135 		else
136 		{
137 			char *esccmd = shell_quote(cmd);
138 			if (esccmd != NULL)
139 			{
140 				int len = (int) (strlen(shell) + strlen(esccmd) + 5);
141 				p = (char *) ecalloc(len, sizeof(char));
142 				SNPRINTF3(p, len, "%s %s %s", shell, shell_coption(), esccmd);
143 				free(esccmd);
144 			}
145 		}
146 	}
147 	if (p == NULL)
148 	{
149 		if (*cmd == '\0')
150 			p = save("sh");
151 		else
152 			p = save(cmd);
153 	}
154 	system(p);
155 	free(p);
156 #else
157 #if MSDOS_COMPILER==DJGPPC
158 	/*
159 	 * Make stdin of the child be in cooked mode.
160 	 */
161 	setmode(0, O_TEXT);
162 	/*
163 	 * We don't need to catch signals of the child (it
164 	 * also makes trouble with some DPMI servers).
165 	 */
166 	__djgpp_exception_toggle();
167   	system(cmd);
168 	__djgpp_exception_toggle();
169 #else
170 	system(cmd);
171 #endif
172 #endif
173 
174 #if HAVE_DUP
175 	/*
176 	 * Restore standard input, reset signals, raw mode, etc.
177 	 */
178 	close(0);
179 	dup(inp);
180 	close(inp);
181 #endif
182 
183 #if MSDOS_COMPILER==WIN32C
184 	open_getchr();
185 #endif
186 	init_signals(1);
187 	raw_mode(1);
188 	if (donemsg != NULL)
189 	{
190 		putstr(donemsg);
191 		putstr("  (press RETURN)");
192 		get_return();
193 		putchr('\n');
194 		flush();
195 	}
196 	init();
197 	screen_trashed = 1;
198 
199 #if MSDOS_COMPILER && MSDOS_COMPILER!=WIN32C
200 	/*
201 	 * Restore the previous directory (possibly
202 	 * changed by the child program we just ran).
203 	 */
204 	chdir(cwd);
205 #if MSDOS_COMPILER != DJGPPC
206 	/*
207 	 * Some versions of chdir() don't change to the drive
208 	 * which is part of CWD.  (DJGPP does this in chdir.)
209 	 */
210 	if (cwd[1] == ':')
211 	{
212 		if (cwd[0] >= 'a' && cwd[0] <= 'z')
213 			setdisk(cwd[0] - 'a');
214 		else if (cwd[0] >= 'A' && cwd[0] <= 'Z')
215 			setdisk(cwd[0] - 'A');
216 	}
217 #endif
218 #endif
219 
220 	/*
221 	 * Reopen the current input file.
222 	 */
223 	reedit_ifile(save_ifile);
224 
225 #if defined(SIGWINCH) || defined(SIGWIND)
226 	/*
227 	 * Since we were ignoring window change signals while we executed
228 	 * the system command, we must assume the window changed.
229 	 * Warning: this leaves a signal pending (in "sigs"),
230 	 * so psignals() should be called soon after lsystem().
231 	 */
232 	winch(0);
233 #endif
234 }
235 
236 #endif
237 
238 #if PIPEC
239 
240 /*
241  * Pipe a section of the input file into the given shell command.
242  * The section to be piped is the section "between" the current
243  * position and the position marked by the given letter.
244  *
245  * If the mark is after the current screen, the section between
246  * the top line displayed and the mark is piped.
247  * If the mark is before the current screen, the section between
248  * the mark and the bottom line displayed is piped.
249  * If the mark is on the current screen, or if the mark is ".",
250  * the whole current screen is piped.
251  */
252 	public int
253 pipe_mark(c, cmd)
254 	int c;
255 	char *cmd;
256 {
257 	POSITION mpos, tpos, bpos;
258 
259 	/*
260 	 * mpos = the marked position.
261 	 * tpos = top of screen.
262 	 * bpos = bottom of screen.
263 	 */
264 	mpos = markpos(c);
265 	if (mpos == NULL_POSITION)
266 		return (-1);
267 	tpos = position(TOP);
268 	if (tpos == NULL_POSITION)
269 		tpos = ch_zero();
270 	bpos = position(BOTTOM);
271 
272  	if (c == '.')
273  		return (pipe_data(cmd, tpos, bpos));
274  	else if (mpos <= tpos)
275  		return (pipe_data(cmd, mpos, bpos));
276  	else if (bpos == NULL_POSITION)
277  		return (pipe_data(cmd, tpos, bpos));
278  	else
279  		return (pipe_data(cmd, tpos, mpos));
280 }
281 
282 /*
283  * Create a pipe to the given shell command.
284  * Feed it the file contents between the positions spos and epos.
285  */
286 	public int
287 pipe_data(cmd, spos, epos)
288 	char *cmd;
289 	POSITION spos;
290 	POSITION epos;
291 {
292 	register FILE *f;
293 	register int c;
294 	extern FILE *popen();
295 
296 	/*
297 	 * This is structured much like lsystem().
298 	 * Since we're running a shell program, we must be careful
299 	 * to perform the necessary deinitialization before running
300 	 * the command, and reinitialization after it.
301 	 */
302 	if (ch_seek(spos) != 0)
303 	{
304 		error("Cannot seek to start position", NULL_PARG);
305 		return (-1);
306 	}
307 
308 	if ((f = popen(cmd, "w")) == NULL)
309 	{
310 		error("Cannot create pipe", NULL_PARG);
311 		return (-1);
312 	}
313 	clear_bot();
314 	putstr("!");
315 	putstr(cmd);
316 	putstr("\n");
317 
318 	deinit();
319 	flush();
320 	raw_mode(0);
321 	init_signals(0);
322 #if MSDOS_COMPILER==WIN32C
323 	close_getchr();
324 #endif
325 #ifdef SIGPIPE
326 	LSIGNAL(SIGPIPE, SIG_IGN);
327 #endif
328 
329 	c = EOI;
330 	while (epos == NULL_POSITION || spos++ <= epos)
331 	{
332 		/*
333 		 * Read a character from the file and give it to the pipe.
334 		 */
335 		c = ch_forw_get();
336 		if (c == EOI)
337 			break;
338 		if (putc(c, f) == EOF)
339 			break;
340 	}
341 
342 	/*
343 	 * Finish up the last line.
344 	 */
345  	while (c != '\n' && c != EOI )
346  	{
347  		c = ch_forw_get();
348  		if (c == EOI)
349  			break;
350  		if (putc(c, f) == EOF)
351  			break;
352  	}
353 
354 	pclose(f);
355 
356 #ifdef SIGPIPE
357 	LSIGNAL(SIGPIPE, SIG_DFL);
358 #endif
359 #if MSDOS_COMPILER==WIN32C
360 	open_getchr();
361 #endif
362 	init_signals(1);
363 	raw_mode(1);
364 	init();
365 	screen_trashed = 1;
366 #if defined(SIGWINCH) || defined(SIGWIND)
367 	/* {{ Probably don't need this here. }} */
368 	winch(0);
369 #endif
370 	return (0);
371 }
372 
373 #endif
374