fbc9d208be7a08a1b4c518894f0e317d1413be5b
[pspp] / host.c
1 /* PSPP - a program for statistical analysis.
2    Copyright (C) 1997-9, 2000, 2009, 2010, 2011 Free Software Foundation, Inc.
3
4    This program is free software: you can redistribute it and/or modify
5    it under the terms of the GNU General Public License as published by
6    the Free Software Foundation, either version 3 of the License, or
7    (at your option) any later version.
8
9    This program is distributed in the hope that it will be useful,
10    but WITHOUT ANY WARRANTY; without even the implied warranty of
11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12    GNU General Public License for more details.
13
14    You should have received a copy of the GNU General Public License
15    along with this program.  If not, see <http://www.gnu.org/licenses/>. */
16
17 #include <config.h>
18
19 #include <stdio.h>
20 #include <stdlib.h>
21 #include <ctype.h>
22 #include <errno.h>
23 #include <unistd.h>
24 #if HAVE_SYS_WAIT_H
25 #include <sys/wait.h>
26 #endif
27
28 #include "data/settings.h"
29 #include "language/command.h"
30 #include "language/lexer/lexer.h"
31 #include "libpspp/assertion.h"
32 #include "libpspp/compiler.h"
33 #include "libpspp/i18n.h"
34 #include "libpspp/message.h"
35 #include "libpspp/str.h"
36
37 #include "gl/localcharset.h"
38 #include "gl/xalloc.h"
39 #include "gl/xmalloca.h"
40
41 #include "gettext.h"
42 #define _(msgid) gettext (msgid)
43 #define N_(msgid) msgid
44 \f
45 #if HAVE_FORK && HAVE_EXECL
46 /* Spawn an interactive shell process. */
47 static bool
48 shell (void)
49 {
50   int pid;
51
52   pid = fork ();
53   switch (pid)
54     {
55     case 0:
56       {
57         const char *shell_fn;
58         char *shell_process;
59
60         {
61           int i;
62
63           for (i = 3; i < 20; i++)
64             close (i);
65         }
66
67         shell_fn = getenv ("SHELL");
68         if (shell_fn == NULL)
69           shell_fn = "/bin/sh";
70
71         {
72           const char *cp = strrchr (shell_fn, '/');
73           cp = cp ? &cp[1] : shell_fn;
74           shell_process = xmalloca (strlen (cp) + 8);
75           strcpy (shell_process, "-");
76           strcat (shell_process, cp);
77           if (strcmp (cp, "sh"))
78             shell_process[0] = '+';
79         }
80
81         execl (shell_fn, shell_process, NULL);
82
83         _exit (1);
84       }
85
86     case -1:
87       msg (SE, _("Couldn't fork: %s."), strerror (errno));
88       return false;
89
90     default:
91       assert (pid > 0);
92       while (wait (NULL) != pid)
93         ;
94       return true;
95     }
96 }
97 #else /* !(HAVE_FORK && HAVE_EXECL) */
98 /* Don't know how to spawn an interactive shell. */
99 static bool
100 shell (void)
101 {
102   msg (SE, _("Interactive shell not supported on this platform."));
103   return false;
104 }
105 #endif
106
107 /* Executes the specified COMMAND in a subshell.  Returns true if
108    successful, false otherwise. */
109 static bool
110 run_command (const char *command)
111 {
112   if (system (NULL) == 0)
113     {
114       msg (SE, _("Command shell not supported on this platform."));
115       return false;
116     }
117
118   /* Execute the command. */
119   if (system (command) == -1)
120     msg (SE, _("Error executing command: %s."), strerror (errno));
121
122   return true;
123 }
124
125 int
126 cmd_host (struct lexer *lexer, struct dataset *ds UNUSED)
127 {
128   if (settings_get_safer_mode ())
129     {
130       msg (SE, _("This command not allowed when the SAFER option is set."));
131       return CMD_FAILURE;
132     }
133
134   if (lex_token (lexer) == T_ENDCMD)
135     return shell () ? CMD_SUCCESS : CMD_FAILURE;
136   else if (lex_match_id (lexer, "COMMAND"))
137     {
138       struct string command;
139       char *locale_command;
140       bool ok;
141
142       lex_match (lexer, T_EQUALS);
143       if (!lex_force_match (lexer, T_LBRACK))
144         return CMD_FAILURE;
145
146       ds_init_empty (&command);
147       while (lex_is_string (lexer))
148         {
149           if (!ds_is_empty (&command))
150             ds_put_byte (&command, '\n');
151           ds_put_substring (&command, lex_tokss (lexer));
152           lex_get (lexer);
153         }
154       if (!lex_force_match (lexer, T_RBRACK))
155         {
156           ds_destroy (&command);
157           return CMD_FAILURE;
158         }
159
160       locale_command = recode_string (locale_charset (), "UTF-8",
161                                       ds_cstr (&command),
162                                       ds_length (&command));
163       ds_destroy (&command);
164
165       ok = run_command (locale_command);
166       free (locale_command);
167
168       return ok ? CMD_SUCCESS : CMD_FAILURE;
169     }
170   else
171     {
172       lex_error (lexer, NULL);
173       return CMD_FAILURE;
174     }
175 }