com32/hdt/hdt-dump.c

   1 /* ----------------------------------------------------------------------- *
   2  *
   3  *   Copyright 2011 Erwan Velu - All Rights Reserved
   4  *
   5  *   Permission is hereby granted, free of charge, to any person
   6  *   obtaining a copy of this software and associated documentation
   7  *   files (the "Software"), to deal in the Software without
   8  *   restriction, including without limitation the rights to use,
   9  *   copy, modify, merge, publish, distribute, sublicense, and/or
  10  *   sell copies of the Software, and to permit persons to whom
  11  *   the Software is furnished to do so, subject to the following
  12  *   conditions:
  13  *
  14  *   The above copyright notice and this permission notice shall
  15  *   be included in all copies or substantial portions of the Software.
  16  *
  17  *   THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  18  *   EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
  19  *   OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  20  *   NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
  21  *   HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
  22  *   WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  23  *   FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
  24  *   OTHER DEALINGS IN THE SOFTWARE.
  25  *
  26  * -----------------------------------------------------------------------
  27  */
  28
  29 #include <stdio.h>
  30 #include <string.h>
  31 #include <stdlib.h>
  32 #include <ctype.h>
  33 #include <getkey.h>
  34 #include <syslinux/config.h>
  35 #include "hdt-common.h"
  36 #include "hdt-dump.h"
  37
  38 struct print_buf p_buf;
  39
  40 struct dump_option {
  41     char *flag;
  42     char *item;
  43 };
  44
  45 char *get_value_from_option(struct s_hardware *hardware, char *option)
  46 {
  47     struct dump_option dump_options[10];
  48     dump_options[0].flag = "%{m}";
  49     dump_options[0].item = hardware->pxe.mac_addr;
  50
  51     dump_options[1].flag = "%{v}";
  52     dump_options[1].item = hardware->dmi.system.manufacturer;
  53
  54     dump_options[2].flag = "%{p}";
  55     dump_options[2].item = hardware->dmi.system.product_name;
  56
  57     dump_options[3].flag = "%{ba}";
  58     dump_options[3].item = hardware->dmi.base_board.asset_tag;
  59
  60     dump_options[4].flag = "%{bs}";
  61     dump_options[4].item = hardware->dmi.base_board.serial;
  62
  63     dump_options[5].flag = "%{ca}";
  64     dump_options[5].item = hardware->dmi.chassis.asset_tag;
  65
  66     dump_options[6].flag = "%{cs}";
  67     dump_options[6].item = hardware->dmi.chassis.serial;
  68
  69     dump_options[7].flag = "%{sk}";
  70     dump_options[7].item = hardware->dmi.system.sku_number;
  71
  72     dump_options[8].flag = "%{ss}";
  73     dump_options[8].item = hardware->dmi.system.serial;
  74
  75     dump_options[9].flag = NULL;
  76     dump_options[9].item = NULL;
  77
  78     for (int i = 0; i < 9; i++) {
  79         if (strcmp(option, dump_options[i].flag) == 0) {
  80             return remove_spaces(dump_options[i].item);
  81         }
  82     }
  83
  84     return NULL;
  85 }
  86
  87 char *compute_filename(struct s_hardware *hardware)
  88 {
  89
  90     char *filename = malloc(512);
  91     snprintf(filename, 512, "%s/%s", hardware->dump_path,
  92              hardware->dump_filename);
  93
  94     /* Until we found some dump parameters */
  95     char *buffer;
  96     while ((buffer = strstr(filename, "%{"))) {
  97         // Find the end of the parameter
  98         char *buffer_end = strstr(buffer, "}");
  99
 100         // Extracting the parameter between %{ and }
 101         char option[8] = { 0 };
 102         strncpy(option, buffer, buffer_end - buffer + 1);
 103
 104         /* Replace this option by its value in the filename
 105          * Filename is longer than the previous filename we had
 106          * so let's restart from the beginning */
 107         filename =
 108             strreplace(filename, option,
 109                        get_value_from_option(hardware, option));
 110     }
 111
 112     /* We replace the ":" in the filename by some "-"
 113      * This will avoid Microsoft FS turning crazy */
 114     chrreplace(filename, ':', '-');
 115
 116     /* Avoid space to make filename easier to manipulate */
 117     chrreplace(filename, ' ', '_');
 118
 119     return filename;
 120 }
 121
 122 int dumpprintf(FILE * p, const char *format, ...)
 123 {
 124     va_list ap;
 125     int rv;
 126
 127     (void)p;
 128     va_start(ap, format);
 129     rv = vbufprintf(&p_buf, format, ap);
 130     va_end(ap);
 131     return rv;
 132 }
 133
 134 void to_cpio(char *filename)
 135 {
 136     cpio_writefile(upload, filename, p_buf.buf, p_buf.len);
 137     if ((p_buf.buf) && (p_buf.len > 0)) {
 138         memset(p_buf.buf, 0, p_buf.len);
 139         free(p_buf.buf);
 140         p_buf.buf = NULL;
 141         p_buf.size = 0;
 142         p_buf.len = 0;
 143     }
 144 }
 145
 146 void flush(ZZJSON_CONFIG * config, ZZJSON ** item)
 147 {
 148     zzjson_print(config, *item);
 149     zzjson_free(config, *item);
 150     *item = NULL;
 151 }
 152
 153 /**
 154  * dump - dump info
 155  **/
 156 void dump(struct s_hardware *hardware)
 157 {
 158     if (hardware->is_pxe_valid == false) {
 159         more_printf("PXE stack was not detected, Dump feature is not available\n");
 160         return;
 161     }
 162
 163     const union syslinux_derivative_info *sdi = syslinux_derivative_info();
 164     int err = 0;
 165     ZZJSON *json = NULL;
 166     ZZJSON_CONFIG config = { ZZJSON_VERY_STRICT, NULL,
 167         (int (*)(void *))fgetc,
 168         NULL,
 169         malloc, calloc, free, realloc,
 170         stderr, NULL, stdout,
 171         (int (*)(void *, const char *,...))dumpprintf,
 172         (int (*)(int, void *))fputc
 173     };
 174
 175     memset(&p_buf, 0, sizeof(p_buf));
 176
 177     /* By now, we only support TFTP reporting */
 178     upload = &upload_tftp;
 179     upload->name = "tftp";
 180
 181     /* The following defines the behavior of the reporting */
 182     char *arg[64];
 183     char *filename = compute_filename(hardware);
 184
 185     /* The filename */
 186     arg[0] = filename;
 187     /* The server to upload the file */
 188     if (strlen(hardware->tftp_ip) != 0) {
 189         arg[1] = hardware->tftp_ip;
 190         arg[2] = NULL;
 191     } else {
 192         arg[1] = NULL;
 193         snprintf(hardware->tftp_ip, sizeof(hardware->tftp_ip),
 194                  "%u.%u.%u.%u",
 195                  ((uint8_t *) & sdi->pxe.ipinfo->serverip)[0],
 196                  ((uint8_t *) & sdi->pxe.ipinfo->serverip)[1],
 197                  ((uint8_t *) & sdi->pxe.ipinfo->serverip)[2],
 198                  ((uint8_t *) & sdi->pxe.ipinfo->serverip)[3]);
 199
 200     }
 201
 202     /* We initiate the cpio to send */
 203     cpio_init(upload, (const char **)arg);
 204
 205     dump_cpu(hardware, &config, &json);
 206     dump_pxe(hardware, &config, &json);
 207     dump_syslinux(hardware, &config, &json);
 208     dump_vpd(hardware, &config, &json);
 209     dump_vesa(hardware, &config, &json);
 210     dump_disks(hardware, &config, &json);
 211     dump_dmi(hardware, &config, &json);
 212     dump_memory(hardware, &config, &json);
 213     dump_pci(hardware, &config, &json);
 214     dump_acpi(hardware, &config, &json);
 215     dump_kernel(hardware, &config, &json);
 216     dump_hdt(hardware, &config, &json);
 217
 218     /* We close & flush the file to send */
 219     cpio_close(upload);
 220
 221     if ((err = flush_data(upload)) != TFTP_OK) {
 222         /* As we manage a tftp connection, let's display the associated error message */
 223         more_printf("Dump failed !\n");
 224         more_printf("TFTP ERROR on  : %s:/%s \n", hardware->tftp_ip, filename);
 225         more_printf("TFTP ERROR msg : %s \n", tftp_string_error_message[-err]);
 226     } else {
 227         more_printf("Dump file sent at %s:/%s\n", hardware->tftp_ip, filename);
 228     }
 229 }