[rsbs2] / firmware.c

#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
#include <strings.h>
#include "rsb-crc.h"
#include "extract.h"

#define FINDSTR(addr, str) (!strncmp((char*)addr, str, strlen(str)))

struct properties {
	unsigned int magic;
	unsigned char unknown0;
	unsigned char unknown1;
	unsigned char right_rw;
	unsigned char rw_mask;
	unsigned char type1;
	unsigned char unknown5;
	unsigned char unknown6;
	unsigned char unknown7;
	unsigned char type2;
	unsigned char val[];
};

#define PROP_ACTION_TRUE	(1<<0)
#define PROP_ACTION_FALSE	(1<<1)
#define PROP_ACTION_RO		(1<<2)
#define PROP_ACTION_RW		(1<<3)

#define PROP_STATUS_NOTFOUND	(0)
#define PROP_STATUS_WRONGTYPE	(1<<0)
#define PROP_STATUS_WRONGRIGHTS	(1<<1)
#define PROP_STATUS_SAMEVAL	(1<<2)
#define PROP_STATUS_SUCCESS	(1<<3)

struct propaction {
	char *property;
	unsigned int action;
	unsigned int status;
	struct propaction *next;
};

void show_properties(unsigned char *fw, int len)
{
	struct file_entry *fent;

	for (fent = get_next_file(fw, len); fent != NULL; fent = get_next_file(NULL, 0)) {
		if (FINDSTR(fent->name, "/default/fw_prop/") ||
		    FINDSTR(fent->name, "/default/fw_setup/") ||
		    FINDSTR(fent->name, "/default/oem_prop/")) {
		    	struct properties *prop;

			printf("0x%08x: found setting: %s ", fent->start - fw, fent->name);

			prop = (struct properties*)fent->start;
			
			if (prop->magic != 0x83011111) {
				printf("ignoring...\n");
				continue;
			}

			if (prop->type1 == 0x00 && prop->type2 == 0x04) {
				printf("STRING: '%s' ", prop->val);
			} else if (prop->type1 == 0x01 && prop->type2 == 0x01) {
				printf("BOOL: %s ",(*prop->val ? "TRUE" : "FALSE"));
			} else if (prop->type1 == 0x04 && prop->type2 == 0x02) {
				printf("VAL: 0x%x ", *((unsigned int*)prop->val));
			} else {
				printf("0x%02x 0x%2x...ignoring\n", prop->type1, prop->type2);
				continue;
			}

			if (prop->right_rw == 0x00 && prop->rw_mask == 0x00) {
				printf("(R-)");
			} else if (prop->right_rw == 0x01) {
				printf("(RW mask: 0x%02x)", prop->rw_mask);
			} else {
				printf("(UNK 0x%02x 0x%02x)", prop->right_rw, prop->rw_mask);
			}
			printf(", length: %d\n", fent->length);
		}
	}
}

void change_properties(unsigned char *fw, int len, struct propaction *paction)
{
	struct file_entry *fent;
	struct propaction *cpaction;

	for (fent = get_next_file(fw, len); fent != NULL; fent = get_next_file(NULL, 0)) {
		cpaction = paction;
		while (cpaction != NULL) {
			if (FINDSTR(fent->name, cpaction->property)) {
				break;
			}
			cpaction = cpaction->next;
		}
		if (cpaction != NULL) {
		    	struct properties *prop;

			prop = (struct properties*)fent->start;

			if (prop->magic != 0x83011111) {
				continue;
			}

			if (cpaction->action & (PROP_ACTION_TRUE|PROP_ACTION_FALSE)) {
				if (prop->type1 == 0x01 && prop->type2 == 0x01) {
					if (cpaction->action & PROP_ACTION_TRUE) {
						if (*prop->val == 0x00) {
							*prop->val = 0x01;
							cpaction->status |= PROP_STATUS_SUCCESS;
						} else {
							cpaction->status |= PROP_STATUS_SAMEVAL;
						}
					} else {
						if (*prop->val == 0x01) {
							*prop->val = 0x00;
							cpaction->status |= PROP_STATUS_SUCCESS;
						} else {
							cpaction->status |= PROP_STATUS_SAMEVAL;
						}
					}
				} else {
					cpaction->status = PROP_STATUS_WRONGTYPE;
				}
			}
			if (cpaction->action & PROP_ACTION_RW) {
				if (prop->right_rw == 0x00 && prop->rw_mask == 0x00) {
					prop->right_rw = 0x01;
					prop->rw_mask = 0x02;
					cpaction->status |= PROP_STATUS_SUCCESS;
				} else {
					cpaction->status |= PROP_STATUS_WRONGRIGHTS;
				}
			}
			if (cpaction->action & PROP_ACTION_RO) {
				if (prop->right_rw == 0x01 && prop->rw_mask == 0x02) {
					prop->right_rw = 0x00;
					prop->rw_mask = 0x00;
					cpaction->status |= PROP_STATUS_SUCCESS;
				} else {
					cpaction->status |= PROP_STATUS_WRONGRIGHTS;
				}
			}
		}
	}
}

#define BD_SERIAL1	0x14,0x02
#define BD_ICMB		0x14,0x04
#define BD_LAN		0x14,0x08
#define BD_SERIAL2	0x14,0x10
#define BD_SERIAL3	0x14,0x20
#define BD_USB		0x14,0x40
#define BD_PCI		0x15,0x03
#define BD_LPC		0x15,0x04
#define BD_VGA		0x15,0x08
#define BD_BATTERY	0x15,0x10
#define BD_ACDC		0x15,0x20
#define BD_STANDBY	0x15,0x40
#define BD_POWERCONN	0x15,0x70
#define BD_DVI		0x15,0x80
#define BD_PWRATX	0x16,0x01
#define BD_PWRRELAY	0x16,0x02
#define BD_PS2A		0x19,0xff

#define MAGIC(fn, args...) fn(args)

#define _BD_IS_SET(bd, byte, bits) (bd[byte] & bits)
#define BD_IS_SET(bd, ident) MAGIC(_BD_IS_SET, bd, BD_##ident)
#define BD_TEXT(bd, ident) (BD_IS_SET(bd, ident) ? "TRUE" : "FALSE")

#define _BD_SET(bd, byte, bits) (bd[byte] |= bits)
#define BD_SET(bd, ident) MAGIC(_BD_SET, bd, BD_##ident)

void print_boarddescription(unsigned char *bd)
{
	int j;

	for (j = 0; j < 32; j++) {
		printf("%02x ", *(bd+j));
	}
	printf("\n");

	/* com/agilent/rmc/amr/AmrMaster.class
	 * com/agilent/rmc/mgui/RmcPanel.class
	 * com/agilent/rmc/mgui/panels/AvrManualConfig.class
	 * com/agilent/rmc/mgui/panels/CardConf.jad
	 * com/agilent/rmc/mgui/panels/PowerMgmtConf.jad
	 * com/agilent/rmc/mgui/panels/RemoteDiskConf.jad
	 */
	printf("\tserial1Present\t\t: %s\n", BD_TEXT(bd, SERIAL1));
	printf("\ticmbPresent\t\t: %s\n", BD_TEXT(bd, ICMB));
	printf("\tlanPresent\t\t: %s\n", BD_TEXT(bd, LAN));
	printf("\tserial2Present\t\t: %s\n", BD_TEXT(bd, SERIAL2));
	printf("\tserial3Present\t\t: %s\n", BD_TEXT(bd, SERIAL3));
	printf("\tusbPresent\t\t: %s\n", BD_TEXT(bd, USB));
	printf("\tpciPresent\t\t: %s\n", BD_TEXT(bd, PCI));
	printf("\tlpcPresent\t\t: %s\n", BD_TEXT(bd, LPC));
	printf("\tvgaPresent\t\t: %s\n", BD_TEXT(bd, VGA));
	printf("\tbatteryPresent\t\t: %s\n", BD_TEXT(bd, BATTERY));
	printf("\tacdcPresent\t\t: %s\n", BD_TEXT(bd, ACDC));
	printf("\tstandbyPresent\t\t: %s\n", BD_TEXT(bd, STANDBY));
	printf("\thasPowerConnectors\t: %s\n", BD_TEXT(bd, POWERCONN));
	printf("\tdviPresent\t\t: %s\n", BD_TEXT(bd, DVI));
	printf("\tpowerSwitchATX\t\t: %s\n", BD_TEXT(bd, PWRATX));
	printf("\tpowerSwitchRelay\t: %s\n", BD_TEXT(bd, PWRRELAY));
	/* 22 & 4 */
	printf("\tps2aPresent\t\t: %s\n", BD_TEXT(bd, PS2A));
}

void handle_boarddescription(unsigned char *fw, int len, int patch)
{
	struct file_entry *fent;
	unsigned char *pos;
	
	for (fent = get_next_file(fw, len); fent != NULL; fent = get_next_file(NULL, 0)) {
		if (!strcmp(fent->name, "pdata"))
			break;
	}

	if (fent == NULL) {
		fprintf(stderr, "pdata file not found, aborting!\n");
		exit(1);
	}


	pos = fent->start;
	/* MAGIC? */
	if (*((unsigned int*)pos) != 0x00002802) {
		fprintf(stderr, "pdata magic does not match, aborting!\n");
		exit(1);
	}

	pos += 26;

	/* MAGIC2? */
	if (*((unsigned int*)pos) != 0x00500101) {
		fprintf(stderr, "pdata magic2 does not match, aborting!\n");
		exit(1);
	}

	if (patch) {
		/* Enable ATX and relay power switching */
		BD_SET(pos, PWRATX);
		BD_SET(pos, PWRRELAY);
	}

	printf("0x%08x: BOARD_DESCRIPTION: ", fent->start - fw);
	print_boarddescription(pos);
}

void syntax(char *name)
{
	fprintf(stderr,"Syntax: %s parameters firmware.bin\n", name);
	fprintf(stderr,"parameters as follows:\n");
	fprintf(stderr,"\t-d\t\tdisplay all properties of the image\n");
	fprintf(stderr,"\t-u\t\tupdate checksum of the image\n");
	fprintf(stderr,"\t-b\t\tmodify BOARD_DESCRIPTION for more power-switch options\n");
	fprintf(stderr,"\t-e\t\textract files in firmware\n");
	fprintf(stderr,"\t-t property\tset 'property' to true\n");
	fprintf(stderr,"\t-f property\tset 'property' to false\n");
	fprintf(stderr,"\t-w property\tallow read-write access to 'property'\n");
	fprintf(stderr,"\t-r property\tallow read-only access to 'property'\n");
	exit(1);
}

void add_action(int opt, char *optarg, struct propaction **paction) {
	struct propaction *pos = *paction;
	struct propaction *prev = NULL;
	
	while (pos != NULL) {
		if (!strcmp(pos->property, optarg))
			break;
		prev = pos;
		pos = pos->next;
	}

	if (pos == NULL) {
		pos = malloc(sizeof(struct propaction));
		if (pos == NULL) {
			perror("malloc");
			exit(1);
		}
		bzero(pos, sizeof(struct propaction));
		pos->property = optarg;

		if (prev == NULL) {
			*paction = pos;
		} else {
			prev->next = pos;
		}
	}

	switch(opt) {
		case 't':
			if (pos->action & PROP_ACTION_FALSE) {
				fprintf(stderr,"inconsistent requests for %s\n",pos->property);
				exit(1);
			}
			pos->action |= PROP_ACTION_TRUE;
			break;
		case 'f':
			if (pos->action & PROP_ACTION_TRUE) {
				fprintf(stderr,"inconsistent requests for %s\n",pos->property);
				exit(1);
			}
			pos->action |= PROP_ACTION_FALSE;
			break;
		case 'w':
			if (pos->action & PROP_ACTION_RO) {
				fprintf(stderr,"inconsistent requests for %s\n",pos->property);
				exit(1);
			}
			pos->action |= PROP_ACTION_RW;
			break;
		case 'r':
			if (pos->action & PROP_ACTION_RW) {
				fprintf(stderr,"inconsistent requests for %s\n",pos->property);
				exit(1);
			}
			pos->action |= PROP_ACTION_RO;
			break;
	}
}

int check_crc(unsigned char *fw, int len)
{
	int ret;
	unsigned int crc, oldcrc;

	ret = rsb_crc2(fw, len, 0x55335053, &crc);
	oldcrc = (unsigned int)*((unsigned int*)(fw + len - 4));

	printf("Checksum: 0x%08x (%s), should be: 0x%08x\n",
		crc,
		(ret ? "NOT OK" : "OK"),
		oldcrc);

	return ret;
}

int check_image(unsigned char *fw, int len)
{
	struct file_entry *fent;
	char *last_name = NULL;

	fent = get_next_file(fw, len);
	while (fent != NULL) {
		last_name = fent->name;
		fent = get_next_file(NULL, 0);
	}

	if (strcmp(last_name, "pdata")) {
		return 1;
	}

	return 0;
}

int main(int argc, char **argv)
{
	struct stat statbuf;
	char *file = NULL;
	unsigned char *fw;
	int fd;
	int remaining;
	int ret;
	int opt;
	unsigned int crc;
	struct propaction *paction = NULL;
	int showall = 0;
	int update_crc = 0;
	int patch_bd = 0;
	int patch_fw = 0;
	int extract = 0;

	if (argc < 2)
		syntax(argv[0]);

	while ((opt = getopt(argc, argv, "dubet:f:w:r:")) != -1) {
		switch(opt) {
			case 'd':
				showall = 1;
				break;
			case 'u':
				update_crc = 1;
				break;
			case 'b':
				patch_bd = 1;
				break;
			case 'e':
				extract = 1;
				break;
			case 't':
			case 'f':
			case 'w':
			case 'r':
				patch_fw = 1;
				add_action(opt, optarg, &paction);
				break;
			default:
				syntax(argv[0]);
		}
	}

	if (argc > optind) {
		file = argv[optind];
	} else {
		syntax(argv[0]);
	}

	if (stat(file, &statbuf) == -1) {
		fprintf(stderr,"%s: ", file);
		perror("stat");
		exit(1);
	}

	if ((fd = open(file, O_RDONLY)) == -1) {
		fprintf(stderr,"%s: ", file);
		perror("open");
		exit(1);
	}

	if ((fw = malloc(statbuf.st_size)) == NULL) {
		perror("malloc");
		exit(1);
	}

	bzero(fw, statbuf.st_size);

	remaining = statbuf.st_size;

	while(remaining) {
		if ((ret = read(fd, fw + (statbuf.st_size - remaining), remaining)) == -1) {
			perror("read");
			exit(1);
		}
		remaining -= ret;
	}
	close(fd);

	ret = check_crc(fw, statbuf.st_size);
	if ((ret != 0) && (!update_crc)) {
		fprintf(stderr,"Checksum incorrect, aborting...\n");
		exit(1);
	}

	if (check_image(fw, statbuf.st_size-4) != 0) {
		fprintf(stderr, "corrupt firmware image found (pdata is not last entry), aborting!\n");
		exit(1);
	}

	if (patch_fw) {
		struct propaction *cpaction = paction;

		change_properties(fw, statbuf.st_size, paction);

		printf("\nProperty change results:\n");
		while(cpaction != NULL) {
			printf("%s: ", cpaction->property);

			if (cpaction->status == PROP_STATUS_NOTFOUND)
				printf("NOTFOUND ");
			if (cpaction->status & PROP_STATUS_SUCCESS)
				printf("SUCCESS ");
			if (cpaction->status & PROP_STATUS_SAMEVAL)
				printf("SAMEVAL ");
			if (cpaction->status & PROP_STATUS_WRONGTYPE)
				printf("WRONGTYPE ");
			if (cpaction->status & PROP_STATUS_WRONGRIGHTS)
				printf("WRONGRIGHTS ");
			printf("\n");
			
			cpaction = cpaction->next;
		}
		printf("\n");
	}

	if (patch_bd) {
		handle_boarddescription(fw, statbuf.st_size -4, 1);
	}

	if (showall) {
		show_properties(fw, statbuf.st_size - 4);
		handle_boarddescription(fw, statbuf.st_size -4, 0);
	}

	if (extract) {
		extract_files(fw, statbuf.st_size - 4);
	}

	if (update_crc || patch_fw || patch_bd) {
		ret = rsb_crc2(fw, statbuf.st_size, 0x55335053, &crc);
		if (ret == 4) {
			*((unsigned int*)(fw + statbuf.st_size - 4)) = crc;
		}

		if (check_image(fw, statbuf.st_size-4) != 0) {
			fprintf(stderr, "corrupt firmware image found (pdata is not last entry), aborting!\n");
			exit(1);
		}

		if (check_crc(fw, statbuf.st_size) == 0) {
			char *newfile;

			newfile = malloc(strlen(file) + strlen(".patched") + 1);
			if (newfile == NULL) {
				perror("malloc");
				exit(1);
			}
			strcpy(newfile, file);
			strcat(newfile, ".patched");

			printf("Writing %s\n", newfile);
			if ((fd = open(newfile, O_WRONLY|O_CREAT, 0644)) == -1) {
				fprintf(stderr,"%s: ", file);
				perror("open");
				exit(1);
			}

			remaining = statbuf.st_size;

			while(remaining) {
				if ((ret = write(fd, fw + (statbuf.st_size - remaining), remaining)) == -1) {
					perror("write");
					exit(1);
				}
				remaining -= ret;
			}
			close(fd);
		} else {
			fprintf(stderr,"Can't set correct checksum, aborting...\n");
		}
	}
	
	exit(0);
}
Commit	Line	Data
972ac24b MG	1	#include <stdio.h>
	2	#include <stdlib.h>
	3	#include <sys/types.h>
	4	#include <sys/stat.h>
	5	#include <fcntl.h>
	6	#include <unistd.h>
	7	#include <errno.h>
7f88d2b6	8	#include <string.h>
972ac24b MG	9	#include <strings.h>
972ac24b MG	10	#include "rsb-crc.h"
1f1fa7b6	11	#include "extract.h"
972ac24b	12
7f88d2b6 MG	13	#define FINDSTR(addr, str) (!strncmp((char*)addr, str, strlen(str)))
7f88d2b6 MG	14
39601b0e MG	15	struct properties {
	16	unsigned int magic;
	17	unsigned char unknown0;
	18	unsigned char unknown1;
	19	unsigned char right_rw;
	20	unsigned char rw_mask;
	21	unsigned char type1;
	22	unsigned char unknown5;
	23	unsigned char unknown6;
	24	unsigned char unknown7;
	25	unsigned char type2;
	26	unsigned char val[];
	27	};
	28
7ac4bfad MG	29	#define PROP_ACTION_TRUE (1<<0)
	30	#define PROP_ACTION_FALSE (1<<1)
	31	#define PROP_ACTION_RO (1<<2)
	32	#define PROP_ACTION_RW (1<<3)
	33
	34	#define PROP_STATUS_NOTFOUND (0)
	35	#define PROP_STATUS_WRONGTYPE (1<<0)
	36	#define PROP_STATUS_WRONGRIGHTS (1<<1)
	37	#define PROP_STATUS_SAMEVAL (1<<2)
	38	#define PROP_STATUS_SUCCESS (1<<3)
	39
	40	struct propaction {
	41	char *property;
	42	unsigned int action;
	43	unsigned int status;
	44	struct propaction *next;
	45	};
	46
39601b0e	47	void show_properties(unsigned char *fw, int len)
7f88d2b6	48	{
0a465cc0	49	struct file_entry *fent;
7f88d2b6	50
0a465cc0 MG	51	for (fent = get_next_file(fw, len); fent != NULL; fent = get_next_file(NULL, 0)) {
	52	if (FINDSTR(fent->name, "/default/fw_prop/") \|\|
	53	FINDSTR(fent->name, "/default/fw_setup/") \|\|
	54	FINDSTR(fent->name, "/default/oem_prop/")) {
39601b0e	55	struct properties *prop;
7f88d2b6	56
0a465cc0	57	printf("0x%08x: found setting: %s ", fent->start - fw, fent->name);
7f88d2b6	58
0a465cc0	59	prop = (struct properties*)fent->start;
39601b0e MG	60
39601b0e MG	61	if (prop->magic != 0x83011111) {
7f88d2b6 MG	62	printf("ignoring...\n");
	63	continue;
	64	}
	65
39601b0e	66	if (prop->type1 == 0x00 && prop->type2 == 0x04) {
7ac4bfad	67	printf("STRING: '%s' ", prop->val);
39601b0e MG	68	} else if (prop->type1 == 0x01 && prop->type2 == 0x01) {
	69	printf("BOOL: %s ",(*prop->val ? "TRUE" : "FALSE"));
	70	} else if (prop->type1 == 0x04 && prop->type2 == 0x02) {
	71	printf("VAL: 0x%x ", ((unsigned int)prop->val));
7f88d2b6	72	} else {
39601b0e	73	printf("0x%02x 0x%2x...ignoring\n", prop->type1, prop->type2);
7f88d2b6 MG	74	continue;
	75	}
	76
39601b0e	77	if (prop->right_rw == 0x00 && prop->rw_mask == 0x00) {
a519eca7	78	printf("(R-)");
39601b0e	79	} else if (prop->right_rw == 0x01) {
a519eca7	80	printf("(RW mask: 0x%02x)", prop->rw_mask);
7f88d2b6	81	} else {
a519eca7	82	printf("(UNK 0x%02x 0x%02x)", prop->right_rw, prop->rw_mask);
7f88d2b6	83	}
0a465cc0	84	printf(", length: %d\n", fent->length);
7f88d2b6 MG	85	}
	86	}
	87	}
	88
7ac4bfad MG	89	void change_properties(unsigned char fw, int len, struct propaction paction)
7ac4bfad MG	90	{
0a465cc0	91	struct file_entry *fent;
7ac4bfad MG	92	struct propaction *cpaction;
7ac4bfad MG	93
0a465cc0	94	for (fent = get_next_file(fw, len); fent != NULL; fent = get_next_file(NULL, 0)) {
7ac4bfad MG	95	cpaction = paction;
7ac4bfad MG	96	while (cpaction != NULL) {
0a465cc0	97	if (FINDSTR(fent->name, cpaction->property)) {
7ac4bfad MG	98	break;
	99	}
	100	cpaction = cpaction->next;
	101	}
	102	if (cpaction != NULL) {
	103	struct properties *prop;
7ac4bfad	104
0a465cc0	105	prop = (struct properties*)fent->start;
7ac4bfad MG	106
	107	if (prop->magic != 0x83011111) {
	108	continue;
	109	}
	110
	111	if (cpaction->action & (PROP_ACTION_TRUE\|PROP_ACTION_FALSE)) {
	112	if (prop->type1 == 0x01 && prop->type2 == 0x01) {
	113	if (cpaction->action & PROP_ACTION_TRUE) {
	114	if (*prop->val == 0x00) {
	115	*prop->val = 0x01;
	116	cpaction->status \|= PROP_STATUS_SUCCESS;
	117	} else {
	118	cpaction->status \|= PROP_STATUS_SAMEVAL;
	119	}
	120	} else {
	121	if (*prop->val == 0x01) {
	122	*prop->val = 0x00;
	123	cpaction->status \|= PROP_STATUS_SUCCESS;
	124	} else {
	125	cpaction->status \|= PROP_STATUS_SAMEVAL;
	126	}
	127	}
	128	} else {
	129	cpaction->status = PROP_STATUS_WRONGTYPE;
	130	}
	131	}
	132	if (cpaction->action & PROP_ACTION_RW) {
	133	if (prop->right_rw == 0x00 && prop->rw_mask == 0x00) {
	134	prop->right_rw = 0x01;
	135	prop->rw_mask = 0x02;
	136	cpaction->status \|= PROP_STATUS_SUCCESS;
	137	} else {
	138	cpaction->status \|= PROP_STATUS_WRONGRIGHTS;
	139	}
	140	}
	141	if (cpaction->action & PROP_ACTION_RO) {
	142	if (prop->right_rw == 0x01 && prop->rw_mask == 0x02) {
	143	prop->right_rw = 0x00;
	144	prop->rw_mask = 0x00;
	145	cpaction->status \|= PROP_STATUS_SUCCESS;
	146	} else {
	147	cpaction->status \|= PROP_STATUS_WRONGRIGHTS;
	148	}
	149	}
	150	}
	151	}
	152	}
	153
68fc92b2 MG	154	#define BD_SERIAL1 0x14,0x02
	155	#define BD_ICMB 0x14,0x04
	156	#define BD_LAN 0x14,0x08
	157	#define BD_SERIAL2 0x14,0x10
	158	#define BD_SERIAL3 0x14,0x20
	159	#define BD_USB 0x14,0x40
	160	#define BD_PCI 0x15,0x03
	161	#define BD_LPC 0x15,0x04
	162	#define BD_VGA 0x15,0x08
	163	#define BD_BATTERY 0x15,0x10
	164	#define BD_ACDC 0x15,0x20
	165	#define BD_STANDBY 0x15,0x40
	166	#define BD_POWERCONN 0x15,0x70
	167	#define BD_DVI 0x15,0x80
	168	#define BD_PWRATX 0x16,0x01
	169	#define BD_PWRRELAY 0x16,0x02
	170	#define BD_PS2A 0x19,0xff
	171
	172	#define MAGIC(fn, args...) fn(args)
	173
	174	#define _BD_IS_SET(bd, byte, bits) (bd[byte] & bits)
	175	#define BD_IS_SET(bd, ident) MAGIC(_BD_IS_SET, bd, BD_##ident)
	176	#define BD_TEXT(bd, ident) (BD_IS_SET(bd, ident) ? "TRUE" : "FALSE")
	177
	178	#define _BD_SET(bd, byte, bits) (bd[byte] \|= bits)
	179	#define BD_SET(bd, ident) MAGIC(_BD_SET, bd, BD_##ident)
	180
453260c6 MG	181	void print_boarddescription(unsigned char *bd)
	182	{
	183	int j;
	184
	185	for (j = 0; j < 32; j++) {
	186	printf("%02x ", *(bd+j));
	187	}
	188	printf("\n");
f370a858 MG	189
	190	/* com/agilent/rmc/amr/AmrMaster.class
	191	* com/agilent/rmc/mgui/RmcPanel.class
	192	* com/agilent/rmc/mgui/panels/AvrManualConfig.class
	193	* com/agilent/rmc/mgui/panels/CardConf.jad
	194	* com/agilent/rmc/mgui/panels/PowerMgmtConf.jad
	195	* com/agilent/rmc/mgui/panels/RemoteDiskConf.jad
	196	*/
68fc92b2 MG	197	printf("\tserial1Present\t\t: %s\n", BD_TEXT(bd, SERIAL1));
	198	printf("\ticmbPresent\t\t: %s\n", BD_TEXT(bd, ICMB));
	199	printf("\tlanPresent\t\t: %s\n", BD_TEXT(bd, LAN));
	200	printf("\tserial2Present\t\t: %s\n", BD_TEXT(bd, SERIAL2));
	201	printf("\tserial3Present\t\t: %s\n", BD_TEXT(bd, SERIAL3));
	202	printf("\tusbPresent\t\t: %s\n", BD_TEXT(bd, USB));
	203	printf("\tpciPresent\t\t: %s\n", BD_TEXT(bd, PCI));
	204	printf("\tlpcPresent\t\t: %s\n", BD_TEXT(bd, LPC));
	205	printf("\tvgaPresent\t\t: %s\n", BD_TEXT(bd, VGA));
	206	printf("\tbatteryPresent\t\t: %s\n", BD_TEXT(bd, BATTERY));
	207	printf("\tacdcPresent\t\t: %s\n", BD_TEXT(bd, ACDC));
	208	printf("\tstandbyPresent\t\t: %s\n", BD_TEXT(bd, STANDBY));
	209	printf("\thasPowerConnectors\t: %s\n", BD_TEXT(bd, POWERCONN));
	210	printf("\tdviPresent\t\t: %s\n", BD_TEXT(bd, DVI));
	211	printf("\tpowerSwitchATX\t\t: %s\n", BD_TEXT(bd, PWRATX));
	212	printf("\tpowerSwitchRelay\t: %s\n", BD_TEXT(bd, PWRRELAY));
f370a858	213	/* 22 & 4 */
68fc92b2	214	printf("\tps2aPresent\t\t: %s\n", BD_TEXT(bd, PS2A));
453260c6 MG	215	}
453260c6 MG	216
7f88d2b6 MG	217	void handle_boarddescription(unsigned char *fw, int len, int patch)
7f88d2b6 MG	218	{
0a465cc0 MG	219	struct file_entry *fent;
0a465cc0 MG	220	unsigned char *pos;
453260c6	221
0a465cc0 MG	222	for (fent = get_next_file(fw, len); fent != NULL; fent = get_next_file(NULL, 0)) {
	223	if (!strcmp(fent->name, "pdata"))
	224	break;
	225	}
453260c6	226
0a465cc0 MG	227	if (fent == NULL) {
	228	fprintf(stderr, "pdata file not found, aborting!\n");
	229	exit(1);
	230	}
453260c6	231
453260c6	232
0a465cc0 MG	233	pos = fent->start;
	234	/* MAGIC? */
	235	if (((unsigned int)pos) != 0x00002802) {
	236	fprintf(stderr, "pdata magic does not match, aborting!\n");
	237	exit(1);
	238	}
453260c6	239
0a465cc0	240	pos += 26;
453260c6	241
0a465cc0 MG	242	/* MAGIC2? */
	243	if (((unsigned int)pos) != 0x00500101) {
	244	fprintf(stderr, "pdata magic2 does not match, aborting!\n");
	245	exit(1);
	246	}
	247
	248	if (patch) {
	249	/* Enable ATX and relay power switching */
	250	BD_SET(pos, PWRATX);
	251	BD_SET(pos, PWRRELAY);
453260c6	252	}
0a465cc0 MG	253
	254	printf("0x%08x: BOARD_DESCRIPTION: ", fent->start - fw);
	255	print_boarddescription(pos);
7f88d2b6 MG	256	}
7f88d2b6 MG	257
7ac4bfad MG	258	void syntax(char *name)
	259	{
	260	fprintf(stderr,"Syntax: %s parameters firmware.bin\n", name);
	261	fprintf(stderr,"parameters as follows:\n");
	262	fprintf(stderr,"\t-d\t\tdisplay all properties of the image\n");
	263	fprintf(stderr,"\t-u\t\tupdate checksum of the image\n");
453260c6	264	fprintf(stderr,"\t-b\t\tmodify BOARD_DESCRIPTION for more power-switch options\n");
14ff7444	265	fprintf(stderr,"\t-e\t\textract files in firmware\n");
7ac4bfad MG	266	fprintf(stderr,"\t-t property\tset 'property' to true\n");
	267	fprintf(stderr,"\t-f property\tset 'property' to false\n");
	268	fprintf(stderr,"\t-w property\tallow read-write access to 'property'\n");
	269	fprintf(stderr,"\t-r property\tallow read-only access to 'property'\n");
	270	exit(1);
	271	}
	272
	273	void add_action(int opt, char optarg, struct propaction *paction) {
	274	struct propaction pos = paction;
	275	struct propaction *prev = NULL;
	276
	277	while (pos != NULL) {
	278	if (!strcmp(pos->property, optarg))
	279	break;
	280	prev = pos;
	281	pos = pos->next;
	282	}
	283
	284	if (pos == NULL) {
	285	pos = malloc(sizeof(struct propaction));
	286	if (pos == NULL) {
	287	perror("malloc");
	288	exit(1);
	289	}
	290	bzero(pos, sizeof(struct propaction));
	291	pos->property = optarg;
	292
	293	if (prev == NULL) {
	294	*paction = pos;
	295	} else {
	296	prev->next = pos;
	297	}
	298	}
	299
	300	switch(opt) {
	301	case 't':
	302	if (pos->action & PROP_ACTION_FALSE) {
	303	fprintf(stderr,"inconsistent requests for %s\n",pos->property);
	304	exit(1);
	305	}
	306	pos->action \|= PROP_ACTION_TRUE;
	307	break;
	308	case 'f':
	309	if (pos->action & PROP_ACTION_TRUE) {
	310	fprintf(stderr,"inconsistent requests for %s\n",pos->property);
	311	exit(1);
	312	}
	313	pos->action \|= PROP_ACTION_FALSE;
	314	break;
	315	case 'w':
	316	if (pos->action & PROP_ACTION_RO) {
	317	fprintf(stderr,"inconsistent requests for %s\n",pos->property);
	318	exit(1);
	319	}
	320	pos->action \|= PROP_ACTION_RW;
	321	break;
	322	case 'r':
	323	if (pos->action & PROP_ACTION_RW) {
	324	fprintf(stderr,"inconsistent requests for %s\n",pos->property);
	325	exit(1);
	326	}
	327	pos->action \|= PROP_ACTION_RO;
	328	break;
	329	}
330	}
331
332	int check_crc(unsigned char *fw, int len)
333	{
334	int ret;
335	unsigned int crc, oldcrc;
336
337	ret = rsb_crc2(fw, len, 0x55335053, &crc);
338	oldcrc = (unsigned int)((unsigned int)(fw + len - 4));
339
340	printf("Checksum: 0x%08x (%s), should be: 0x%08x\n",
341	crc,
342	(ret ? "NOT OK" : "OK"),
343	oldcrc);
344
345	return ret;
346	}
347
0a465cc0 MG	348	int check_image(unsigned char *fw, int len)
	349	{
	350	struct file_entry *fent;
	351	char *last_name = NULL;
	352
	353	fent = get_next_file(fw, len);
	354	while (fent != NULL) {
	355	last_name = fent->name;
	356	fent = get_next_file(NULL, 0);
	357	}
	358
	359	if (strcmp(last_name, "pdata")) {
	360	return 1;
	361	}
	362
	363	return 0;
	364	}
	365
972ac24b MG	366	int main(int argc, char **argv)
	367	{
	368	struct stat statbuf;
7ac4bfad	369	char *file = NULL;
972ac24b MG	370	unsigned char *fw;
	371	int fd;
	372	int remaining;
	373	int ret;
7ac4bfad MG	374	int opt;
	375	unsigned int crc;
	376	struct propaction *paction = NULL;
	377	int showall = 0;
	378	int update_crc = 0;
453260c6	379	int patch_bd = 0;
7ac4bfad	380	int patch_fw = 0;
14ff7444	381	int extract = 0;
972ac24b	382
7ac4bfad MG	383	if (argc < 2)
	384	syntax(argv[0]);
	385
14ff7444	386	while ((opt = getopt(argc, argv, "dubet:f:w:r:")) != -1) {
7ac4bfad MG	387	switch(opt) {
	388	case 'd':
	389	showall = 1;
	390	break;
	391	case 'u':
	392	update_crc = 1;
	393	break;
453260c6 MG	394	case 'b':
	395	patch_bd = 1;
	396	break;
14ff7444 MG	397	case 'e':
	398	extract = 1;
	399	break;
7ac4bfad MG	400	case 't':
	401	case 'f':
	402	case 'w':
	403	case 'r':
	404	patch_fw = 1;
	405	add_action(opt, optarg, &paction);
	406	break;
	407	default:
	408	syntax(argv[0]);
	409	}
972ac24b MG	410	}
972ac24b MG	411
7ac4bfad MG	412	if (argc > optind) {
	413	file = argv[optind];
	414	} else {
	415	syntax(argv[0]);
	416	}
	417
	418	if (stat(file, &statbuf) == -1) {
	419	fprintf(stderr,"%s: ", file);
972ac24b MG	420	perror("stat");
	421	exit(1);
	422	}
	423
7ac4bfad MG	424	if ((fd = open(file, O_RDONLY)) == -1) {
7ac4bfad MG	425	fprintf(stderr,"%s: ", file);
972ac24b MG	426	perror("open");
	427	exit(1);
	428	}
	429
	430	if ((fw = malloc(statbuf.st_size)) == NULL) {
	431	perror("malloc");
	432	exit(1);
	433	}
	434
	435	bzero(fw, statbuf.st_size);
	436
	437	remaining = statbuf.st_size;
	438
	439	while(remaining) {
	440	if ((ret = read(fd, fw + (statbuf.st_size - remaining), remaining)) == -1) {
	441	perror("read");
	442	exit(1);
	443	}
	444	remaining -= ret;
	445	}
7ac4bfad	446	close(fd);
972ac24b	447
7ac4bfad MG	448	ret = check_crc(fw, statbuf.st_size);
	449	if ((ret != 0) && (!update_crc)) {
	450	fprintf(stderr,"Checksum incorrect, aborting...\n");
9eb4b3c6	451	exit(1);
7ac4bfad	452	}
972ac24b	453
0a465cc0 MG	454	if (check_image(fw, statbuf.st_size-4) != 0) {
	455	fprintf(stderr, "corrupt firmware image found (pdata is not last entry), aborting!\n");
	456	exit(1);
	457	}
	458
7ac4bfad MG	459	if (patch_fw) {
	460	struct propaction *cpaction = paction;
	461
	462	change_properties(fw, statbuf.st_size, paction);
	463
	464	printf("\nProperty change results:\n");
	465	while(cpaction != NULL) {
	466	printf("%s: ", cpaction->property);
	467
	468	if (cpaction->status == PROP_STATUS_NOTFOUND)
	469	printf("NOTFOUND ");
	470	if (cpaction->status & PROP_STATUS_SUCCESS)
	471	printf("SUCCESS ");
	472	if (cpaction->status & PROP_STATUS_SAMEVAL)
	473	printf("SAMEVAL ");
	474	if (cpaction->status & PROP_STATUS_WRONGTYPE)
	475	printf("WRONGTYPE ");
	476	if (cpaction->status & PROP_STATUS_WRONGRIGHTS)
	477	printf("WRONGRIGHTS ");
	478	printf("\n");
	479
	480	cpaction = cpaction->next;
7f88d2b6	481	}
7ac4bfad	482	printf("\n");
7f88d2b6	483	}
972ac24b	484
453260c6 MG	485	if (patch_bd) {
	486	handle_boarddescription(fw, statbuf.st_size -4, 1);
	487	}
	488
	489	if (showall) {
7ac4bfad	490	show_properties(fw, statbuf.st_size - 4);
453260c6	491	handle_boarddescription(fw, statbuf.st_size -4, 0);
14ff7444 MG	492	}
	493
	494	if (extract) {
1f1fa7b6	495	extract_files(fw, statbuf.st_size - 4);
453260c6	496	}
7ac4bfad	497
453260c6	498	if (update_crc \|\| patch_fw \|\| patch_bd) {
7ac4bfad MG	499	ret = rsb_crc2(fw, statbuf.st_size, 0x55335053, &crc);
	500	if (ret == 4) {
	501	((unsigned int)(fw + statbuf.st_size - 4)) = crc;
	502	}
	503
79234162 MG	504	if (check_image(fw, statbuf.st_size-4) != 0) {
	505	fprintf(stderr, "corrupt firmware image found (pdata is not last entry), aborting!\n");
	506	exit(1);
	507	}
	508
7ac4bfad MG	509	if (check_crc(fw, statbuf.st_size) == 0) {
	510	char *newfile;
	511
	512	newfile = malloc(strlen(file) + strlen(".patched") + 1);
	513	if (newfile == NULL) {
	514	perror("malloc");
	515	exit(1);
	516	}
	517	strcpy(newfile, file);
	518	strcat(newfile, ".patched");
	519
	520	printf("Writing %s\n", newfile);
	521	if ((fd = open(newfile, O_WRONLY\|O_CREAT, 0644)) == -1) {
	522	fprintf(stderr,"%s: ", file);
	523	perror("open");
	524	exit(1);
	525	}
	526
	527	remaining = statbuf.st_size;
	528
	529	while(remaining) {
	530	if ((ret = write(fd, fw + (statbuf.st_size - remaining), remaining)) == -1) {
	531	perror("write");
	532	exit(1);
	533	}
	534	remaining -= ret;
	535	}
	536	close(fd);
	537	} else {
	538	fprintf(stderr,"Can't set correct checksum, aborting...\n");
	539	}
	540	}
	541
972ac24b MG	542	exit(0);
972ac24b MG	543	}