session: Fix two memory leaks.
[pspp] / src / language / dictionary / numeric.c
1 /* PSPP - a program for statistical analysis.
2    Copyright (C) 1997-9, 2000, 2006, 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 <stdlib.h>
20
21 #include "data/dataset.h"
22 #include "data/dictionary.h"
23 #include "data/variable.h"
24 #include "data/format.h"
25 #include "language/command.h"
26 #include "language/lexer/format-parser.h"
27 #include "language/lexer/lexer.h"
28 #include "language/lexer/variable-parser.h"
29 #include "libpspp/assertion.h"
30 #include "libpspp/message.h"
31 #include "libpspp/str.h"
32
33 #include "gettext.h"
34 #define _(msgid) gettext (msgid)
35
36 /* Parses the NUMERIC command. */
37 int
38 cmd_numeric (struct lexer *lexer, struct dataset *ds)
39 {
40   size_t i;
41
42   /* Names of variables to create. */
43   char **v;
44   size_t nv;
45
46   do
47     {
48       /* Format spec for variables to create.  f.type==-1 if default is to
49          be used. */
50       struct fmt_spec f;
51
52       if (!parse_DATA_LIST_vars (lexer, dataset_dict (ds),
53                                  &v, &nv, PV_NO_DUPLICATE))
54         return CMD_FAILURE;
55
56       /* Get the optional format specification. */
57       if (lex_match (lexer, T_LPAREN))
58         {
59           if (!parse_format_specifier (lexer, &f))
60             goto fail;
61
62           if ( ! fmt_check_output (&f))
63             goto fail;
64
65           if (fmt_is_string (f.type))
66             {
67               char str[FMT_STRING_LEN_MAX + 1];
68               msg (SE, _("Format type %s may not be used with a numeric "
69                          "variable."), fmt_to_string (&f, str));
70               goto fail;
71             }
72
73           if (!lex_match (lexer, T_RPAREN))
74             {
75               lex_error_expecting (lexer, "`)'", NULL_SENTINEL);
76               goto fail;
77             }
78         }
79       else
80         f.type = -1;
81
82       /* Create each variable. */
83       for (i = 0; i < nv; i++)
84         {
85           struct variable *new_var = dict_create_var (dataset_dict (ds), v[i], 0);
86           if (!new_var)
87             msg (SE, _("There is already a variable named %s."), v[i]);
88           else
89             {
90               if (f.type != -1)
91                 var_set_both_formats (new_var, &f);
92             }
93         }
94
95       /* Clean up. */
96       for (i = 0; i < nv; i++)
97         free (v[i]);
98       free (v);
99     }
100   while (lex_match (lexer, T_SLASH));
101
102   return CMD_SUCCESS;
103
104   /* If we have an error at a point where cleanup is required,
105      flow-of-control comes here. */
106 fail:
107   for (i = 0; i < nv; i++)
108     free (v[i]);
109   free (v);
110   return CMD_FAILURE;
111 }
112
113 /* Parses the STRING command. */
114 int
115 cmd_string (struct lexer *lexer, struct dataset *ds)
116 {
117   size_t i;
118
119   /* Names of variables to create. */
120   char **v;
121   size_t nv;
122
123   /* Format spec for variables to create. */
124   struct fmt_spec f;
125
126   /* Width of variables to create. */
127   int width;
128
129   do
130     {
131       if (!parse_DATA_LIST_vars (lexer, dataset_dict (ds),
132                                  &v, &nv, PV_NO_DUPLICATE))
133         return CMD_FAILURE;
134
135       if (!lex_force_match (lexer, T_LPAREN)
136           || !parse_format_specifier (lexer, &f)
137           || !lex_force_match (lexer, T_RPAREN))
138         goto fail;
139       if (!fmt_is_string (f.type))
140         {
141           char str[FMT_STRING_LEN_MAX + 1];
142           msg (SE, _("Format type %s may not be used with a string "
143                      "variable."), fmt_to_string (&f, str));
144           goto fail;
145         }
146       if (!fmt_check_output (&f))
147         goto fail;
148
149       width = fmt_var_width (&f);
150
151       /* Create each variable. */
152       for (i = 0; i < nv; i++)
153         {
154           struct variable *new_var = dict_create_var (dataset_dict (ds), v[i],
155                                                       width);
156           if (!new_var)
157             msg (SE, _("There is already a variable named %s."), v[i]);
158           else
159             var_set_both_formats (new_var, &f);
160         }
161
162       /* Clean up. */
163       for (i = 0; i < nv; i++)
164         free (v[i]);
165       free (v);
166     }
167   while (lex_match (lexer, T_SLASH));
168
169   return CMD_SUCCESS;
170
171   /* If we have an error at a point where cleanup is required,
172      flow-of-control comes here. */
173 fail:
174   for (i = 0; i < nv; i++)
175     free (v[i]);
176   free (v);
177   return CMD_FAILURE;
178 }
179
180 /* Parses the LEAVE command. */
181 int
182 cmd_leave (struct lexer *lexer, struct dataset *ds)
183 {
184   struct variable **v;
185   size_t nv;
186
187   size_t i;
188
189   if (!parse_variables (lexer, dataset_dict (ds), &v, &nv, PV_NONE))
190     return CMD_CASCADING_FAILURE;
191   for (i = 0; i < nv; i++)
192     var_set_leave (v[i], true);
193   free (v);
194
195   return CMD_SUCCESS;
196 }