Revert "Let prom_claim_chunk do the work for tftp images to support larger images."
[yaboot.git] / second / fs_of.c
1 /*
2  *  fs_of.c - an implementation for OpenFirmware supported filesystems
3  *
4  *  Copyright (C) 2001, 2002 Ethan Benson
5  *
6  *  Copyright (C) 1999 Benjamin Herrenschmidt
7  *
8  *  This program is free software; you can redistribute it and/or modify
9  *  it under the terms of the GNU General Public License as published by
10  *  the Free Software Foundation; either version 2 of the License, or
11  *  (at your option) any later version.
12  *
13  *  This program is distributed in the hope that it will be useful,
14  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
15  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  *  GNU General Public License for more details.
17  *
18  *  You should have received a copy of the GNU General Public License
19  *  along with this program; if not, write to the Free Software
20  *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
21  */
22
23 /*
24  * BrokenFirmware cannot "read" from the network. We use tftp "load"
25  * method for network boot for now, we may provide our own NFS
26  * implementation in a later version. That means that we allocate a
27  * huge block of memory for the entire file before loading it. We use
28  * the location where the kernel puts RTAS, it's not used by the
29  * bootloader and if freed when the kernel is booted.  This will have
30  * to be changed if we plan to instanciate RTAS in the bootloader
31  * itself
32  */
33
34 #include "ctype.h"
35 #include "types.h"
36 #include "stddef.h"
37 #include "stdlib.h"
38 #include "file.h"
39 #include "prom.h"
40 #include "string.h"
41 #include "partition.h"
42 #include "fdisk-part.h"
43 #include "fs.h"
44 #include "errors.h"
45 #include "debug.h"
46
47 #define LOAD_BUFFER_POS         0x1000000
48 #define LOAD_BUFFER_SIZE        0x1000000
49
50 static int of_open(struct boot_file_t* file, const char* dev_name,
51                    struct partition_t* part, const char* file_name);
52 static int of_read(struct boot_file_t* file, unsigned int size, void* buffer);
53 static int of_seek(struct boot_file_t* file, unsigned int newpos);
54 static int of_close(struct boot_file_t* file);
55
56
57 static int of_net_open(struct boot_file_t* file, const char* dev_name,
58                        struct partition_t* part, const char* file_name);
59 static int of_net_read(struct boot_file_t* file, unsigned int size, void* buffer);
60 static int of_net_seek(struct boot_file_t* file, unsigned int newpos);
61
62
63 struct fs_t of_filesystem =
64 {
65      "built-in",
66      of_open,
67      of_read,
68      of_seek,
69      of_close
70 };
71
72 struct fs_t of_net_filesystem =
73 {
74      "built-in network",
75      of_net_open,
76      of_net_read,
77      of_net_seek,
78      of_close
79 };
80
81 static int
82 of_open(struct boot_file_t* file, const char* dev_name,
83         struct partition_t* part, const char* file_name)
84 {
85      static char        buffer[1024];
86      char               *filename;
87      char               *p;
88
89      DEBUG_ENTER;
90      DEBUG_OPEN;
91
92      strncpy(buffer, dev_name, 768);
93      strcat(buffer, ":");
94      if (part) {
95           if (part->sys_ind == LINUX_RAID) {
96                DEBUG_F("skipping because partition is marked LINUX_RAID\n");
97                DEBUG_LEAVE(FILE_ERR_BAD_FSYS);
98                return FILE_ERR_BAD_FSYS;
99           }
100           char pn[3];
101           sprintf(pn, "%02d", part->part_number);
102           strcat(buffer, pn);
103      }
104      if (file_name && strlen(file_name)) {
105           if (part)
106                strcat(buffer, ",");
107           filename = strdup(file_name);
108           for (p = filename; *p; p++)
109                if (*p == '/')
110                     *p = '\\';
111           strcat(buffer, filename);
112           free(filename);
113      }
114
115      DEBUG_F("opening: \"%s\"\n", buffer);
116
117      file->of_device = prom_open(buffer);
118
119      DEBUG_F("file->of_device = %p\n", file->of_device);
120
121      file->pos = 0;
122      file->buffer = NULL;
123      if ((file->of_device == PROM_INVALID_HANDLE) || (file->of_device == 0))
124      {
125           DEBUG_LEAVE(FILE_ERR_BAD_FSYS);
126           return FILE_ERR_BAD_FSYS;
127      }
128
129      DEBUG_LEAVE(FILE_ERR_OK);
130      return FILE_ERR_OK;
131 }
132
133 static int
134 of_net_open(struct boot_file_t* file, const char* dev_name,
135             struct partition_t* part, const char* file_name)
136 {
137      static char        buffer[1024];
138      char               *filename;
139      char               *p;
140
141      DEBUG_ENTER;
142      DEBUG_OPEN;
143
144      strncpy(buffer, dev_name, 768);
145      if (file_name && strlen(file_name)) {
146           strcat(buffer, ",");
147           filename = strdup(file_name);
148           for (p = filename; *p; p++)
149                if (*p == '/')
150                     *p = '\\';
151           strcat(buffer, filename);
152           free(filename);
153      }
154
155      DEBUG_F("Opening: \"%s\"\n", buffer);
156
157      file->of_device = prom_open(buffer);
158
159      DEBUG_F("file->of_device = %p\n", file->of_device);
160
161      file->pos = 0;
162      if ((file->of_device == PROM_INVALID_HANDLE) || (file->of_device == 0))
163      {
164           DEBUG_LEAVE(FILE_ERR_BAD_FSYS);
165           return FILE_ERR_BAD_FSYS;
166      }
167
168
169      file->buffer = prom_claim_chunk((void *)LOAD_BUFFER_POS,
170                                      LOAD_BUFFER_SIZE, 0);
171      if (file->buffer == (void *)-1) {
172           prom_printf("Can't claim memory for TFTP download\n");
173           prom_close(file->of_device);
174           DEBUG_LEAVE(FILE_IOERR);
175           return FILE_IOERR;
176      }
177      memset(file->buffer, 0, LOAD_BUFFER_SIZE);
178
179      DEBUG_F("TFP...\n");
180
181      file->len = prom_loadmethod(file->of_device, file->buffer);
182
183      DEBUG_F("result: %Ld\n", file->len);
184
185      DEBUG_LEAVE(FILE_ERR_OK);
186      return FILE_ERR_OK;
187 }
188
189 static int
190 of_read(struct boot_file_t* file, unsigned int size, void* buffer)
191 {
192      unsigned int count;
193
194      count = prom_read(file->of_device, buffer, size);
195      file->pos += count;
196      return count;
197 }
198
199 static int
200 of_net_read(struct boot_file_t* file, unsigned int size, void* buffer)
201 {
202      unsigned int count, av;
203
204      av = file->len - file->pos;
205      count = size > av ? av : size;
206      memcpy(buffer, file->buffer + file->pos, count);
207      file->pos += count;
208      return count;
209 }
210
211 static int
212 of_seek(struct boot_file_t* file, unsigned int newpos)
213 {
214      if (prom_seek(file->of_device, newpos)) {
215           file->pos = newpos;
216           return FILE_ERR_OK;
217      }
218
219      return FILE_CANT_SEEK;
220 }
221
222 static int
223 of_net_seek(struct boot_file_t* file, unsigned int newpos)
224 {
225      file->pos = (newpos > file->len) ? file->len : newpos;
226      return FILE_ERR_OK;
227 }
228
229 static int
230 of_close(struct boot_file_t* file)
231 {
232
233      DEBUG_ENTER;
234      DEBUG_F("<@%p>\n", file->of_device);
235
236      if (file->buffer) {
237           prom_release(file->buffer, LOAD_BUFFER_SIZE);
238      }
239      prom_close(file->of_device);
240      DEBUG_F("of_close called\n");
241
242      DEBUG_LEAVE(0);
243      return 0;
244 }
245
246 /*
247  * Local variables:
248  * c-file-style: "k&r"
249  * c-basic-offset: 5
250  * End:
251  */