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 void compute_filename(struct s_hardware *hardware, char *filename, int size) {
  41
  42    snprintf(filename,size,"%s/",hardware->dump_path);
  43
  44     if (hardware->is_pxe_valid) {
  45             strncat(filename, hardware->pxe.mac_addr, sizeof(hardware->pxe.mac_addr));
  46             strncat(filename, "+", 1);
  47     }
  48
  49     if (hardware->is_dmi_valid) {
  50             strncat(filename, remove_spaces(hardware->dmi.system.product_name), sizeof(hardware->dmi.system.manufacturer));
  51             strncat(filename, "+", 1);
  52             strncat(filename, remove_spaces(hardware->dmi.system.manufacturer), sizeof(hardware->dmi.system.product_name));
  53     }
  54
  55     /* We replace the ":" in the filename by some "-"
  56      * This will avoid Microsoft FS turning crazy */
  57     chrreplace(filename,':','-');
  58
  59     /* Avoid space to make filename easier to manipulate */
  60     chrreplace(filename,' ','_');
  61
  62 }
  63
  64 int dumpprintf(FILE *p, const char *format, ...) {
  65    va_list ap;
  66    int rv;
  67
  68   (void) p;
  69    va_start(ap, format);
  70    rv = vbufprintf(&p_buf,format, ap);
  71    va_end(ap);
  72    return rv;
  73 }
  74
  75 void to_cpio(char *filename) {
  76    cpio_writefile(upload,filename,p_buf.buf,p_buf.len);
  77    if ((p_buf.buf) && (p_buf.len > 0)){
  78       memset(p_buf.buf,0,p_buf.len);
  79       free(p_buf.buf);
  80       p_buf.buf=NULL;
  81       p_buf.size=0;
  82       p_buf.len=0;
  83    }
  84 }
  85
  86 void flush (ZZJSON_CONFIG *config, ZZJSON ** item) {
  87    zzjson_print(config, *item);
  88    zzjson_free(config, *item);
  89    *item=NULL;
  90 }
  91
  92 /**
  93  * dump - dump info
  94  **/
  95 void dump(struct s_hardware *hardware)
  96 {
  97     const union syslinux_derivative_info *sdi = syslinux_derivative_info();
  98     int err=0;
  99     ZZJSON *json = NULL;
 100     ZZJSON_CONFIG config = { ZZJSON_VERY_STRICT, NULL,
 101                 (int(*)(void*)) fgetc,
 102                 NULL,
 103                 malloc, calloc, free, realloc,
 104                 stderr, NULL, stdout,
 105                 (int(*)(void *,const char*,...)) dumpprintf,
 106                 (int(*)(int,void*)) fputc
 107     };
 108
 109     memset(&p_buf,0,sizeof(p_buf));
 110
 111     more_printf("Preparing dump file\n");
 112
 113     /* By now, we only support TFTP reporting */
 114     upload=&upload_tftp;
 115     upload->name="tftp";
 116
 117     /* The following defines the behavior of the reporting */
 118     char *arg[64];
 119     char filename[512]={0};
 120     compute_filename(hardware, filename, sizeof(filename));
 121
 122     /* The filename */
 123     arg[0] = filename;
 124     /* The server to upload the file */
 125     if (strlen(hardware->tftp_ip) != 0) {
 126             arg[1] = hardware->tftp_ip;
 127             arg[2] = NULL;
 128     } else {
 129             arg[1] = NULL;
 130             snprintf(hardware->tftp_ip, sizeof(hardware->tftp_ip),
 131                             "%u.%u.%u.%u",
 132                             ((uint8_t *)&sdi->pxe.ipinfo->serverip)[0],
 133                             ((uint8_t *)&sdi->pxe.ipinfo->serverip)[1],
 134                             ((uint8_t *)&sdi->pxe.ipinfo->serverip)[2],
 135                             ((uint8_t *)&sdi->pxe.ipinfo->serverip)[3]);
 136
 137     }
 138
 139     /* We initiate the cpio to send */
 140     cpio_init(upload,(const char **)arg);
 141
 142     dump_cpu(hardware, &config, &json);
 143     dump_pxe(hardware, &config, &json);
 144     dump_syslinux(hardware, &config, &json);
 145     dump_vpd(hardware, &config, &json);
 146     dump_vesa(hardware, &config, &json);
 147     dump_disks(hardware, &config, &json);
 148     dump_dmi(hardware, &config, &json);
 149     dump_memory(hardware, &config, &json);
 150     dump_pci(hardware, &config, &json);
 151     dump_acpi(hardware, &config, &json);
 152     dump_kernel(hardware, &config, &json);
 153     dump_hdt(hardware, &config, &json);
 154
 155     /* We close & flush the file to send */
 156     cpio_close(upload);
 157
 158     if ((err=flush_data(upload)) != TFTP_OK) {
 159         /* As we manage a tftp connection, let's display the associated error message */
 160         more_printf("Dump failed !\n");
 161         more_printf("TFTP ERROR on  : %s:/%s \n",hardware->tftp_ip, filename);
 162         more_printf("TFTP ERROR msg : %s \n",tftp_string_error_message[-err]);
 163     } else {
 164         more_printf("Dump file sent at %s:/%s\n",hardware->tftp_ip, filename);
 165     }
 166 }