|
Packit Service |
f629e6 |
/*
|
|
Packit Service |
f629e6 |
* readfile.c - Read an entire file into a string.
|
|
Packit Service |
f629e6 |
*
|
|
Packit Service |
f629e6 |
* Arnold Robbins
|
|
Packit Service |
f629e6 |
* Tue Apr 23 17:43:30 IDT 2002
|
|
Packit Service |
f629e6 |
* Revised per Peter Tillier
|
|
Packit Service |
f629e6 |
* Mon Jun 9 17:05:11 IDT 2003
|
|
Packit Service |
f629e6 |
* Revised for new dynamic function facilities
|
|
Packit Service |
f629e6 |
* Mon Jun 14 14:53:07 IDT 2004
|
|
Packit Service |
f629e6 |
* Revised for formal API May 2012
|
|
Packit Service |
f629e6 |
* Added input parser March 2014
|
|
Packit Service |
f629e6 |
*/
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
/*
|
|
Packit Service |
f629e6 |
* Copyright (C) 2002, 2003, 2004, 2011, 2012, 2013, 2014
|
|
Packit Service |
f629e6 |
* the Free Software Foundation, Inc.
|
|
Packit Service |
f629e6 |
*
|
|
Packit Service |
f629e6 |
* This file is part of GAWK, the GNU implementation of the
|
|
Packit Service |
f629e6 |
* AWK Programming Language.
|
|
Packit Service |
f629e6 |
*
|
|
Packit Service |
f629e6 |
* GAWK is free software; you can redistribute it and/or modify
|
|
Packit Service |
f629e6 |
* it under the terms of the GNU General Public License as published by
|
|
Packit Service |
f629e6 |
* the Free Software Foundation; either version 3 of the License, or
|
|
Packit Service |
f629e6 |
* (at your option) any later version.
|
|
Packit Service |
f629e6 |
*
|
|
Packit Service |
f629e6 |
* GAWK is distributed in the hope that it will be useful,
|
|
Packit Service |
f629e6 |
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
Packit Service |
f629e6 |
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
Packit Service |
f629e6 |
* GNU General Public License for more details.
|
|
Packit Service |
f629e6 |
*
|
|
Packit Service |
f629e6 |
* You should have received a copy of the GNU General Public License
|
|
Packit Service |
f629e6 |
* along with this program; if not, write to the Free Software
|
|
Packit Service |
f629e6 |
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
|
|
Packit Service |
f629e6 |
*/
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
#ifdef HAVE_CONFIG_H
|
|
Packit Service |
f629e6 |
#include <config.h>
|
|
Packit Service |
f629e6 |
#endif
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
#define _BSD_SOURCE
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
#include <stdio.h>
|
|
Packit Service |
f629e6 |
#include <assert.h>
|
|
Packit Service |
f629e6 |
#include <errno.h>
|
|
Packit Service |
f629e6 |
#include <fcntl.h>
|
|
Packit Service |
f629e6 |
#include <stdlib.h>
|
|
Packit Service |
f629e6 |
#include <string.h>
|
|
Packit Service |
f629e6 |
#include <unistd.h>
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
#include <sys/types.h>
|
|
Packit Service |
f629e6 |
#include <sys/stat.h>
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
#include "gawkapi.h"
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
#include "gettext.h"
|
|
Packit Service |
f629e6 |
#define _(msgid) gettext(msgid)
|
|
Packit Service |
f629e6 |
#define N_(msgid) msgid
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
#ifndef O_BINARY
|
|
Packit Service |
f629e6 |
#define O_BINARY 0
|
|
Packit Service |
f629e6 |
#endif
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
static const gawk_api_t *api; /* for convenience macros to work */
|
|
Packit Service |
f629e6 |
static awk_ext_id_t ext_id;
|
|
Packit Service |
f629e6 |
static const char *ext_version = "readfile extension: version 2.0";
|
|
Packit Service |
f629e6 |
static awk_bool_t init_readfile();
|
|
Packit Service |
f629e6 |
static awk_bool_t (*init_func)(void) = init_readfile;
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
int plugin_is_GPL_compatible;
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
/* read_file_to_buffer --- handle the mechanics of reading the file */
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
static char *
|
|
Packit Service |
f629e6 |
read_file_to_buffer(int fd, const struct stat *sbuf)
|
|
Packit Service |
f629e6 |
{
|
|
Packit Service |
f629e6 |
char *text;
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
if ((sbuf->st_mode & S_IFMT) != S_IFREG) {
|
|
Packit Service |
f629e6 |
errno = EINVAL;
|
|
Packit Service |
f629e6 |
update_ERRNO_int(errno);
|
|
Packit Service |
f629e6 |
return NULL;
|
|
Packit Service |
f629e6 |
}
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
emalloc(text, char *, sbuf->st_size + 1, "do_readfile");
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
if (read(fd, text, sbuf->st_size) != sbuf->st_size) {
|
|
Packit Service |
f629e6 |
update_ERRNO_int(errno);
|
|
Packit Service |
f629e6 |
gawk_free(text);
|
|
Packit Service |
f629e6 |
return NULL;
|
|
Packit Service |
f629e6 |
}
|
|
Packit Service |
f629e6 |
text[sbuf->st_size] = '\0';
|
|
Packit Service |
f629e6 |
return text;
|
|
Packit Service |
f629e6 |
}
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
/* do_readfile --- read a file into memory */
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
static awk_value_t *
|
|
Packit Service |
f629e6 |
do_readfile(int nargs, awk_value_t *result, struct awk_ext_func *unused)
|
|
Packit Service |
f629e6 |
{
|
|
Packit Service |
f629e6 |
awk_value_t filename;
|
|
Packit Service |
f629e6 |
int ret;
|
|
Packit Service |
f629e6 |
struct stat sbuf;
|
|
Packit Service |
f629e6 |
char *text;
|
|
Packit Service |
f629e6 |
int fd;
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
assert(result != NULL);
|
|
Packit Service |
f629e6 |
make_null_string(result); /* default return value */
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
unset_ERRNO();
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
if (get_argument(0, AWK_STRING, &filename)) {
|
|
Packit Service |
f629e6 |
ret = stat(filename.str_value.str, & sbuf);
|
|
Packit Service |
f629e6 |
if (ret < 0) {
|
|
Packit Service |
f629e6 |
update_ERRNO_int(errno);
|
|
Packit Service |
f629e6 |
goto done;
|
|
Packit Service |
f629e6 |
}
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
if ((fd = open(filename.str_value.str, O_RDONLY|O_BINARY)) < 0) {
|
|
Packit Service |
f629e6 |
update_ERRNO_int(errno);
|
|
Packit Service |
f629e6 |
goto done;
|
|
Packit Service |
f629e6 |
}
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
text = read_file_to_buffer(fd, & sbuf);
|
|
Packit Service |
f629e6 |
if (text == NULL)
|
|
Packit Service |
f629e6 |
goto done; /* ERRNO already updated */
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
close(fd);
|
|
Packit Service |
f629e6 |
make_malloced_string(text, sbuf.st_size, result);
|
|
Packit Service |
f629e6 |
goto done;
|
|
Packit Service |
f629e6 |
} else if (do_lint)
|
|
Packit Service |
f629e6 |
lintwarn(ext_id, _("readfile: called with wrong kind of argument"));
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
done:
|
|
Packit Service |
f629e6 |
/* Set the return value */
|
|
Packit Service |
f629e6 |
return result;
|
|
Packit Service |
f629e6 |
}
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
/* readfile_get_record --- read the whole file as one record */
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
static int
|
|
Packit Service |
f629e6 |
readfile_get_record(char **out, awk_input_buf_t *iobuf, int *errcode,
|
|
Packit Service |
f629e6 |
char **rt_start, size_t *rt_len,
|
|
Packit Service |
f629e6 |
const awk_fieldwidth_info_t **unused)
|
|
Packit Service |
f629e6 |
{
|
|
Packit Service |
f629e6 |
char *text;
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
/*
|
|
Packit Service |
f629e6 |
* The caller sets *errcode to 0, so we should set it only if an
|
|
Packit Service |
f629e6 |
* error occurs.
|
|
Packit Service |
f629e6 |
*/
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
if (out == NULL || iobuf == NULL)
|
|
Packit Service |
f629e6 |
return EOF;
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
if (iobuf->opaque != NULL) {
|
|
Packit Service |
f629e6 |
/*
|
|
Packit Service |
f629e6 |
* Already read the whole file,
|
|
Packit Service |
f629e6 |
* free up stuff and return EOF
|
|
Packit Service |
f629e6 |
*/
|
|
Packit Service |
f629e6 |
gawk_free(iobuf->opaque);
|
|
Packit Service |
f629e6 |
iobuf->opaque = NULL;
|
|
Packit Service |
f629e6 |
return EOF;
|
|
Packit Service |
f629e6 |
}
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
/* read file */
|
|
Packit Service |
f629e6 |
text = read_file_to_buffer(iobuf->fd, & iobuf->sbuf);
|
|
Packit Service |
f629e6 |
if (text == NULL)
|
|
Packit Service |
f629e6 |
return EOF;
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
/* set up the iobuf for next time */
|
|
Packit Service |
f629e6 |
iobuf->opaque = text;
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
/* set return values */
|
|
Packit Service |
f629e6 |
*rt_start = NULL;
|
|
Packit Service |
f629e6 |
*rt_len = 0;
|
|
Packit Service |
f629e6 |
*out = text;
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
/* return count */
|
|
Packit Service |
f629e6 |
return iobuf->sbuf.st_size;
|
|
Packit Service |
f629e6 |
}
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
/* readfile_can_take_file --- return true if we want the file */
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
static awk_bool_t
|
|
Packit Service |
f629e6 |
readfile_can_take_file(const awk_input_buf_t *iobuf)
|
|
Packit Service |
f629e6 |
{
|
|
Packit Service |
f629e6 |
awk_value_t array, index, value;
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
if (iobuf == NULL)
|
|
Packit Service |
f629e6 |
return awk_false;
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
/*
|
|
Packit Service |
f629e6 |
* This could fail if PROCINFO isn't referenced from
|
|
Packit Service |
f629e6 |
* the awk program. It's not a "can't happen" error.
|
|
Packit Service |
f629e6 |
*/
|
|
Packit Service |
f629e6 |
if (! sym_lookup("PROCINFO", AWK_ARRAY, & array)) {
|
|
Packit Service |
f629e6 |
return awk_false;
|
|
Packit Service |
f629e6 |
}
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
(void) make_const_string("readfile", 8, & index);
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
if (! get_array_element(array.array_cookie, & index, AWK_UNDEFINED, & value)) {
|
|
Packit Service |
f629e6 |
return awk_false;
|
|
Packit Service |
f629e6 |
}
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
return awk_true;
|
|
Packit Service |
f629e6 |
}
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
/* readfile_take_control_of --- take over the file */
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
static awk_bool_t
|
|
Packit Service |
f629e6 |
readfile_take_control_of(awk_input_buf_t *iobuf)
|
|
Packit Service |
f629e6 |
{
|
|
Packit Service |
f629e6 |
if (iobuf == NULL)
|
|
Packit Service |
f629e6 |
return awk_false;
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
iobuf->get_record = readfile_get_record;
|
|
Packit Service |
f629e6 |
return awk_true;
|
|
Packit Service |
f629e6 |
}
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
static awk_input_parser_t readfile_parser = {
|
|
Packit Service |
f629e6 |
"readfile",
|
|
Packit Service |
f629e6 |
readfile_can_take_file,
|
|
Packit Service |
f629e6 |
readfile_take_control_of,
|
|
Packit Service |
f629e6 |
NULL
|
|
Packit Service |
f629e6 |
};
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
/* init_readfile --- set things up */
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
static awk_bool_t
|
|
Packit Service |
f629e6 |
init_readfile()
|
|
Packit Service |
f629e6 |
{
|
|
Packit Service |
f629e6 |
register_input_parser(& readfile_parser);
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
return awk_true;
|
|
Packit Service |
f629e6 |
}
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
static awk_ext_func_t func_table[] = {
|
|
Packit Service |
f629e6 |
{ "readfile", do_readfile, 1, 1, awk_false, NULL },
|
|
Packit Service |
f629e6 |
};
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
/* define the dl_load function using the boilerplate macro */
|
|
Packit Service |
f629e6 |
|
|
Packit Service |
f629e6 |
dl_load_func(func_table, readfile, "")
|