summaryrefslogtreecommitdiffstats
path: root/sw/programmer/linux/src/flash.c
blob: 48189dae7ddfa5a1babe6fe1d5505710cd72dc87 (plain)
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
#include "flash.h"

static int flash_serial_fd = -1;

int flash_open(const char *path, unsigned long baudrate)
{
	flash_serial_fd = serial_open(path, baudrate);

	if (flash_serial_fd < 0)
		return -1;

	return 0;
}

int flash_write(const char *romfile, void (*log)(const char *))
{
	int romfd;
	ssize_t written;

	struct stat romst;
	struct flash_blk head;

	uint8_t *buffer = malloc(FLASH_BLOCK_SIZE);

	romfd = open(romfile, O_RDONLY);

	if (fstat(romfd, &romst) != 0)
		goto exit_romfd;

	while ((head.size = read(romfd, buffer, FLASH_BLOCK_SIZE))) {
		head.addr = (uint16_t) lseek(romfd, 0, SEEK_CUR);

		written = write(flash_serial_fd, &head, sizeof(struct flash_blk));
		if (written < 0) {
			// error
		}

		written = write(flash_serial_fd, buffer, head.size);		
		if (written < 0) {
			// error
		}

		char logbuf[64];
		sprintf(logbuf, "[@] Written %d bytes at address %d\n", head.size, head.addr);
		log(logbuf);
	}

exit_romfd:
	close(romfd);

	free(buffer);
	return 0;
}

void flash_close(void)
{
	if (flash_serial_fd >= 0) 
		close(flash_serial_fd);
}