97 lines
2.7 KiB
C
97 lines
2.7 KiB
C
/* Copyright (c) 2022, 2023 Leah Rowe <info@minifree.org> */
|
|
/* SPDX-License-Identifier: MIT */
|
|
|
|
#include <sys/stat.h>
|
|
|
|
#include <dirent.h>
|
|
#include <err.h>
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <stdint.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
|
|
void openFiles(const char *path);
|
|
void readGbeFile(const char *path);
|
|
void cmd_setmac(void);
|
|
int invalidMacAddress(const char *strMac, uint16_t *mac);
|
|
uint8_t hextonum(char chs);
|
|
uint8_t rhex(void);
|
|
void cmd_dump(void);
|
|
void showmac(int partnum);
|
|
void hexdump(int partnum);
|
|
void cmd_setchecksum(void);
|
|
void cmd_brick(void);
|
|
void cmd_swap(void);
|
|
void cmd_copy(void);
|
|
int validChecksum(int partnum);
|
|
void setWord(int pos16, int partnum, uint16_t val16);
|
|
void xorswap_buf(int partnum);
|
|
void writeGbeFile(const char *filename);
|
|
|
|
#define FILENAME argv[1]
|
|
#define COMMAND argv[2]
|
|
#define MAC_ADDRESS argv[3]
|
|
#define PARTNUM argv[3]
|
|
#define SIZE_4KB 0x1000
|
|
#define SIZE_8KB 0x2000
|
|
|
|
uint16_t buf16[SIZE_4KB], mac[3] = {0, 0, 0};
|
|
uint8_t *buf = (uint8_t *) &buf16;
|
|
size_t nf = 128, gbe[2];
|
|
uint8_t nvmPartModified[2] = {0, 0}, skipread[2] = {0, 0};
|
|
int endian = 1, flags = O_RDWR, rfd, fd, part, gbeFileModified = 0;
|
|
|
|
const char *strMac = NULL, *strRMac = "??:??:??:??:??:??";
|
|
|
|
typedef struct op {
|
|
char *str;
|
|
void (*cmd)(void);
|
|
int args;
|
|
} op_t;
|
|
op_t op[] = {
|
|
{ .str = "dump", .cmd = cmd_dump, .args = 3},
|
|
{ .str = "setmac", .cmd = cmd_setmac, .args = 3},
|
|
{ .str = "swap", .cmd = cmd_swap, .args = 3},
|
|
{ .str = "copy", .cmd = cmd_copy, .args = 4},
|
|
{ .str = "brick", .cmd = cmd_brick, .args = 4},
|
|
{ .str = "setchecksum", .cmd = cmd_setchecksum, .args = 4},
|
|
};
|
|
void (*cmd)(void) = NULL;
|
|
|
|
#define ERR() errno = errno ? errno : ECANCELED
|
|
#define err_if(x) if (x) err(ERR(), NULL)
|
|
|
|
#define xopen(f,l,p) if (opendir(l) != NULL) err(errno = EISDIR, "%s", l); \
|
|
if ((f = open(l, p)) == -1) err(ERR(), "%s", l); \
|
|
if (fstat(f, &st) == -1) err(ERR(), "%s", l)
|
|
#define xpread(f, b, n, o, l) if (pread(f, b, n, o) == -1) err(ERR(), "%s", l)
|
|
#define handle_endianness(r) if (((uint8_t *) &endian)[0] ^ 1) xorswap_buf(r)
|
|
#define xpwrite(f, b, n, o, l) if (pwrite(f, b, n, o) == -1) err(ERR(), "%s", l)
|
|
#define xclose(f, l) if (close(f) == -1) err(ERR(), "%s", l)
|
|
|
|
#define xorswap(x, y) x ^= y, y ^= x, x ^= y
|
|
#define word(pos16, partnum) buf16[pos16 + (partnum << 11)]
|
|
|
|
void
|
|
xpledge(const char *promises, const char *execpromises)
|
|
{
|
|
(void)promises; (void)execpromises;
|
|
#ifdef __OpenBSD__
|
|
if (pledge(promises, execpromises) == -1)
|
|
err(ERR(), "pledge");
|
|
#endif
|
|
}
|
|
|
|
void
|
|
xunveil(const char *path, const char *permissions)
|
|
{
|
|
(void)path; (void)permissions;
|
|
#ifdef __OpenBSD__
|
|
if (unveil(path, permissions) == -1)
|
|
err(ERR(), "unveil");
|
|
#endif
|
|
}
|