/*      $NetBSD: mkboot.c,v 1.21 2025/04/05 19:57:46 tsutsui Exp $      */

/*
* Copyright (c) 1990, 1993
*      The Regents of the University of California.  All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
*    notice, this list of conditions and the following disclaimer.
* 2. 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.
* 3. Neither the name of the University 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 BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS 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.
*
*      @(#)mkboot.c    8.1 (Berkeley) 7/15/93
*/

#if HAVE_NBTOOL_CONFIG_H
#include "nbtool_config.h"
#endif

#include <sys/cdefs.h>

#ifndef lint
__COPYRIGHT("@(#) Copyright (c) 1990, 1993\
The Regents of the University of California.  All rights reserved.");
#endif /* not lint */

#ifndef lint
#ifdef notdef
static char sccsid[] = "@(#)mkboot.c    7.2 (Berkeley) 12/16/90";
#endif
__RCSID("$NetBSD: mkboot.c,v 1.21 2025/04/05 19:57:46 tsutsui Exp $");
#endif /* not lint */

#include <sys/param.h>
#include <sys/file.h>
#include <sys/stat.h>
#if HAVE_NBTOOL_CONFIG_H
#include "nbtool_config.h"
#include "../../sys/sys/bootblock.h"
#else
#include <sys/bootblock.h>
#include <sys/endian.h>
#endif

#include <time.h>

#include <ctype.h>
#include <err.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#define bintobcd(bin)   ((((bin) / 10) << 4) | ((bin) % 10))

static uint32_t loadpoint = ULONG_MAX;
static struct hp300_load ld;
static struct hp300_lifvol lifv;
static struct hp300_lifdir lifd[HP300_LIF_NUMDIR];
static time_t repro_epoch = 0;

int      main(int, char **);

static void      bcddate(char *, char *);
static char     *lifname(char *);
static size_t    putfile(char *, int);
static void      usage(void);

#ifndef __CTASSERT
#define __CTASSERT(X)
#endif

#define CLEAR(a, b, c)  \
__CTASSERT(sizeof(b) - 1 == c); \
memcpy((a), (b), (c))

/*
* Old Format:
*      sector 0:       LIF volume header (40 bytes)
*      sector 1:       <unused>
*      sector 2:       LIF directory (8 x 32 == 256 bytes)
*      sector 3-:      LIF file 0, LIF file 1, etc.
* where sectors are 256 bytes.
*
* New Format:
*      sector 0:       LIF volume header (40 bytes)
*      sector 1:       <unused>
*      sector 2:       LIF directory (8 x 32 == 256 bytes)
*      sector 3:       <unused>
*      sector 4-31:    disklabel (~300 bytes right now)
*      sector 32-:     LIF file 0, LIF file 1, etc.
*/
int
main(int argc, char **argv)
{
       char *name1, *name2, *name3;
       int to, ch;
       uint32_t count, nsec;

       while ((ch = getopt(argc, argv, "l:t:")) != -1)
               switch (ch) {
               case 'l':
                       loadpoint = strtoul(optarg, NULL, 0);
                       break;
               case 't':
                       repro_epoch = (time_t)atoll(optarg);
                       break;
               default:
                       usage();
               }

       argc -= optind;
       argv += optind;
       if (loadpoint == ULONG_MAX || argc == 0)
               usage();
       name1 = argv[0];
       argv++;
       argc--;
       if (argc == 0)
               usage();
       if (argc > 1) {
               name2 = argv[0];
               argv++;
               argc--;
               if (argc > 1) {
                       name3 = argv[0];
                       argv++;
                       argc--;
               } else
                       name3 = NULL;
       } else
               name2 = name3 = NULL;

       if ((to = open(argv[0], O_WRONLY | O_TRUNC | O_CREAT, 0644)) == -1)
               err(1, "Can't open `%s'", argv[0]);

       /* clear possibly unused directory entries */
       CLEAR(lifd[1].dir_name, "          ", sizeof(lifd[1].dir_name));
       lifd[1].dir_type = htobe16(0xFFFF);
       lifd[1].dir_addr = htobe32(0);
       lifd[1].dir_length = htobe32(0);
       lifd[1].dir_flag = htobe16(0x00FF);
       lifd[1].dir_exec = htobe32(0);
       lifd[7] = lifd[6] = lifd[5] = lifd[4] = lifd[3] = lifd[2] = lifd[1];

       /* record volume info */
       lifv.vol_id = htobe16(HP300_VOL_ID);
       CLEAR(lifv.vol_label, "BOOT43", sizeof(lifv.vol_label));
       lifv.vol_addr = htobe32(hp300_btolifs(HP300_LIF_DIRSTART));
       lifv.vol_oct = htobe16(HP300_VOL_OCT);
       lifv.vol_dirsize = htobe32(hp300_btolifs(HP300_LIF_DIRSIZE));
       lifv.vol_version = htobe16(1);

       /* output bootfile one */
       lseek(to, HP300_LIF_FILESTART, SEEK_SET);
       count = putfile(name1, to);
       nsec = hp300_btolifs(count);
       strcpy(lifd[0].dir_name, lifname(name1));
       lifd[0].dir_type = htobe16(HP300_DIR_TYPE);
       lifd[0].dir_addr = htobe32(hp300_btolifs(HP300_LIF_FILESTART));
       lifd[0].dir_length = htobe32(nsec);
       bcddate(name1, lifd[0].dir_toc);
       lifd[0].dir_flag = htobe16(HP300_DIR_FLAG);
       lifd[0].dir_exec = htobe32(loadpoint);
       lifv.vol_length = htobe32(be32toh(lifd[0].dir_addr) +
           be32toh(lifd[0].dir_length));

       /* if there is an optional second boot program, output it */
       if (name2 != NULL) {
               lseek(to, HP300_LIF_FILESTART + hp300_lifstob(nsec), SEEK_SET);
               count = putfile(name2, to);
               nsec = hp300_btolifs(count);
               strcpy(lifd[1].dir_name, lifname(name2));
               lifd[1].dir_type = htobe16(HP300_DIR_TYPE);
               lifd[1].dir_addr = htobe32(lifv.vol_length);
               lifd[1].dir_length = htobe32(nsec);
               bcddate(name2, lifd[1].dir_toc);
               lifd[1].dir_flag = htobe16(HP300_DIR_FLAG);
               lifd[1].dir_exec = htobe32(loadpoint);
               lifv.vol_length = htobe32(be32toh(lifd[1].dir_addr) +
                   be32toh(lifd[1].dir_length));
       }

       /* ditto for three */
       if (name3 != NULL) {
               lseek(to, HP300_LIF_FILESTART + hp300_lifstob(lifd[0].dir_length
                   + nsec), SEEK_SET);
               count = putfile(name3, to);
               nsec = hp300_btolifs(count);
               strcpy(lifd[2].dir_name, lifname(name3));
               lifd[2].dir_type = htobe16(HP300_DIR_TYPE);
               lifd[2].dir_addr = htobe32(lifv.vol_length);
               lifd[2].dir_length = htobe32(nsec);
               bcddate(name3, lifd[2].dir_toc);
               lifd[2].dir_flag = htobe16(HP300_DIR_FLAG);
               lifd[2].dir_exec = htobe32(loadpoint);
               lifv.vol_length = htobe32(be32toh(lifd[2].dir_addr) +
                   be32toh(lifd[2].dir_length));
       }

       /* output volume/directory header info */
       lseek(to, HP300_LIF_VOLSTART, SEEK_SET);
       write(to, &lifv, HP300_LIF_VOLSIZE);
       lseek(to, HP300_LIF_DIRSTART, SEEK_SET);
       write(to, lifd, HP300_LIF_DIRSIZE);

       return EXIT_SUCCESS;
}

static size_t
putfile(char *from, int to)
{
       int fd;
       struct stat statb;
       void *bp;

       if ((fd = open(from, 0)) < 0)
               err(EXIT_FAILURE, "Unable to open file `%s'", from);
       fstat(fd, &statb);
       ld.address = htobe32(loadpoint);
       ld.count = htobe32(statb.st_size);
       if ((bp = malloc(statb.st_size)) == NULL)
               err(EXIT_FAILURE, "Can't allocate buffer");
       if (read(fd, bp, statb.st_size) < 0)
               err(EXIT_FAILURE, "Error reading from file `%s'", from);
       (void)close(fd);
       write(to, &ld, sizeof(ld));
       write(to, bp, statb.st_size);
       free(bp);

       return statb.st_size + sizeof(ld);
}

static void
usage(void)
{

       fprintf(stderr, "Usage: %s -l <loadpoint> [-t <timestamp>] prog1 "
           "[ prog2 ] outfile\n", getprogname());
       exit(EXIT_FAILURE);
}

static char *
lifname(char *str)
{
       static char lname[10] = "SYS_XXXXX";
       char *cp;
       int i;

       if ((cp = strrchr(str, '/')) != NULL)
               str = ++cp;
       for (i = 4; i < 9; i++) {
               if (islower((unsigned char)*str))
                       lname[i] = toupper((unsigned char)*str);
               else if (isalnum((unsigned char)*str) || *str == '_')
                       lname[i] = *str;
               else
                       break;
               str++;
       }
       for (; i < 10; i++)
               lname[i] = '\0';

       return lname;
}

static void
bcddate(char *name, char *toc)
{
       struct stat statb;
       struct tm *tm;

       if (repro_epoch != 0)
               tm = gmtime(&repro_epoch);
       else {
               stat(name, &statb);
               tm = localtime(&statb.st_ctime);
       }
       *toc++ = bintobcd(tm->tm_mon + 1);
       *toc++ = bintobcd(tm->tm_mday);
       *toc++ = bintobcd(tm->tm_year);
       *toc++ = bintobcd(tm->tm_hour);
       *toc++ = bintobcd(tm->tm_min);
       *toc   = bintobcd(tm->tm_sec);
}