/* $NetBSD: ustarfs.c,v 1.9 2019/01/09 03:28:31 christos Exp $ */
/*-
* Copyright (c) 2004 The NetBSD Foundation, Inc.
* All rights reserved.
*
* This code is derived from software contributed to The NetBSD Foundation
* by UCHIYAMA Yasushi.
*
* 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.
*
* THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. 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 FOUNDATION 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.
*/
#include <lib/libsa/stand.h>
#include <lib/libkern/libkern.h>
#include <sys/param.h>
#include <machine/sbd.h>
#include <machine/sector.h>
#include "local.h"
#include "common.h"
static bool __ustarfs_file(int, char *, size_t *);
static bool __block_read(uint8_t *, int);
static bool __block_read_n(uint8_t *, int, int);
static void __change_volume(int);
static int __block2volume(int);
static int __volume_offset(int);
static int __next_block(int);
enum { USTAR_BLOCK_SIZE = 8192 };/* Check src/distrib/common/buildfloppies.sh */
struct volume {
int max_block;
int current_volume;
} __volume;
bool
ustarfs_load(const char *file, void **addrp, size_t *sizep)
{
char fname[16];
int block = 16;
size_t sz;
int maxblk;
if (DEVICE_CAPABILITY.active_device == NVSRAM_BOOTDEV_HARDDISK)
maxblk = 0x1fffffff; /* no limit */
else if (DEVICE_CAPABILITY.active_device == NVSRAM_BOOTDEV_FLOPPYDISK)
/*
* Although phisical format isn't 2D, volume size is
* limited to size of 2D.
*/
maxblk = (77 + 76) * 13;
else {
printf("not supported device.\n");
return false;
}
/* Truncate to ustar block boundary */
__volume.max_block = (maxblk / (USTAR_BLOCK_SIZE >> DEV_BSHIFT)) *
(USTAR_BLOCK_SIZE >> DEV_BSHIFT);
__volume.current_volume = 0;
/* Find file */
while (/*CONSTCOND*/1) {
if (!__ustarfs_file(block, fname, &sz))
return false;
if (strcmp(file, fname) == 0)
break;
block += (ROUND_SECTOR(sz) >> DEV_BSHIFT) + 1;
}
block++; /* skip tar header */
*sizep = sz;
/* Load file */
sz = ROUND_SECTOR(sz);
if ((*addrp = alloc(sz)) == 0) {
printf("%s: can't allocate memory.\n", __func__);
return false;
}
if (!__block_read_n(*addrp, block, sz >> DEV_BSHIFT)) {
printf("%s: can't load file.\n", __func__);
dealloc(*addrp, sz);
return false;
}
return true;
}
bool
__ustarfs_file(int start_block, char *file, size_t *size)
{
uint8_t buf[512];
if (!__block_read(buf, start_block)) {
printf("can't read tar header.\n");
return false;
}
if (((*(uint32_t *)(buf + 256)) & 0xffffff) != 0x757374) {
printf("bad tar magic.\n");
return false;
}
*size = strtoul((char *)buf + 124, 0, 0);
strncpy(file, (char *)buf, 16);
return true;
}
bool
__block_read_n(uint8_t *buf, int blk, int count)
{
int vol, cnt;
while (count > 0) {
vol = __block2volume(blk);
if (vol != __volume.current_volume) {
__change_volume(vol);
__volume.current_volume = vol;
}
cnt = __next_block(vol) - blk;
if (cnt > count)
cnt = count;
if (!sector_read_n(0, buf, __volume_offset(blk), cnt))
return false;
count -= cnt;
blk += cnt;
buf += cnt * DEV_BSIZE;
}
return true;
}
bool
__block_read(uint8_t *buf, int blk)
{
int vol;
vol = __block2volume(blk);
if (vol != __volume.current_volume) {
__change_volume(vol);
__volume.current_volume = vol;
}
return sector_read(0, buf, __volume_offset(blk));
}
void
__change_volume(int volume)
{
uint8_t buf[512];
int i;
while (/*CONSTCOND*/1) {
printf("insert disk %d, and press return...", volume + 1);
while (getchar() != '\r')
;
printf("\n");
if (!sector_read(0, buf, 0))
continue;
if (*(uint32_t *)buf != 0x55535441) { /* "USTAR" */
printf("invalid magic.\n");
continue;
}
if ((i = buf[8] - '0') != volume + 1) {
printf("invalid volume number. disk=%d requested=%d\n",
i, volume + 1);
continue;
}
return;
}
}
int
__block2volume(int blk)
{
if (blk < __volume.max_block)
return 0;
blk -= __volume.max_block;
return (blk / (__volume.max_block - 16)) + 1;
}
int
__volume_offset(int blk)
{
if (blk < __volume.max_block)
return blk;
blk -= __volume.max_block;
return blk % (__volume.max_block - 16) + 16;
}
int
__next_block(int vol)
{
return __volume.max_block + vol * (__volume.max_block - 16);
}