1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
/*
* readfile.c - Read an entire file into a string.
*
* Arnold Robbins
* Tue Apr 23 17:43:30 IDT 2002
* Revised per Peter Tillier
* Mon Jun 9 17:05:11 IDT 2003
* Revised for new dynamic function facilities
* Mon Jun 14 14:53:07 IDT 2004
*/
/*
* Copyright (C) 2002, 2003, 2004 the Free Software Foundation, Inc.
*
* This file is part of GAWK, the GNU implementation of the
* AWK Programming Language.
*
* GAWK 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 2 of the License, or
* (at your option) any later version.
*
* GAWK 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, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
*/
#include "awk.h"
#include <fcntl.h>
#ifndef O_BINARY
#define O_BINARY 0
#endif
/* do_readfile --- read a file into memory */
NODE *
do_readfile(tree)
NODE *tree;
{
NODE *filename;
int ret = -1;
struct stat sbuf;
char *text;
int fd;
if (do_lint && get_curfunc_arg_count() > 1)
lintwarn("readfile: called with too many arguments");
filename = get_argument(tree, 0);
if (filename != NULL) {
(void) force_string(filename);
ret = stat(filename->stptr, & sbuf);
if (ret < 0) {
update_ERRNO();
free_temp(filename);
goto done;
} else if ((sbuf.st_mode & S_IFMT) != S_IFREG) {
errno = EINVAL;
ret = -1;
update_ERRNO();
free_temp(filename);
goto done;
}
if ((fd = open(filename->stptr, O_RDONLY|O_BINARY)) < 0) {
ret = -1;
update_ERRNO();
free_temp(filename);
goto done;
}
emalloc(text, char *, sbuf.st_size + 2, "do_readfile");
memset(text, '\0', sbuf.st_size + 2);
if ((ret = read(fd, text, sbuf.st_size)) != sbuf.st_size) {
(void) close(fd);
ret = -1;
update_ERRNO();
free_temp(filename);
goto done;
}
close(fd);
free_temp(filename);
set_value(tmp_string(text, sbuf.st_size));
return tmp_number((AWKNUM) 0);
} else if (do_lint)
lintwarn("filename: called with no arguments");
done:
/* Set the return value */
set_value(tmp_number((AWKNUM) ret));
/* Just to make the interpreter happy */
return tmp_number((AWKNUM) 0);
}
/* dlload --- load new builtins in this library */
NODE *
dlload(tree, dl)
NODE *tree;
void *dl;
{
make_builtin("readfile", do_readfile, 1);
return tmp_number((AWKNUM) 0);
}
|
