usr/src/cmd/fm/eversholt/common/eftwrite.c

   1 /*
   2  * CDDL HEADER START
   3  *
   4  * The contents of this file are subject to the terms of the
   5  * Common Development and Distribution License (the "License").
   6  * You may not use this file except in compliance with the License.
   7  *
   8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
   9  * or http://www.opensolaris.org/os/licensing.
  10  * See the License for the specific language governing permissions
  11  * and limitations under the License.
  12  *
  13  * When distributing Covered Code, include this CDDL HEADER in each
  14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
  15  * If applicable, add the following below this CDDL HEADER, with the
  16  * fields enclosed by brackets "[]" replaced with your own identifying
  17  * information: Portions Copyright [yyyy] [name of copyright owner]
  18  *
  19  * CDDL HEADER END
  20  */
  21 /*
  22  * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
  23  * Use is subject to license terms.
  24  *
  25  * eftwrite.c -- routines for writing .eft files
  26  *
  27  * this module emits the table resulting from compilation of the
  28  * source files.  this code done nothing unless the -o option
  29  * was given on the command line.
  30  */
  31
  32 #include <stdio.h>
  33 #include <string.h>
  34 #include <strings.h>
  35 #include <unistd.h>
  36 #include <errno.h>
  37 #include "out.h"
  38 #include "stats.h"
  39 #include "stable.h"
  40 #include "lut.h"
  41 #include "tree.h"
  42 #include "eft.h"
  43 #include "eftwrite.h"
  44 #include "esclex.h"
  45 #include "version.h"
  46 #include "ptree.h"
  47
  48 /* for uintX_t, htonl(), etc */
  49 #include <sys/types.h>
  50 #include <netinet/in.h>
  51 #include <inttypes.h>
  52
  53 extern char Args[];
  54
  55 static struct stats *Outbytes;
  56
  57 static int Identlen;
  58 static int Dictlen;
  59
  60 void
  61 eftwrite_init(void)
  62 {
  63         Outbytes = stats_new_counter("eftwrite.total", "bytes written", 1);
  64 }
  65
  66 /*ARGSUSED*/
  67 static void
  68 ident_lencalc(const char *s, void *rhs, void *arg)
  69 {
  70         Identlen += strlen(s) + 1;
  71 }
  72
  73 /*ARGSUSED*/
  74 static void
  75 dict_lencalc(const char *s, void *rhs, void *arg)
  76 {
  77         Dictlen += strlen(s) + 1;
  78 }
  79
  80 /*ARGSUSED*/
  81 static void
  82 ident_printer(const char *s, void *rhs, void *arg)
  83 {
  84         FILE *fp = (FILE *)arg;
  85
  86         (void) fwrite(s, strlen(s) + 1, 1, fp);
  87 }
  88
  89 /*ARGSUSED*/
  90 static void
  91 dict_printer(const char *s, void *rhs, void *arg)
  92 {
  93         FILE *fp = (FILE *)arg;
  94
  95         (void) fwrite(s, strlen(s) + 1, 1, fp);
  96 }
  97
  98 void
  99 eftwrite(const char *fname)
 100 {
 101         FILE *fp;
 102         FILE *tfp;
 103         struct eftheader hdr;
 104 #define BUFLEN  8192
 105         char buf[BUFLEN];
 106         int cc;
 107
 108         if ((tfp = tmpfile()) == NULL)
 109                 out(O_DIE|O_SYS, "cannot create temporary file");
 110
 111         /* XXX switch stdout to tfp temporarily */
 112         /* XXX for now */
 113         out_altfp(tfp);
 114         ptree(O_ALTFP, tree_root(NULL), 0, 1);
 115
 116         rewind(tfp);
 117
 118         lut_walk(Ident, (lut_cb)ident_lencalc, NULL);
 119         lut_walk(Dicts, (lut_cb)dict_lencalc, NULL);
 120
 121         bzero(&hdr, sizeof (hdr));
 122         hdr.magic = EFT_HDR_MAGIC;
 123         hdr.major = EFT_HDR_MAJOR;
 124         hdr.minor = EFT_HDR_MINOR;
 125         hdr.cmajor = VERSION_MAJOR;
 126         hdr.cminor = VERSION_MINOR;
 127         hdr.identlen = Identlen;
 128         hdr.dictlen = Dictlen;
 129         buf[BUFLEN - 1] = '\0';
 130
 131 #ifdef DEBUG
 132         (void) snprintf(hdr.comment, EFT_HDR_MAXCOMMENT,
 133             "Built using esc-%d.%d\tArgs: \"%s\"\n", VERSION_MAJOR,
 134             VERSION_MINOR, Args);
 135 #else
 136         (void) snprintf(hdr.comment, EFT_HDR_MAXCOMMENT,
 137             "Built using esc-%d.%d\n", VERSION_MAJOR, VERSION_MINOR);
 138 #endif
 139
 140         if ((fp = fopen(fname, "w")) == NULL)
 141                 out(O_DIE|O_SYS, "can't open output file: %s", fname);
 142
 143         while ((cc = fread(buf, 1, BUFLEN, tfp)) > 0) {
 144                 char *ptr;
 145
 146                 for (ptr = buf; ptr < &buf[cc]; ptr++)
 147                         hdr.csum += (uint32_t)*ptr;
 148         }
 149         if (ferror(tfp))
 150                 out(O_DIE|O_SYS, "fread on tmpfile");
 151         rewind(tfp);
 152
 153         hdr.magic = htonl(hdr.magic);
 154         hdr.major = htons(hdr.major);
 155         hdr.minor = htons(hdr.minor);
 156         hdr.cmajor = htons(hdr.cmajor);
 157         hdr.cminor = htons(hdr.cminor);
 158         hdr.identlen = htonl(hdr.identlen);
 159         hdr.dictlen = htonl(hdr.dictlen);
 160         hdr.csum = htonl(hdr.csum);
 161
 162         (void) fwrite(&hdr, sizeof (hdr), 1, fp);
 163         if (ferror(fp))
 164                 out(O_DIE|O_SYS, "%s: can't write header", fname);
 165         stats_counter_add(Outbytes, sizeof (hdr));
 166
 167         lut_walk(Ident, (lut_cb)ident_printer, (void *)fp);
 168         stats_counter_add(Outbytes, Identlen);
 169         lut_walk(Dicts, (lut_cb)dict_printer, (void *)fp);
 170         stats_counter_add(Outbytes, Dictlen);
 171
 172         while ((cc = fread(buf, 1, BUFLEN, tfp)) > 0) {
 173                 char *ptr;
 174
 175                 for (ptr = buf; ptr < &buf[cc]; ptr++)
 176                         *ptr = ~((unsigned char)*ptr);
 177                 if (cc != fwrite(buf, 1, cc, fp) || ferror(fp))
 178                         out(O_DIE|O_SYS, "fwrite on %s", fname);
 179                 stats_counter_add(Outbytes, cc);
 180         }
 181         if (ferror(tfp))
 182                 out(O_DIE|O_SYS, "fread on tmpfile");
 183         (void) fclose(tfp);
 184         (void) fclose(fp);
 185 }