diff options
Diffstat (limited to 'datatop/src/datatop_sys_snap.c')
-rw-r--r-- | datatop/src/datatop_sys_snap.c | 199 |
1 files changed, 199 insertions, 0 deletions
diff --git a/datatop/src/datatop_sys_snap.c b/datatop/src/datatop_sys_snap.c new file mode 100644 index 0000000..ee9ea95 --- /dev/null +++ b/datatop/src/datatop_sys_snap.c @@ -0,0 +1,199 @@ +/************************************************************************ +Copyright (c) 2015, The Linux Foundation. All rights reserved. + +Redistribution and use in source and binary forms, with or without +modification, are permitted provided that the following conditions are +met: + * Redistributions of source code must retain the above copyright + notice, this list of conditions and the following disclaimer. + * Redistributions in binary form must reproduce the above + copyright notice, this list of conditions and the following + disclaimer in the documentation and/or other materials provided + with the distribution. + * Neither the name of The Linux Foundation nor the names of its + contributors may be used to endorse or promote products derived + from this software without specific prior written permission. + +THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESS OR IMPLIED +WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF +MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT +ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS +BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR +CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF +SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR +BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, +WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE +OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN +IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. +************************************************************************/ + +#include <unistd.h> +#include <errno.h> +#include <stdio.h> +#include <string.h> +#include <stdlib.h> +#include <fcntl.h> +#include <sys/types.h> +#include <sys/wait.h> +#include <sys/stat.h> + +#include "datatop_interface.h" +#include "datatop_fileops.h" +#include "datatop_str.h" +#include "datatop_opt.h" + +#define DTOP_SNAP_SIZE 8192 +#define DTOP_SNAP_LINE (DTOP_SNAP_SIZE>>2) + +static int dtop_system_snapshot_helper_print(char *file, const char *str) +{ + FILE *snap_file = fopen(file, "a"); + + if (snap_file) { + if (fprintf(snap_file, "%s", str) < 0) { + fclose(snap_file); + return FILE_ERROR; + } + } else { + return FILE_ERROR; + } + fflush(snap_file); + fclose(snap_file); + return FILE_SUCCESS; +} + +/** + * @brief A helper function to dtop_print_system_snapshot. + * + * @param fw File that desired system data is printed to. + * @return FILE_ERROR - Writing to file was unsuccessful. + * @return FILE_SUCCESS - Writing to file was successful. + */ +static int dtop_run_and_log(char *file, const char *c1, const char **args) +{ + int i; + pid_t child_pid; + + i = 0; + dtop_system_snapshot_helper_print(file, "\n" + "--------------------------------------------------------------\n" + "Command: "); + while(args[i] != 0) { + dtop_system_snapshot_helper_print(file, args[i++]); + dtop_system_snapshot_helper_print(file, " "); + } + dtop_system_snapshot_helper_print(file, "\n"); + + + child_pid = fork(); + if (child_pid == 0) { + int fd = open(file, O_WRONLY | O_APPEND | O_CREAT, + S_IRUSR | S_IWUSR); + dup2(fd, STDOUT_FILENO); + dup2(fd, STDERR_FILENO); + close(fd); + execvp(c1, (char * const *)args); + printf("Failed to execute %s\n", c1); + printf("errno=%d error=%s\n", errno, strerror(errno)); + close(STDOUT_FILENO); + close(STDERR_FILENO); + exit(0); + } else if (child_pid < 0) { + return FILE_ERROR; + } else { + int return_status; + waitpid(child_pid, &return_status, 0); + + if (return_status != 0) + return FILE_ERROR; + } + + return FILE_SUCCESS; +} + + + +/* IPv4 */ +const char *ip_addr_cmd[] = {"ip", "addr", 0}; +const char *ip_route_cmd[] = {"ip", "route", 0}; +const char *ip_route_all_tables_cmd[] = {"ip", "route", "show", "table", "all", 0}; +const char *ip_rule_cmd[] = {"ip", "rule", "show", 0}; +const char *ip_tables_cmd[] = {"iptables", "-L", "-n", "-v", 0}; +const char *ip_tables_nat_cmd[] = {"iptables", "-t", "nat", "-L", "-n", "-v", 0}; +const char *ip_tables_mangle_cmd[] = {"iptables", "-t", "mangle", "-L", "-n", "-v", 0}; +const char *ip_tables_raw_cmd[] = {"iptables", "-t", "raw", "-L", "-n", "-v", 0}; + +/* IPv6 */ +const char *ip6_addr_cmd[] = {"ip", "-6", "addr", 0}; +const char *ip6_route_cmd[] = {"ip", "-6", "route", 0}; +const char *ip6_route_all_tables_cmd[] = {"ip", "-6", "route", "show", "table", "all", 0}; +const char *ip6_rule_cmd[] = {"ip", "-6", "rule", "show", 0}; +const char *ip6_tables_cmd[] = {"ip6tables", "-L", "-n", "-v", 0}; +const char *ip6_tables_nat_cmd[] = {"ip6tables", "-t", "nat", "-L", "-n", "-v", 0}; +const char *ip6_tables_mangle_cmd[] = {"ip6tables", "-t", "mangle", "-L", "-n", "-v", 0}; +const char *ip6_tables_raw_cmd[] = {"ip6tables", "-t", "raw", "-L", "-n", "-v", 0}; + +/* Misc */ +const char *rps_config[] = {"cat", "/sys/class/net/rmnet_mhi0/queues/rx-0/rps_cpus", 0}; +const char *if_config[] = {"/data/busybox/busybox", "ifconfig", 0}; +const char *netcfg[] = {"netcfg", 0}; +const char *softnet_stat[] = {"cat", "/proc/net/softnet_stat", 0}; + +/* XFRM logging */ +const char *xfrm_state[] = {"ip", "xfrm", "state", "show", 0}; +const char *xfrm_policy[] = {"ip", "xfrm", "policy", "show", 0}; +const char *xfrm_netstat[] = {"cat", "/proc/net/xfrm_stat", 0}; + +#define DO_DTOP_RUN_AND_LOG(X) \ + dtop_run_and_log(file, X[0], X); +/** + * @brief Prints a System snapshot to a file specified by the user. + * + * @param fw File that system snapshot is printed to. + * @return FILE_ERROR - Writing to file was unsuccessful. + * @return FILE_SUCCESS - Writing to file was successful. + */ +int dtop_print_system_snapshot(char *file) +{ + dtop_system_snapshot_helper_print(file, + "==============================================================\n" + " System Data Snapshot - Captured with Data Top \n" + " Version "); + dtop_system_snapshot_helper_print(file, VERSION); + dtop_system_snapshot_helper_print(file, "\n" + "==============================================================\n" + "\n"); + + /* IPv4 */ + DO_DTOP_RUN_AND_LOG(ip_addr_cmd); + DO_DTOP_RUN_AND_LOG(ip_route_cmd); + DO_DTOP_RUN_AND_LOG(ip_route_all_tables_cmd); + DO_DTOP_RUN_AND_LOG(ip_rule_cmd); + DO_DTOP_RUN_AND_LOG(ip_tables_cmd); + DO_DTOP_RUN_AND_LOG(ip_tables_nat_cmd); + DO_DTOP_RUN_AND_LOG(ip_tables_mangle_cmd); + DO_DTOP_RUN_AND_LOG(ip_tables_raw_cmd); + + /* IPv6 */ + DO_DTOP_RUN_AND_LOG(ip6_addr_cmd); + DO_DTOP_RUN_AND_LOG(ip6_route_cmd); + DO_DTOP_RUN_AND_LOG(ip6_route_all_tables_cmd); + DO_DTOP_RUN_AND_LOG(ip6_rule_cmd); + DO_DTOP_RUN_AND_LOG(ip6_tables_cmd); + DO_DTOP_RUN_AND_LOG(ip6_tables_nat_cmd); + DO_DTOP_RUN_AND_LOG(ip6_tables_mangle_cmd); + DO_DTOP_RUN_AND_LOG(ip6_tables_raw_cmd); + + /* Misc */ + DO_DTOP_RUN_AND_LOG(rps_config); + DO_DTOP_RUN_AND_LOG(if_config); + DO_DTOP_RUN_AND_LOG(netcfg); + DO_DTOP_RUN_AND_LOG(softnet_stat); + + /* XFRM logging */ + DO_DTOP_RUN_AND_LOG(xfrm_state); + DO_DTOP_RUN_AND_LOG(xfrm_policy); + DO_DTOP_RUN_AND_LOG(xfrm_netstat); + + return FILE_SUCCESS; +} |