微處理器 - Allwinner V3S (Lichee Pi Zero) - mksunxi.c


?
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
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
  
#define __ALIGN_MASK(x, mask) (((x) + (mask)) & ~(mask))
#define ALIGN(x, a) __ALIGN_MASK((x), (typeof(x))(a)-1)
  
#if 0
static inline uint32_t __swab32(uint32_t x)
{
  return ( (x<<24) | (x>>24) | \
    ((x & (uint32_t)0x0000ff00UL)<<8) | \
    ((x & (uint32_t)0x00ff0000UL)>>8) );
}
#define cpu_to_le32(x) (__swab32((uint32_t)(x)))
#define le32_to_cpu(x) (__swab32((uint32_t)(x)))
#else
#define cpu_to_le32(x) (x)
#define le32_to_cpu(x) (x)
#endif
  
struct boot_head_t {
    uint32_t instruction;
    uint8_t magic[8];
    uint32_t checksum;
    uint32_t length;
    uint8_t spl_signature[4];
    uint32_t fel_script_address;
    uint32_t fel_uenv_length;
    uint32_t dt_name_offset;
    uint32_t reserved1;
    uint32_t boot_media;
    uint32_t string_pool[13];
};
  
int main(int argc, char *argv[])
{
    struct boot_head_t *h;
    FILE *fp;
    char *buffer;
    int buflen, filelen;
    uint32_t *p;
    uint32_t sum;
    int i, l, loop;
  
    if (argc != 2) {
        printf("Usage: mksunxi <bootloader>\n");
        return -1;
    }
  
    fp = fopen(argv[1], "r+b");
    if (fp == NULL) {
        printf("Open bootloader error\n");
        return -1;
    }
    fseek(fp, 0L, SEEK_END);
    filelen = ftell(fp);
    fseek(fp, 0L, SEEK_SET);
  
    if (filelen <= sizeof(struct boot_head_t)) {
        fclose(fp);
        printf("The size of bootloader too small\n");
        return -1;
    }
  
    buflen = ALIGN(filelen, 512);
    buffer = malloc(buflen);
    memset(buffer, 0, buflen);
    if (fread(buffer, 1, filelen, fp) != filelen) {
        printf("Can't read bootloader\n");
        free(buffer);
        fclose(fp);
        return -1;
    }
  
    h = (struct boot_head_t *)buffer;
    p = (uint32_t *)h;
    l = le32_to_cpu(h->length);
    l = ALIGN(l, 512);
    h->length = cpu_to_le32(l);
    h->checksum = cpu_to_le32(0x5F0A6C39);
    loop = l >> 2;
    for (i = 0, sum = 0; i < loop; i++) sum += le32_to_cpu(p[i]);
    h->checksum = cpu_to_le32(sum);
  
    fseek(fp, 0L, SEEK_SET);
    if (fwrite(buffer, 1, buflen, fp) != buflen) {
        printf("Write bootloader error\n");
        free(buffer);
        fclose(fp);
        return -1;
    }
  
    fclose(fp);
    printf("The bootloader head has been fixed, spl size is %d bytes.\r\n", l);
    return 0;
}