- Change NANDFlash again for Versatile support. Now the LLB and OS Loader are created...
[reactos.git] / reactos / tools / nandflash / main.c
1 /*
2 * PROJECT: OMAP3 NAND Flashing Utility
3 * LICENSE: BSD - See COPYING.ARM in the top level directory
4 * FILE: tools/nandflash/main.c
5 * PURPOSE: Flashes OmapLDR, FreeLDR and a Root FS into a NAND image
6 * PROGRAMMERS: ReactOS Portable Systems Group
7 */
8
9 /* INCLUDES *******************************************************************/
10
11 #include "nandflash.h"
12
13 /* GLOBALS ********************************************************************/
14
15 /* File Names */
16 PCHAR NandImageName = "reactos.bin";
17 PCHAR LlbImageName = "./output-arm/boot/armllb/armllb.bin";
18 PCHAR BootLdrImageName = "./output-arm/boot/freeldr/freeldr/freeldr.sys";
19 PCHAR FsImageName = "ramdisk.img";
20 PCHAR RamImageName = "ramdisk.bin";
21
22 /* NAND On-Disk Memory Map */
23 ULONG LlbStart = 0x00000000, LlbEnd = 0x00010000; // 64 KB
24 ULONG BootLdrStart = 0x00010000, BootLdrEnd = 0x00090000; // 512 KB
25 ULONG FsStart = 0x00090000, FsEnd = 0x10000000; // 255 MB
26
27 /* Should we write OOB data? */
28 ULONG NeedsOob = TRUE;
29
30 /* FUNCTIONS ******************************************************************/
31
32 INT
33 NTAPI
34 CreateFlashFile(VOID)
35 {
36 INT FileDescriptor, i;
37 CHAR Buffer[NAND_PAGE_SIZE + NAND_OOB_SIZE];
38
39 /* Try open NAND image */
40 FileDescriptor = open(NandImageName, O_RDWR);
41 if (FileDescriptor)
42 {
43 /* Create NAND image */
44 FileDescriptor = open(NandImageName, O_RDWR | O_CREAT);
45 if (FileDescriptor) return FileDescriptor;
46
47 /* Create zero buffer */
48 memset(Buffer, 0xff, sizeof(Buffer));
49
50 /* Write zero buffer */
51 for (i = 0; i < NAND_PAGES; i++) write(FileDescriptor, Buffer, sizeof(Buffer));
52 }
53
54 /* Return NAND descriptor */
55 return FileDescriptor;
56 }
57
58 VOID
59 NTAPI
60 WriteToFlash(IN INT NandImageFile,
61 IN INT ImageFile,
62 IN ULONG ImageStart,
63 IN ULONG ImageEnd)
64 {
65 CHAR Data[NAND_PAGE_SIZE], Oob[NAND_OOB_SIZE];
66 ULONG StartPage, EndPage, i, OobSize = 0;
67 BOOLEAN KeepGoing = TRUE;
68
69 /* Offset to NAND Page convert */
70 StartPage = ImageStart / NAND_PAGE_SIZE;
71 EndPage = ImageEnd / NAND_PAGE_SIZE;
72
73 /* Jump to NAND offset */
74 if (NeedsOob) OobSize = NAND_OOB_SIZE;
75 lseek(NandImageFile, StartPage * (NAND_PAGE_SIZE + OobSize), SEEK_SET);
76
77 /* Set input image offset */
78 lseek(ImageFile, 0, SEEK_SET);
79
80 /* Create zero buffer */
81 memset(Data, 0xff, NAND_PAGE_SIZE);
82 memset(Oob, 0xff, NAND_OOB_SIZE);
83
84 /* Parse NAND Pages */
85 for (i = StartPage; i < EndPage; i++)
86 {
87 /* Read NAND page from input image */
88 if (read(ImageFile, Data, NAND_PAGE_SIZE) < NAND_PAGE_SIZE)
89 {
90 /* Do last write and quit after */
91 KeepGoing = FALSE;
92 }
93
94 /* Write OOB and NAND Data */
95 write(NandImageFile, Data, NAND_PAGE_SIZE);
96 if (NeedsOob) write(NandImageFile, Oob, NAND_OOB_SIZE);
97
98 /* Next page if data continues */
99 if (!KeepGoing) break;
100 }
101 }
102
103 VOID
104 NTAPI
105 WriteLlb(IN INT NandImageFile)
106 {
107 INT FileDescriptor;
108
109 /* Open LLB and write it */
110 FileDescriptor = open(LlbImageName, O_RDWR);
111 WriteToFlash(NandImageFile, FileDescriptor, LlbStart, LlbEnd);
112 close(FileDescriptor);
113 }
114
115 VOID
116 NTAPI
117 WriteBootLdr(IN INT NandImageFile)
118 {
119 INT FileDescriptor;
120
121 /* Open FreeLDR and write it */
122 FileDescriptor = open(BootLdrImageName, O_RDWR);
123 WriteToFlash(NandImageFile, FileDescriptor, BootLdrStart, BootLdrEnd);
124 close(FileDescriptor);
125 }
126
127 VOID
128 NTAPI
129 WriteFileSystem(IN INT NandImageFile)
130 {
131 INT FileDescriptor;
132
133 /* Open FS image and write it */
134 FileDescriptor = open(FsImageName, O_RDWR);
135 WriteToFlash(NandImageFile, FileDescriptor, FsStart, FsEnd);
136 close(FileDescriptor);
137 }
138
139 VOID
140 NTAPI
141 WriteRamDisk(IN INT RamDiskFile)
142 {
143 INT FileDescriptor;
144
145 /* Open FS image and write it 16MB later */
146 FileDescriptor = open(FsImageName, O_RDWR);
147 WriteToFlash(RamDiskFile, FileDescriptor, 16 * 1024 * 1024, (32 + 16) * 1024 * 1024);
148 close(FileDescriptor);
149 }
150
151 int
152 main(ULONG argc,
153 char **argv)
154 {
155 INT NandImageFile, RamDiskFile;
156
157 /* Flat NAND, no OOB */
158 if (argc == 2) NeedsOob = FALSE;
159
160 /* Open or create NAND Image File */
161 NandImageFile = CreateFlashFile();
162 if (!NandImageFile) exit(-1);
163
164 /* Write components */
165 WriteLlb(NandImageFile);
166 WriteBootLdr(NandImageFile);
167 if (NeedsOob)
168 {
169 /* Write the ramdisk normaly */
170 WriteFileSystem(NandImageFile);
171 }
172 else
173 {
174 /* Open a new file for the ramdisk */
175 RamDiskFile = open(RamImageName, O_RDWR | O_CREAT);
176 if (!RamDiskFile) exit(-1);
177
178 /* Write it */
179 WriteRamDisk(RamDiskFile);
180
181 /* Close */
182 close(RamDiskFile);
183 }
184
185
186 /* Close and return */
187 close(NandImageFile);
188 return 0;
189 }
190
191 /* EOF */