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