1 /* Implementation of the GETARG and IARGC g77, and
2 corresponding F2003, intrinsics.
3 Copyright (C) 2004-2023 Free Software Foundation, Inc.
4 Contributed by Bud Davis and Janne Blomqvist.
5
6 This file is part of the GNU Fortran 95 runtime library (libgfortran).
7
8 Libgfortran is free software; you can redistribute it and/or
9 modify it under the terms of the GNU General Public
10 License as published by the Free Software Foundation; either
11 version 3 of the License, or (at your option) any later version.
12
13 Libgfortran is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 GNU General Public License for more details.
17
18 Under Section 7 of GPL version 3, you are granted additional
19 permissions described in the GCC Runtime Library Exception, version
20 3.1, as published by the Free Software Foundation.
21
22 You should have received a copy of the GNU General Public License and
23 a copy of the GCC Runtime Library Exception along with this program;
24 see the files COPYING3 and COPYING.RUNTIME respectively. If not, see
25 <http://www.gnu.org/licenses/>. */
26
27 #include "libgfortran.h"
28 #include <string.h>
29
30
31 /* Get a commandline argument. */
32
33 extern void getarg_i4 (GFC_INTEGER_4 *, char *, gfc_charlen_type);
34 iexport_proto(getarg_i4);
35
36 void
37 getarg_i4 (GFC_INTEGER_4 *pos, char *val, gfc_charlen_type val_len)
38 {
39 int argc;
40 char **argv;
41
42 get_args (&argc, &argv);
43
44 if (val_len < 1 || !val )
45 return; /* something is wrong , leave immediately */
46
47 memset (val, ' ', val_len);
48
49 if ((*pos) + 1 <= argc && *pos >=0 )
50 {
51 gfc_charlen_type arglen = strlen (argv[*pos]);
52 if (arglen > val_len)
53 arglen = val_len;
54 memcpy (val, argv[*pos], arglen);
55 }
56 }
57 iexport(getarg_i4);
58
59
60 /* INTEGER*8 wrapper of getarg. */
61
62 extern void getarg_i8 (GFC_INTEGER_8 *, char *, gfc_charlen_type);
63 export_proto (getarg_i8);
64
65 void
66 getarg_i8 (GFC_INTEGER_8 *pos, char *val, gfc_charlen_type val_len)
67 {
68 GFC_INTEGER_4 pos4 = (GFC_INTEGER_4) *pos;
69 getarg_i4 (&pos4, val, val_len);
70 }
71
72
73 /* Return the number of commandline arguments. The g77 info page
74 states that iargc does not include the specification of the
75 program name itself. */
76
77 extern GFC_INTEGER_4 iargc (void);
78 export_proto(iargc);
79
80 GFC_INTEGER_4
81 iargc (void)
82 {
83 int argc;
84 char **argv;
85
86 get_args (&argc, &argv);
87
88 return (argc - 1);
89 }
90
91
92 /* F2003 intrinsic functions and subroutines related to command line
93 arguments.
94
95 - function command_argument_count() is converted to iargc by the compiler.
96
97 - subroutine get_command([command, length, status]).
98
99 - subroutine get_command_argument(number, [value, length, status]).
100 */
101
102 /* These two status codes are specified in the standard. */
103 #define GFC_GC_SUCCESS 0
104 #define GFC_GC_VALUE_TOO_SHORT -1
105
106 /* Processor-specific status failure code. */
107 #define GFC_GC_FAILURE 42
108
109
110 extern void get_command_argument_i4 (GFC_INTEGER_4 *, char *, GFC_INTEGER_4 *,
111 GFC_INTEGER_4 *, gfc_charlen_type);
112 iexport_proto(get_command_argument_i4);
113
114 /* Get a single commandline argument. */
115
116 void
117 get_command_argument_i4 (GFC_INTEGER_4 *number, char *value,
118 GFC_INTEGER_4 *length, GFC_INTEGER_4 *status,
119 gfc_charlen_type value_len)
120 {
121 int argc, stat_flag = GFC_GC_SUCCESS;
122 gfc_charlen_type arglen = 0;
123 char **argv;
124
125 if (number == NULL )
126 /* Should never happen. */
127 runtime_error ("Missing argument to get_command_argument");
128
129 if (value == NULL && length == NULL && status == NULL)
130 return; /* No need to do anything. */
131
132 get_args (&argc, &argv);
133
134 if (*number < 0 || *number >= argc)
135 stat_flag = GFC_GC_FAILURE;
136 else
137 arglen = strlen(argv[*number]);
138
139 if (value != NULL)
140 {
141 if (value_len < 1)
142 stat_flag = GFC_GC_FAILURE;
143 else
144 memset (value, ' ', value_len);
145 }
146
147 if (value != NULL && stat_flag != GFC_GC_FAILURE)
148 {
149 if (arglen > value_len)
150 stat_flag = GFC_GC_VALUE_TOO_SHORT;
151
152 memcpy (value, argv[*number], arglen <= value_len ? arglen : value_len);
153 }
154
155 if (length != NULL)
156 *length = arglen;
157
158 if (status != NULL)
159 *status = stat_flag;
160 }
161 iexport(get_command_argument_i4);
162
163
164 /* INTEGER*8 wrapper for get_command_argument. */
165
166 extern void get_command_argument_i8 (GFC_INTEGER_8 *, char *, GFC_INTEGER_8 *,
167 GFC_INTEGER_8 *, gfc_charlen_type);
168 export_proto(get_command_argument_i8);
169
170 void
171 get_command_argument_i8 (GFC_INTEGER_8 *number, char *value,
172 GFC_INTEGER_8 *length, GFC_INTEGER_8 *status,
173 gfc_charlen_type value_len)
174 {
175 GFC_INTEGER_4 number4;
176 GFC_INTEGER_4 length4;
177 GFC_INTEGER_4 status4;
178
179 number4 = (GFC_INTEGER_4) *number;
180 get_command_argument_i4 (&number4, value, &length4, &status4, value_len);
181 if (length)
182 *length = length4;
183 if (status)
184 *status = status4;
185 }
186
187
188 /* Return the whole commandline. */
189
190 extern void get_command_i4 (char *, GFC_INTEGER_4 *, GFC_INTEGER_4 *,
191 gfc_charlen_type);
192 iexport_proto(get_command_i4);
193
194 void
195 get_command_i4 (char *command, GFC_INTEGER_4 *length, GFC_INTEGER_4 *status,
196 gfc_charlen_type command_len)
197 {
198 int i, argc, thisarg;
199 int stat_flag = GFC_GC_SUCCESS;
200 char **argv;
201 gfc_charlen_type arglen, tot_len = 0;
202
203 if (command == NULL && length == NULL && status == NULL)
204 return; /* No need to do anything. */
205
206 get_args (&argc, &argv);
207
208 if (command != NULL)
209 {
210 /* Initialize the string to blanks. */
211 if (command_len < 1)
212 stat_flag = GFC_GC_FAILURE;
213 else
214 memset (command, ' ', command_len);
215 }
216
217 for (i = 0; i < argc ; i++)
218 {
219 arglen = strlen(argv[i]);
220
221 if (command != NULL && stat_flag == GFC_GC_SUCCESS)
222 {
223 thisarg = arglen;
224 if (tot_len + thisarg > command_len)
225 {
226 thisarg = command_len - tot_len; /* Truncate. */
227 stat_flag = GFC_GC_VALUE_TOO_SHORT;
228 }
229 /* Also a space before the next arg. */
230 else if (i != argc - 1 && tot_len + arglen == command_len)
231 stat_flag = GFC_GC_VALUE_TOO_SHORT;
232
233 memcpy (&command[tot_len], argv[i], thisarg);
234 }
235
236 /* Add the legth of the argument. */
237 tot_len += arglen;
238 if (i != argc - 1)
239 tot_len++;
240 }
241
242 if (length != NULL)
243 *length = tot_len;
244
245 if (status != NULL)
246 *status = stat_flag;
247 }
248 iexport(get_command_i4);
249
250
251 /* INTEGER*8 wrapper for get_command. */
252
253 extern void get_command_i8 (char *, GFC_INTEGER_8 *, GFC_INTEGER_8 *,
254 gfc_charlen_type);
255 export_proto(get_command_i8);
256
257 void
258 get_command_i8 (char *command, GFC_INTEGER_8 *length, GFC_INTEGER_8 *status,
259 gfc_charlen_type command_len)
260 {
261 GFC_INTEGER_4 length4;
262 GFC_INTEGER_4 status4;
263
264 get_command_i4 (command, &length4, &status4, command_len);
265 if (length)
266 *length = length4;
267 if (status)
268 *status = status4;
269 }