com32/hdt/hdt-dump.c

   1 /* ----------------------------------------------------------------------- *
   2  *
   3  *   Copyright 20011 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 <bufprintf.h>
  34 #include <zzjson/zzjson.h>
  35 #include "hdt-common.h"
  36
  37 #define add_i(name,value) *item = zzjson_object_append(config, *item, name, zzjson_create_number_i(config, value))
  38 #define add_s(name,value) *item = zzjson_object_append(config, *item, name, zzjson_create_string(config, value))
  39 #define add_bool_true(name) *item = zzjson_object_append(config, *item, name, zzjson_create_true(config))
  40 #define add_bool_false(name) *item = zzjson_object_append(config, *item, name, zzjson_create_false(config))
  41 #define add_hi(value) add_i(#value,hardware->value)
  42 #define add_hs(value) add_s(#value,hardware->value)
  43 #define add_b(name,value) if (value==true) {add_bool_true((char *)name);} else {add_bool_false((char *)name);}
  44
  45 static struct print_buf p_buf;
  46
  47 static void compute_filename(struct s_hardware *hardware, char *filename, int size) {
  48
  49    snprintf(filename,size,"%s/","hdt");
  50
  51     if (hardware->is_pxe_valid) {
  52             strncat(filename, hardware->pxe.mac_addr, sizeof(hardware->pxe.mac_addr));
  53             strncat(filename, "+", 1);
  54     }
  55
  56     if (hardware->is_dmi_valid) {
  57             strncat(filename, remove_spaces(hardware->dmi.system.product_name), sizeof(hardware->dmi.system.manufacturer));
  58             strncat(filename, "+", 1);
  59             strncat(filename, remove_spaces(hardware->dmi.system.manufacturer), sizeof(hardware->dmi.system.product_name));
  60     }
  61
  62     /* We replace the ":" in the filename by some "-"
  63      * This will avoid Microsoft FS turning crazy */
  64     chrreplace(filename,':','-');
  65
  66     /* Avoid space to make filename easier to manipulate */
  67     chrreplace(filename,' ','_');
  68
  69 }
  70
  71 void print_and_flush(ZZJSON_CONFIG *config, ZZJSON **item) {
  72         zzjson_print(config, *item);
  73         zzjson_free(config, *item);
  74 }
  75
  76 void dump_cpu(struct s_hardware *hardware, ZZJSON_CONFIG *config, ZZJSON **item) {
  77
  78         *item = zzjson_create_object(config, NULL); /* empty object */
  79         add_hs(cpu.vendor);
  80         add_hs(cpu.model);
  81         add_hi(cpu.vendor_id);
  82         add_hi(cpu.family);
  83         add_hi(cpu.model_id);
  84         add_hi(cpu.stepping);
  85         add_hi(cpu.num_cores);
  86         add_hi(cpu.l1_data_cache_size);
  87         add_hi(cpu.l1_instruction_cache_size);
  88         add_hi(cpu.l2_cache_size);
  89         size_t i;
  90         for (i = 0; i < cpu_flags_count; i++) {
  91                 char temp[128]={0};
  92                 snprintf(temp,sizeof(temp),"cpu.flags.%s",cpu_flags_names[i]);
  93                 add_b(temp,get_cpu_flag_value_from_name(&hardware->cpu,cpu_flags_names[i]));
  94         }
  95         print_and_flush(config,item);
  96 }
  97
  98 int dumpprintf(FILE *p, const char *format, ...) {
  99    va_list ap;
 100    int rv;
 101
 102   (void) p;
 103    va_start(ap, format);
 104    rv = vbufprintf(&p_buf,format, ap);
 105    va_end(ap);
 106    return rv;
 107 }
 108
 109 /**
 110  * dump - dump info
 111  **/
 112 void dump(struct s_hardware *hardware)
 113 {
 114     ZZJSON *json = NULL;
 115     ZZJSON_CONFIG config = { ZZJSON_VERY_STRICT, NULL,
 116                 (int(*)(void*)) fgetc,
 117                 NULL,
 118                 malloc, calloc, free, realloc,
 119                 stderr, NULL, stdout,
 120                 (int(*)(void *,const char*,...)) dumpprintf,
 121                 (int(*)(int,void*)) fputc
 122     };
 123
 124     detect_hardware(hardware);
 125     dump_cpu(hardware, &config, &json);
 126
 127
 128     /* By now, we only support TFTP reporting */
 129     upload=&upload_tftp;
 130     upload->name="tftp";
 131
 132     /* The following defines the behavior of the reporting */
 133     char *arg[64];
 134     char filename[512]={0};
 135     compute_filename(hardware, filename, sizeof(filename));
 136
 137     /* The filename */
 138     arg[0] = filename;
 139     /* More to come */
 140     arg[1] = NULL;
 141
 142     /* We initiate the cpio to send */
 143     cpio_init(upload,(const char **)arg);
 144
 145     cpio_writefile(upload,"cpu",p_buf.buf,p_buf.len);
 146
 147     /* We close & flush the file to send */
 148     cpio_close(upload);
 149     flush_data(upload);
 150     if (p_buf.buf) {
 151             free(p_buf.buf);
 152     }
 153 }