iusutils/mkbootsd.c at master · Ithamar/iusutils · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <stdlib.h>
#include "sdheader.h"

#define ARRAY_SIZE(x) (sizeof(x)/sizeof(x[0]))

static int valid_bootscr(unsigned char * addr)
{
	unsigned char header[] = {
		0x0a, 0x2d, 0x2d, 0x5b, 0x5b, 0x75, 0x62, 0x6f,
		0x6f, 0x74, 0x2e, 0x73, 0x63, 0x72, 0x69, 0x70,
		0x74, 0x0a,};
	unsigned char *p = (unsigned char *)addr;

	unsigned int i;
	for(i = 0; i < ARRAY_SIZE(header); i++)
	  if(*p++ ^ header[i])
		return 0;

	for(i = 0; i < 0x1000; i++, p++)
	  if(*p == 0xa
		 && *(p + 1) == 0x5d
		 && *(p + 2) == 0x5d)
		return 1;		// valid script
	return 0;
}

int main(int argc, char *argv[])
{
	char buf[4096];
	int fdi, fdo, ret;

	if(argc < 3)
	{
		printf("Usage:\nmkbootsd /dev/dev_node script [iusfile]\n");
		exit(0);
	}

confirm:
	printf("All data on SD card will be lost, type 'confirm' to continue: ");
	ret = scanf("%s", buf);
	if(strcmp(buf, "confirm"))
	  goto confirm;

	fdo = open(argv[1], O_WRONLY);
	if(fdo < 0)
	{
		printf("Can not open %s\n", argv[1]);
		exit(1);
	}

	/* Write header */
	ret = write(fdo, sd_sector_0, 512);
	if(ret != 512)
	{
		printf("Write SD header failed.\n");
		goto __exit__;
	}

	/* Write Script */
	fdi = open(argv[2], O_RDONLY);
	if(fdo < 0)
	{
		printf("Can not open %s\n", argv[2]);
		exit(1);
	}

	memset(buf, 0, 4096);
	ret = read(fdi, buf, 4096);
	close(fdi);

	if(!valid_bootscr((unsigned char *)buf))
	{
		printf("Boot script is invalid.\n");
		goto __exit__;
	}

	ret = lseek(fdo, 0x80000, SEEK_SET);
	if(ret != 0x80000)
	{
		printf("Seek script location failed.\n");
		goto __exit__;
	}

	ret = write(fdo, buf, 4096);
	if(ret != 4096)
	{
		printf("Write script failed.\n");
		goto __exit__;
	}

	/* Write .ius */
	if (argc == 4) {
		fdi = open(argv[3], O_RDONLY);
		if (fdo < 0) {
			printf("Can not open %s\n", argv[3]);
			exit(1);
		}

		ret = lseek(fdo, 0x100000, SEEK_SET);
		if (ret != 0x100000) {
			printf("Seek ius location failed.\n");
			goto __exit__;
		}

		while((ret=read(fdi, buf, 4096)) > 0) {
			if (write(fdo, buf, ret) < 0) {
				printf("Write ius failed\n");
				break;
			}
		}

		if (ret < 0)
			printf("Read ius failed\n");

		close(fdi);
	}

	printf("Boot SD created successfully!\n");

__exit__:
	close(fdo);
	return 0;
}