+++ /dev/null
-/* PSPP - a program for statistical analysis.
- Copyright (C) 1997-2000, 2006, 2010-2013, 2016 Free Software Foundation, Inc.
-
- This program is free software: you can redistribute it and/or modify
- it under the terms of the GNU General Public License as published by
- the Free Software Foundation, either version 3 of the License, or
- (at your option) any later version.
-
- This program is distributed in the hope that it will be useful,
- but WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- GNU General Public License for more details.
-
- You should have received a copy of the GNU General Public License
- along with this program. If not, see <http://www.gnu.org/licenses/>. */
-
-#include <config.h>
-
-#include "data/file-handle-def.h"
-
-#include <limits.h>
-#include <errno.h>
-#include <stdlib.h>
-
-#include "data/file-name.h"
-#include "data/session.h"
-#include "data/variable.h"
-#include "language/command.h"
-#include "language/data-io/file-handle.h"
-#include "language/lexer/lexer.h"
-#include "libpspp/assertion.h"
-#include "libpspp/cast.h"
-#include "libpspp/message.h"
-#include "libpspp/str.h"
-
-#include "gl/xalloc.h"
-
-#include "gettext.h"
-#define _(msgid) gettext (msgid)
-
-int
-cmd_file_handle (struct lexer *lexer, struct dataset *ds UNUSED)
-{
- enum cmd_result result = CMD_CASCADING_FAILURE;
- char *handle_name = NULL;
- char *file_name = NULL;
- int lrecl = 0;
- int tabwidth = -1;
- enum { MODE_DEFAULT, MODE_CHARACTER, MODE_BINARY, MODE_IMAGE, MODE_360 }
- mode = MODE_DEFAULT;
- int ends = -1;
- enum { RECFORM_FIXED = 1, RECFORM_VARIABLE, RECFORM_SPANNED } recform = 0;
- char *encoding = NULL;
-
- if (!lex_force_id (lexer))
- goto exit;
-
- handle_name = xstrdup (lex_tokcstr (lexer));
- struct file_handle *fh = fh_from_id (handle_name);
- if (fh)
- {
- fh_unref (fh);
- lex_error (lexer, _("File handle %s is already defined. "
- "Use %s before redefining a file handle."),
- handle_name, "CLOSE FILE HANDLE");
- goto exit;
- }
-
- lex_get (lexer);
- if (!lex_force_match (lexer, T_SLASH))
- goto exit;
-
- int mode_start = 0;
- int mode_end = 0;
- while (lex_token (lexer) != T_ENDCMD)
- {
- if (lex_match_id (lexer, "NAME"))
- {
- if (file_name)
- {
- lex_sbc_only_once (lexer, "NAME");
- goto exit;
- }
-
- lex_match (lexer, T_EQUALS);
- if (!lex_force_string (lexer))
- goto exit;
- free (file_name);
- file_name = ss_xstrdup (lex_tokss (lexer));
- lex_get (lexer);
- }
- else if (lex_match_id (lexer, "LRECL"))
- {
- if (lrecl)
- {
- lex_sbc_only_once (lexer, "LRECL");
- goto exit;
- }
-
- lex_match (lexer, T_EQUALS);
- if (!lex_force_int_range (lexer, "LRECL", 1, (1UL << 31) - 1))
- goto exit;
- lrecl = lex_integer (lexer);
- lex_get (lexer);
- }
- else if (lex_match_id (lexer, "TABWIDTH"))
- {
- if (tabwidth >= 0)
- {
- lex_sbc_only_once (lexer, "TABWIDTH");
- goto exit;
- }
- lex_match (lexer, T_EQUALS);
-
- if (!lex_force_int_range (lexer, "TABWIDTH", 1, INT_MAX))
- goto exit;
- tabwidth = lex_integer (lexer);
- lex_get (lexer);
- }
- else if (lex_match_id (lexer, "MODE"))
- {
- if (mode != MODE_DEFAULT)
- {
- lex_sbc_only_once (lexer, "MODE");
- goto exit;
- }
- mode_start = lex_ofs (lexer) - 1;
- lex_match (lexer, T_EQUALS);
-
- if (lex_match_id (lexer, "CHARACTER"))
- mode = MODE_CHARACTER;
- else if (lex_match_id (lexer, "BINARY"))
- mode = MODE_BINARY;
- else if (lex_match_id (lexer, "IMAGE"))
- mode = MODE_IMAGE;
- else if (lex_match_int (lexer, 360))
- mode = MODE_360;
- else
- {
- lex_error_expecting (lexer, "CHARACTER", "BINARY",
- "IMAGE", "360");
- goto exit;
- }
- mode_end = lex_ofs (lexer) - 1;
- }
- else if (lex_match_id (lexer, "ENDS"))
- {
- if (ends >= 0)
- {
- lex_sbc_only_once (lexer, "ENDS");
- goto exit;
- }
- lex_match (lexer, T_EQUALS);
-
- if (lex_match_id (lexer, "LF"))
- ends = FH_END_LF;
- else if (lex_match_id (lexer, "CRLF"))
- ends = FH_END_CRLF;
- else
- {
- lex_error_expecting (lexer, "LF", "CRLF");
- goto exit;
- }
- }
- else if (lex_match_id (lexer, "RECFORM"))
- {
- if (recform)
- {
- lex_sbc_only_once (lexer, "RECFORM");
- goto exit;
- }
- lex_match (lexer, T_EQUALS);
- if (lex_match_id (lexer, "FIXED") || lex_match_id (lexer, "F"))
- recform = RECFORM_FIXED;
- else if (lex_match_id (lexer, "VARIABLE")
- || lex_match_id (lexer, "V"))
- recform = RECFORM_VARIABLE;
- else if (lex_match_id (lexer, "SPANNED")
- || lex_match_id (lexer, "VS"))
- recform = RECFORM_SPANNED;
- else
- {
- lex_error_expecting (lexer, "FIXED", "VARIABLE", "SPANNED");
- goto exit;
- }
- }
- else if (lex_match_id (lexer, "ENCODING"))
- {
- if (encoding)
- {
- lex_sbc_only_once (lexer, "ENCODING");
- goto exit;
- }
-
- lex_match (lexer, T_EQUALS);
- if (!lex_force_string (lexer))
- goto exit;
- free (encoding);
- encoding = ss_xstrdup (lex_tokss (lexer));
- lex_get (lexer);
- }
- if (!lex_match (lexer, T_SLASH))
- break;
- }
-
- if (lex_end_of_command (lexer) != CMD_SUCCESS)
- goto exit;
-
- struct fh_properties properties = *fh_default_properties ();
- if (file_name == NULL)
- {
- lex_sbc_missing (lexer, "NAME");
- goto exit;
- }
-
- switch (mode)
- {
- case MODE_DEFAULT:
- case MODE_CHARACTER:
- properties.mode = FH_MODE_TEXT;
- if (tabwidth >= 0)
- properties.tab_width = tabwidth;
- if (ends)
- properties.line_ends = ends;
- break;
- case MODE_IMAGE:
- properties.mode = FH_MODE_FIXED;
- break;
- case MODE_BINARY:
- properties.mode = FH_MODE_VARIABLE;
- break;
- case MODE_360:
- properties.encoding = CONST_CAST (char *, "EBCDIC-US");
- if (recform == RECFORM_FIXED)
- properties.mode = FH_MODE_FIXED;
- else if (recform == RECFORM_VARIABLE)
- {
- properties.mode = FH_MODE_360_VARIABLE;
- properties.record_width = 8192;
- }
- else if (recform == RECFORM_SPANNED)
- {
- properties.mode = FH_MODE_360_SPANNED;
- properties.record_width = 8192;
- }
- else
- {
- lex_ofs_error (lexer, mode_start, mode_end,
- _("%s must be specified with %s."),
- "RECFORM", "MODE=360");
- goto exit;
- }
- break;
- default:
- NOT_REACHED ();
- }
-
- if (properties.mode == FH_MODE_FIXED || lrecl)
- {
- if (!lrecl)
- msg (SE, _("The specified file mode requires LRECL. "
- "Assuming %zu-character records."),
- properties.record_width);
- else
- properties.record_width = lrecl;
- }
-
- if (encoding)
- properties.encoding = encoding;
-
- fh_create_file (handle_name, file_name, lex_get_encoding (lexer),
- &properties);
-
- result = CMD_SUCCESS;
-
-exit:
- free (handle_name);
- free (file_name);
- free (encoding);
- return result;
-}
-
-int
-cmd_close_file_handle (struct lexer *lexer, struct dataset *ds UNUSED)
-{
- struct file_handle *handle;
-
- if (!lex_force_id (lexer))
- return CMD_CASCADING_FAILURE;
- handle = fh_from_id (lex_tokcstr (lexer));
- if (handle == NULL)
- return CMD_CASCADING_FAILURE;
-
- fh_unname (handle);
- return CMD_SUCCESS;
-}
-
-/* Returns the name for REFERENT. */
-static const char *
-referent_name (enum fh_referent referent)
-{
- switch (referent)
- {
- case FH_REF_FILE:
- return _("file");
- case FH_REF_INLINE:
- return _("inline file");
- case FH_REF_DATASET:
- return _("dataset");
- default:
- NOT_REACHED ();
- }
-}
-
-/* Parses a file handle name:
-
- - If SESSION is nonnull, then the parsed syntax may be the name of a
- dataset within SESSION. Dataset names take precedence over file handle
- names.
-
- - If REFERENT_MASK includes FH_REF_FILE, the parsed syntax may be a file
- name as a string or a file handle name as an identifier.
-
- - If REFERENT_MASK includes FH_REF_INLINE, the parsed syntax may be the
- identifier INLINE to represent inline data.
-
- Returns the file handle when successful, a null pointer on failure.
-
- The caller is responsible for fh_unref()'ing the returned file handle when
- it is no longer needed. */
-struct file_handle *
-fh_parse (struct lexer *lexer, enum fh_referent referent_mask,
- struct session *session)
-{
- if (session != NULL && lex_token (lexer) == T_ID)
- {
- struct dataset *ds;
-
- ds = session_lookup_dataset (session, lex_tokcstr (lexer));
- if (ds != NULL)
- {
- lex_get (lexer);
- return fh_create_dataset (ds);
- }
- }
-
- int start_ofs = lex_ofs (lexer);
- struct file_handle *handle;
- if (lex_match_id (lexer, "INLINE"))
- handle = fh_inline_file ();
- else
- {
- if (lex_token (lexer) != T_ID && !lex_is_string (lexer))
- {
- lex_error (lexer,
- _("Syntax error expecting a file name or handle name."));
- return NULL;
- }
-
- handle = NULL;
- if (lex_token (lexer) == T_ID)
- handle = fh_from_id (lex_tokcstr (lexer));
- if (handle == NULL)
- handle = fh_create_file (NULL, lex_tokcstr (lexer), lex_get_encoding (lexer),
- fh_default_properties ());
- lex_get (lexer);
- }
-
- if (!(fh_get_referent (handle) & referent_mask))
- {
- lex_ofs_error (lexer, start_ofs, lex_ofs (lexer) - 1,
- _("Handle for %s not allowed here."),
- referent_name (fh_get_referent (handle)));
- fh_unref (handle);
- return NULL;
- }
-
- return handle;
-}