1 /***************************************************************************
2 * Copyright (C) 2007-2010 by Øyvind Harboe *
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 2 of the License, or *
7 * (at your option) any later version. *
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. *
14 * You should have received a copy of the GNU General Public License *
15 * along with this program; if not, write to the *
16 * Free Software Foundation, Inc., *
17 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. *
18 ***************************************************************************/
20 /* this file contains various functionality useful to standalone systems */
27 #include "time_support.h"
29 #ifdef HAVE_ARPA_INET_H
30 #include <arpa/inet.h>
41 #ifdef HAVE_SYS_IOCTL_H
42 #include <sys/ioctl.h>
44 #ifdef HAVE_SYS_STAT_H
54 /* loads a file and returns a pointer to it in memory. The file contains
55 * a 0 byte(sentinel) after len bytes - the length of the file. */
56 int loadFile(const char *fileName, void **data, size_t *len)
58 /* ensure returned length is always sane */
62 pFile = fopen(fileName, "rb");
64 LOG_ERROR("Can't open %s", fileName);
67 if (fseek(pFile, 0, SEEK_END) != 0) {
68 LOG_ERROR("Can't open %s", fileName);
72 long fsize = ftell(pFile);
74 LOG_ERROR("Can't open %s", fileName);
80 if (fseek(pFile, 0, SEEK_SET) != 0) {
81 LOG_ERROR("Can't open %s", fileName);
85 *data = malloc(*len + 1);
87 LOG_ERROR("Can't open %s", fileName);
92 if (fread(*data, 1, *len, pFile) != *len) {
95 LOG_ERROR("Can't open %s", fileName);
100 /* 0-byte after buffer (not included in *len) serves as a sentinel */
101 char *buf = (char *)*data;
107 COMMAND_HANDLER(handle_cat_command)
110 return ERROR_COMMAND_SYNTAX_ERROR;
112 /* NOTE!!! we only have line printing capability so we print the entire file as a single
117 int retval = loadFile(CMD_ARGV[0], &data, &len);
118 if (retval == ERROR_OK) {
119 command_print(CMD_CTX, "%s", (char *)data);
122 command_print(CMD_CTX, "%s not found", CMD_ARGV[0]);
127 COMMAND_HANDLER(handle_trunc_command)
130 return ERROR_COMMAND_SYNTAX_ERROR;
132 FILE *config_file = NULL;
133 config_file = fopen(CMD_ARGV[0], "w");
134 if (config_file != NULL)
141 COMMAND_HANDLER(handle_meminfo_command)
144 struct mallinfo info;
147 return ERROR_COMMAND_SYNTAX_ERROR;
152 command_print(CMD_CTX, "Diff: %d", prev - info.fordblks);
153 prev = info.fordblks;
155 command_print(CMD_CTX, "Available ram: %d", info.fordblks);
161 COMMAND_HANDLER(handle_append_command)
164 return ERROR_COMMAND_SYNTAX_ERROR;
166 int retval = ERROR_FAIL;
167 FILE *config_file = NULL;
169 config_file = fopen(CMD_ARGV[0], "a");
170 if (config_file != NULL) {
171 fseek(config_file, 0, SEEK_END);
174 for (i = 1; i < CMD_ARGC; i++) {
175 if (fwrite(CMD_ARGV[i], 1, strlen(CMD_ARGV[i]),
176 config_file) != strlen(CMD_ARGV[i]))
178 if (i != CMD_ARGC - 1) {
179 if (fwrite(" ", 1, 1, config_file) != 1)
183 if ((i == CMD_ARGC) && (fwrite("\n", 1, 1, config_file) == 1))
192 COMMAND_HANDLER(handle_cp_command)
195 return ERROR_COMMAND_SYNTAX_ERROR;
197 /* NOTE!!! we only have line printing capability so we print the entire file as a single
202 int retval = loadFile(CMD_ARGV[0], &data, &len);
203 if (retval != ERROR_OK)
206 FILE *f = fopen(CMD_ARGV[1], "wb");
208 retval = ERROR_COMMAND_SYNTAX_ERROR;
212 size_t chunk = len - pos;
213 static const size_t maxChunk = 512 * 1024; /* ~1/sec */
214 if (chunk > maxChunk)
217 if ((retval == ERROR_OK) && (fwrite(((char *)data) + pos, 1, chunk, f) != chunk))
218 retval = ERROR_COMMAND_SYNTAX_ERROR;
220 if (retval != ERROR_OK)
223 command_print(CMD_CTX, "%zu", len - pos);
231 if (retval == ERROR_OK)
232 command_print(CMD_CTX, "Copied %s to %s", CMD_ARGV[0], CMD_ARGV[1]);
234 command_print(CMD_CTX, "copy failed");
241 if (retval != ERROR_OK)
247 #define SHOW_RESULT(a, b) LOG_ERROR(# a " failed %d\n", (int)b)
250 void copyfile(char *name2, char *name1)
258 fd1 = open(name1, O_WRONLY | O_CREAT, 0664);
260 SHOW_RESULT(open, fd1);
262 fd2 = open(name2, O_RDONLY);
264 SHOW_RESULT(open, fd2);
267 done = read(fd2, buf, IOSIZE);
269 SHOW_RESULT(read, done);
276 wrote = write(fd1, buf, done);
278 SHOW_RESULT(write, wrote);
286 SHOW_RESULT(close, err);
290 SHOW_RESULT(close, err);
293 /* utility fn to copy a directory */
294 void copydir(char *name, char *destdir)
299 dirp = opendir(destdir);
301 mkdir(destdir, 0777);
303 err = closedir(dirp);
305 dirp = opendir(name);
307 SHOW_RESULT(opendir, -1);
310 struct dirent *entry = readdir(dirp);
315 if (strcmp(entry->d_name, ".") == 0)
317 if (strcmp(entry->d_name, "..") == 0)
322 char fullPath[PATH_MAX];
323 strncpy(fullPath, name, PATH_MAX);
324 strcat(fullPath, "/");
325 strncat(fullPath, entry->d_name, PATH_MAX - strlen(fullPath));
327 if (stat(fullPath, &buf) == -1) {
328 LOG_ERROR("unable to read status from %s", fullPath);
331 isDir = S_ISDIR(buf.st_mode) != 0;
336 /* diag_printf("<INFO>: entry %14s",entry->d_name); */
337 char fullname[PATH_MAX];
338 char fullname2[PATH_MAX];
340 strcpy(fullname, name);
341 strcat(fullname, "/");
342 strcat(fullname, entry->d_name);
344 strcpy(fullname2, destdir);
345 strcat(fullname2, "/");
346 strcat(fullname2, entry->d_name);
347 /* diag_printf("from %s to %s\n", fullname, fullname2); */
348 copyfile(fullname, fullname2);
350 /* diag_printf("\n"); */
353 err = closedir(dirp);
355 SHOW_RESULT(stat, err);
358 COMMAND_HANDLER(handle_rm_command)
361 return ERROR_COMMAND_SYNTAX_ERROR;
364 if (rmdir(CMD_ARGV[0]) == 0)
366 else if (unlink(CMD_ARGV[0]) == 0)
369 return del ? ERROR_OK : ERROR_FAIL;
372 static int ioutil_Jim_Command_ls(Jim_Interp *interp,
374 Jim_Obj * const *argv)
377 Jim_WrongNumArgs(interp, 1, argv, "ls ?dir?");
381 char *name = (char *) Jim_GetString(argv[1], NULL);
384 dirp = opendir(name);
387 Jim_Obj *objPtr = Jim_NewListObj(interp, NULL, 0);
390 struct dirent *entry = NULL;
391 entry = readdir(dirp);
395 if ((strcmp(".", entry->d_name) == 0) || (strcmp("..", entry->d_name) == 0))
398 Jim_ListAppendElement(interp, objPtr,
399 Jim_NewStringObj(interp, entry->d_name, strlen(entry->d_name)));
403 Jim_SetResult(interp, objPtr);
408 static int ioutil_Jim_Command_peek(Jim_Interp *interp,
410 Jim_Obj *const *argv)
413 Jim_WrongNumArgs(interp, 1, argv, "peek ?address?");
418 if (Jim_GetLong(interp, argv[1], &address) != JIM_OK)
421 int value = *((volatile int *) address);
423 Jim_SetResult(interp, Jim_NewIntObj(interp, value));
428 static int ioutil_Jim_Command_poke(Jim_Interp *interp,
430 Jim_Obj *const *argv)
433 Jim_WrongNumArgs(interp, 1, argv, "poke ?address? ?value?");
438 if (Jim_GetLong(interp, argv[1], &address) != JIM_OK)
441 if (Jim_GetLong(interp, argv[2], &value) != JIM_OK)
444 *((volatile int *) address) = value;
449 /* not so pretty code to fish out ip number*/
450 static int ioutil_Jim_Command_ip(Jim_Interp *interp, int argc,
451 Jim_Obj *const *argv)
453 #if !defined(__CYGWIN__)
454 Jim_Obj *tclOutput = Jim_NewStringObj(interp, "", 0);
456 struct ifaddrs *ifa = NULL, *ifp = NULL;
458 if (getifaddrs(&ifp) < 0)
461 for (ifa = ifp; ifa; ifa = ifa->ifa_next) {
465 if (ifa->ifa_addr->sa_family == AF_INET)
466 salen = sizeof(struct sockaddr_in);
467 else if (ifa->ifa_addr->sa_family == AF_INET6)
468 salen = sizeof(struct sockaddr_in6);
472 if (getnameinfo(ifa->ifa_addr, salen, ip, sizeof(ip), NULL, 0,
476 Jim_AppendString(interp, tclOutput, ip, strlen(ip));
483 Jim_Obj *tclOutput = Jim_NewStringObj(interp, "fixme!!!", 0);
484 LOG_ERROR("NOT IMPLEMENTED!!!");
486 Jim_SetResult(interp, tclOutput);
491 #ifdef HAVE_SYS_IOCTL_H
493 /* not so pretty code to fish out eth0 mac address */
494 static int ioutil_Jim_Command_mac(Jim_Interp *interp, int argc,
495 Jim_Obj *const *argv)
497 struct ifreq *ifr, *ifend;
503 SockFD = socket(AF_INET, SOCK_DGRAM, 0);
507 ifc.ifc_len = sizeof(ifs);
509 if (ioctl(SockFD, SIOCGIFCONF, &ifc) < 0) {
514 ifend = ifs + (ifc.ifc_len / sizeof(struct ifreq));
515 for (ifr = ifc.ifc_req; ifr < ifend; ifr++) {
516 /* if (ifr->ifr_addr.sa_family == AF_INET) */
518 if (strcmp("eth0", ifr->ifr_name) != 0)
520 strncpy(ifreq.ifr_name, ifr->ifr_name, sizeof(ifreq.ifr_name));
521 if (ioctl(SockFD, SIOCGIFHWADDR, &ifreq) < 0) {
528 Jim_Obj *tclOutput = Jim_NewStringObj(interp, "", 0);
531 sprintf(buffer, "%02x-%02x-%02x-%02x-%02x-%02x",
532 ifreq.ifr_hwaddr.sa_data[0]&0xff,
533 ifreq.ifr_hwaddr.sa_data[1]&0xff,
534 ifreq.ifr_hwaddr.sa_data[2]&0xff,
535 ifreq.ifr_hwaddr.sa_data[3]&0xff,
536 ifreq.ifr_hwaddr.sa_data[4]&0xff,
537 ifreq.ifr_hwaddr.sa_data[5]&0xff);
539 Jim_AppendString(interp, tclOutput, buffer, strlen(buffer));
541 Jim_SetResult(interp, tclOutput);
554 static const struct command_registration ioutil_command_handlers[] = {
557 .handler = handle_cat_command,
559 .help = "display text file content",
560 .usage = "file_name",
564 .handler = handle_trunc_command,
566 .help = "truncate a file to zero length",
567 .usage = "file_name",
571 .handler = handle_cp_command,
573 .help = "copy a file",
574 .usage = "src_file_name dst_file_name",
577 .name = "append_file",
578 .handler = handle_append_command,
580 .help = "append a variable number of strings to a file",
581 .usage = "file_name [<string1>, [<string2>, ...]]",
586 .handler = handle_meminfo_command,
588 .help = "display free heap space",
594 .handler = handle_rm_command,
595 .help = "remove a directory or file",
596 .usage = "file_name",
600 * Peek and poke are security holes -- they manipulate
601 * server-internal addresses.
608 .jim_handler = ioutil_Jim_Command_peek,
609 .help = "peek at a memory address",
615 .jim_handler = ioutil_Jim_Command_poke,
616 .help = "poke at a memory address",
617 .usage = "address value",
622 .jim_handler = ioutil_Jim_Command_ls,
623 .help = "show a listing of files",
626 #ifdef HAVE_SYS_IOCTL_H
631 .jim_handler = ioutil_Jim_Command_mac,
632 .help = "show MAC address",
638 .jim_handler = ioutil_Jim_Command_ip,
640 .help = "show IP address",
642 COMMAND_REGISTRATION_DONE
645 int ioutil_init(struct command_context *cmd_ctx)
647 return register_commands(cmd_ctx, NULL, ioutil_command_handlers);