ascii: Correct multiple bugs regarding output width.
[pspp] / src / ui / terminal / main.c
1 /* PSPP - a program for statistical analysis.
2    Copyright (C) 1997-2000, 2006-2007, 2009-2014 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 <signal.h>
20 #include <stdio.h>
21 #include <stdlib.h>
22 #if HAVE_FPU_CONTROL_H
23 #include <fpu_control.h>
24 #endif
25 #if HAVE_FENV_H
26 #include <fenv.h>
27 #endif
28 #if HAVE_IEEEFP_H
29 #include <ieeefp.h>
30 #endif
31 #include <unistd.h>
32
33 #include "data/dataset.h"
34 #include "data/dictionary.h"
35 #include "data/file-handle-def.h"
36 #include "data/session.h"
37 #include "data/settings.h"
38 #include "data/variable.h"
39 #include "gsl/gsl_errno.h"
40 #include "language/command.h"
41 #include "language/lexer/lexer.h"
42 #include "language/lexer/include-path.h"
43 #include "libpspp/argv-parser.h"
44 #include "libpspp/compiler.h"
45 #include "libpspp/i18n.h"
46 #include "libpspp/message.h"
47 #include "libpspp/version.h"
48 #include "math/random.h"
49 #include "output/driver.h"
50 #include "output/message-item.h"
51 #include "ui/source-init-opts.h"
52 #include "ui/terminal/terminal-opts.h"
53 #include "ui/terminal/terminal-reader.h"
54
55 #include "gl/fatal-signal.h"
56 #include "gl/progname.h"
57 #include "gl/relocatable.h"
58
59 #include "gettext.h"
60 #define _(msgid) gettext (msgid)
61
62 static struct session *the_session;
63
64 static void add_syntax_reader (struct lexer *, const char *file_name,
65                                const char *encoding, enum lex_syntax_mode);
66 static void bug_handler(int sig);
67 static void fpu_init (void);
68 static void output_msg (const struct msg *, void *);
69
70 /* Program entry point. */
71 int
72 main (int argc, char **argv)
73 {
74   struct terminal_opts *terminal_opts;
75   struct argv_parser *parser;
76   enum lex_syntax_mode syntax_mode;
77   char *syntax_encoding;
78   bool process_statrc;
79   struct lexer *lexer;
80
81   set_program_name (argv[0]);
82
83   signal (SIGABRT, bug_handler);
84   signal (SIGSEGV, bug_handler);
85   signal (SIGFPE, bug_handler);
86
87   i18n_init ();
88   fpu_init ();
89   gsl_set_error_handler_off ();
90
91   output_engine_push ();
92   fh_init ();
93   settings_init ();
94   random_init ();
95
96   lexer = lex_create ();
97   the_session = session_create (NULL);
98   dataset_create (the_session, "");
99
100   parser = argv_parser_create ();
101   terminal_opts = terminal_opts_init (parser, &syntax_mode, &process_statrc,
102                                       &syntax_encoding);
103   source_init_register_argv_parser (parser);
104   if (!argv_parser_run (parser, argc, argv))
105     exit (EXIT_FAILURE);
106   terminal_opts_done (terminal_opts, argc, argv);
107   argv_parser_destroy (parser);
108
109   msg_set_handler (output_msg, lexer);
110   session_set_default_syntax_encoding (the_session, syntax_encoding);
111
112   /* Add syntax files to source stream. */
113   if (process_statrc)
114     {
115       char *rc = include_path_search ("rc");
116       if (rc != NULL)
117         {
118           add_syntax_reader (lexer, rc, "Auto", LEX_SYNTAX_AUTO);
119           free (rc);
120         }
121     }
122   if (optind < argc)
123     {
124       int i;
125
126       for (i = optind; i < argc; i++)
127         add_syntax_reader (lexer, argv[i], syntax_encoding, syntax_mode);
128     }
129   else
130     add_syntax_reader (lexer, "-", syntax_encoding, syntax_mode);
131
132   /* Parse and execute syntax. */
133   lex_get (lexer);
134   for (;;)
135     {
136       int result = cmd_parse (lexer, session_active_dataset (the_session));
137
138       if (result == CMD_EOF || result == CMD_FINISH)
139         break;
140       else if (cmd_result_is_failure (result) && lex_token (lexer) != T_STOP)
141         {
142           if (lex_get_error_mode (lexer) == LEX_ERROR_STOP)
143             {
144               msg (MW, _("Error encountered while ERROR=STOP is effective."));
145               lex_discard_noninteractive (lexer);
146             }
147           else if (result == CMD_CASCADING_FAILURE
148                    && lex_get_error_mode (lexer) != LEX_ERROR_TERMINAL)
149             {
150               msg (SE, _("Stopping syntax file processing here to avoid "
151                          "a cascade of dependent command failures."));
152               lex_discard_noninteractive (lexer);
153             }
154         }
155
156       if (msg_ui_too_many_errors ())
157         lex_discard_noninteractive (lexer);
158     }
159
160
161   output_engine_pop ();
162   session_destroy (the_session);
163
164   random_done ();
165   settings_done ();
166   fh_done ();
167   lex_destroy (lexer);
168   i18n_done ();
169
170   return msg_ui_any_errors ();
171 }
172 \f
173 static void
174 fpu_init (void)
175 {
176 #if HAVE_FEHOLDEXCEPT
177   fenv_t foo;
178   feholdexcept (&foo);
179 #elif HAVE___SETFPUCW && defined(_FPU_IEEE)
180   __setfpucw (_FPU_IEEE);
181 #elif HAVE_FPSETMASK
182   fpsetmask (0);
183 #endif
184 }
185
186 /* If a segfault happens, issue a message to that effect and halt */
187 static void
188 bug_handler(int sig)
189 {
190   /* Reset SIG to its default handling so that if it happens again we won't
191      recurse. */
192   signal (sig, SIG_DFL);
193
194   switch (sig)
195     {
196     case SIGABRT:
197       request_bug_report("Assertion Failure/Abort");
198       break;
199     case SIGFPE:
200       request_bug_report("Floating Point Exception");
201       break;
202     case SIGSEGV:
203       request_bug_report("Segmentation Violation");
204       break;
205     default:
206       request_bug_report("Unknown");
207       break;
208     }
209
210   /* Re-raise the signal so that we terminate with the correct status. */
211   raise (sig);
212 }
213
214 static void
215 output_msg (const struct msg *m_, void *lexer_)
216 {
217   struct lexer *lexer = lexer_;
218   struct msg m = *m_;
219
220   if (m.file_name == NULL)
221     {
222       m.file_name = CONST_CAST (char *, lex_get_file_name (lexer));
223       m.first_line = lex_get_first_line_number (lexer, 0);
224       m.last_line = lex_get_last_line_number (lexer, 0);
225     }
226
227   m.command_name = output_get_command_name ();
228
229   message_item_submit (message_item_create (&m));
230
231   free (m.command_name);
232 }
233
234 static void
235 add_syntax_reader (struct lexer *lexer, const char *file_name,
236                    const char *encoding, enum lex_syntax_mode syntax_mode)
237 {
238   struct lex_reader *reader;
239
240   reader = (!strcmp (file_name, "-") && isatty (STDIN_FILENO)
241             ? terminal_reader_create ()
242             : lex_reader_for_file (file_name, encoding, syntax_mode,
243                                    LEX_ERROR_CONTINUE));
244
245   if (reader)
246     lex_append (lexer, reader);
247 }