Statistics
| Branch: | Revision:

root / block-vvfat.c @ f97572e5

History | View | Annotate | Download (79.3 kB)

1 a046433a bellard
/* vim:set shiftwidth=4 ts=8: */
2 de167e41 bellard
/*
3 de167e41 bellard
 * QEMU Block driver for virtual VFAT (shadows a local directory)
4 5fafdf24 ths
 *
5 a046433a bellard
 * Copyright (c) 2004,2005 Johannes E. Schindelin
6 5fafdf24 ths
 *
7 de167e41 bellard
 * Permission is hereby granted, free of charge, to any person obtaining a copy
8 de167e41 bellard
 * of this software and associated documentation files (the "Software"), to deal
9 de167e41 bellard
 * in the Software without restriction, including without limitation the rights
10 de167e41 bellard
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
11 de167e41 bellard
 * copies of the Software, and to permit persons to whom the Software is
12 de167e41 bellard
 * furnished to do so, subject to the following conditions:
13 de167e41 bellard
 *
14 de167e41 bellard
 * The above copyright notice and this permission notice shall be included in
15 de167e41 bellard
 * all copies or substantial portions of the Software.
16 de167e41 bellard
 *
17 de167e41 bellard
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18 de167e41 bellard
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19 de167e41 bellard
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
20 de167e41 bellard
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21 de167e41 bellard
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
22 de167e41 bellard
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
23 de167e41 bellard
 * THE SOFTWARE.
24 de167e41 bellard
 */
25 de167e41 bellard
#include <sys/stat.h>
26 de167e41 bellard
#include <dirent.h>
27 de167e41 bellard
#include <assert.h>
28 faf07963 pbrook
#include "qemu-common.h"
29 de167e41 bellard
#include "block_int.h"
30 de167e41 bellard
31 a046433a bellard
#ifndef S_IWGRP
32 a046433a bellard
#define S_IWGRP 0
33 a046433a bellard
#endif
34 a046433a bellard
#ifndef S_IWOTH
35 a046433a bellard
#define S_IWOTH 0
36 a046433a bellard
#endif
37 a046433a bellard
38 a046433a bellard
/* TODO: add ":bootsector=blabla.img:" */
39 a046433a bellard
/* LATER TODO: add automatic boot sector generation from
40 a046433a bellard
    BOOTEASY.ASM and Ranish Partition Manager
41 5fafdf24 ths
    Note that DOS assumes the system files to be the first files in the
42 a046433a bellard
    file system (test if the boot sector still relies on that fact)! */
43 a046433a bellard
/* MAYBE TODO: write block-visofs.c */
44 a046433a bellard
/* TODO: call try_commit() only after a timeout */
45 a046433a bellard
46 a046433a bellard
/* #define DEBUG */
47 a046433a bellard
48 a046433a bellard
#ifdef DEBUG
49 a046433a bellard
50 a046433a bellard
#define DLOG(a) a
51 a046433a bellard
52 a046433a bellard
#undef stderr
53 a046433a bellard
#define stderr STDERR
54 a046433a bellard
FILE* stderr = NULL;
55 de167e41 bellard
56 3f47aa8c blueswir1
static void checkpoint(void);
57 de167e41 bellard
58 a046433a bellard
#ifdef __MINGW32__
59 a046433a bellard
void nonono(const char* file, int line, const char* msg) {
60 a046433a bellard
    fprintf(stderr, "Nonono! %s:%d %s\n", file, line, msg);
61 a046433a bellard
    exit(-5);
62 a046433a bellard
}
63 a046433a bellard
#undef assert
64 6bcb76c3 bellard
#define assert(a) do {if (!(a)) nonono(__FILE__, __LINE__, #a);}while(0)
65 a046433a bellard
#endif
66 a046433a bellard
67 a046433a bellard
#else
68 a046433a bellard
69 a046433a bellard
#define DLOG(a)
70 a046433a bellard
71 a046433a bellard
#endif
72 de167e41 bellard
73 de167e41 bellard
/* dynamic array functions */
74 de167e41 bellard
typedef struct array_t {
75 de167e41 bellard
    char* pointer;
76 de167e41 bellard
    unsigned int size,next,item_size;
77 de167e41 bellard
} array_t;
78 de167e41 bellard
79 de167e41 bellard
static inline void array_init(array_t* array,unsigned int item_size)
80 de167e41 bellard
{
81 de167e41 bellard
    array->pointer=0;
82 de167e41 bellard
    array->size=0;
83 de167e41 bellard
    array->next=0;
84 de167e41 bellard
    array->item_size=item_size;
85 de167e41 bellard
}
86 de167e41 bellard
87 de167e41 bellard
static inline void array_free(array_t* array)
88 de167e41 bellard
{
89 de167e41 bellard
    if(array->pointer)
90 de167e41 bellard
        free(array->pointer);
91 de167e41 bellard
    array->size=array->next=0;
92 de167e41 bellard
}
93 de167e41 bellard
94 a046433a bellard
/* does not automatically grow */
95 de167e41 bellard
static inline void* array_get(array_t* array,unsigned int index) {
96 a046433a bellard
    assert(index < array->next);
97 a046433a bellard
    return array->pointer + index * array->item_size;
98 a046433a bellard
}
99 a046433a bellard
100 a046433a bellard
static inline int array_ensure_allocated(array_t* array, int index)
101 a046433a bellard
{
102 a046433a bellard
    if((index + 1) * array->item_size > array->size) {
103 a046433a bellard
        int new_size = (index + 32) * array->item_size;
104 2137b4cc ths
        array->pointer = qemu_realloc(array->pointer, new_size);
105 a046433a bellard
        if (!array->pointer)
106 a046433a bellard
            return -1;
107 a046433a bellard
        array->size = new_size;
108 a046433a bellard
        array->next = index + 1;
109 de167e41 bellard
    }
110 a046433a bellard
111 a046433a bellard
    return 0;
112 de167e41 bellard
}
113 de167e41 bellard
114 de167e41 bellard
static inline void* array_get_next(array_t* array) {
115 a046433a bellard
    unsigned int next = array->next;
116 a046433a bellard
    void* result;
117 a046433a bellard
118 a046433a bellard
    if (array_ensure_allocated(array, next) < 0)
119 a046433a bellard
        return NULL;
120 a046433a bellard
121 a046433a bellard
    array->next = next + 1;
122 a046433a bellard
    result = array_get(array, next);
123 a046433a bellard
124 de167e41 bellard
    return result;
125 de167e41 bellard
}
126 de167e41 bellard
127 de167e41 bellard
static inline void* array_insert(array_t* array,unsigned int index,unsigned int count) {
128 de167e41 bellard
    if((array->next+count)*array->item_size>array->size) {
129 de167e41 bellard
        int increment=count*array->item_size;
130 2137b4cc ths
        array->pointer=qemu_realloc(array->pointer,array->size+increment);
131 de167e41 bellard
        if(!array->pointer)
132 de167e41 bellard
            return 0;
133 de167e41 bellard
        array->size+=increment;
134 de167e41 bellard
    }
135 de167e41 bellard
    memmove(array->pointer+(index+count)*array->item_size,
136 de167e41 bellard
                array->pointer+index*array->item_size,
137 de167e41 bellard
                (array->next-index)*array->item_size);
138 de167e41 bellard
    array->next+=count;
139 de167e41 bellard
    return array->pointer+index*array->item_size;
140 de167e41 bellard
}
141 de167e41 bellard
142 de167e41 bellard
/* this performs a "roll", so that the element which was at index_from becomes
143 de167e41 bellard
 * index_to, but the order of all other elements is preserved. */
144 de167e41 bellard
static inline int array_roll(array_t* array,int index_to,int index_from,int count)
145 de167e41 bellard
{
146 de167e41 bellard
    char* buf;
147 de167e41 bellard
    char* from;
148 de167e41 bellard
    char* to;
149 de167e41 bellard
    int is;
150 de167e41 bellard
151 de167e41 bellard
    if(!array ||
152 de167e41 bellard
            index_to<0 || index_to>=array->next ||
153 de167e41 bellard
            index_from<0 || index_from>=array->next)
154 de167e41 bellard
        return -1;
155 3b46e624 ths
156 de167e41 bellard
    if(index_to==index_from)
157 de167e41 bellard
        return 0;
158 de167e41 bellard
159 de167e41 bellard
    is=array->item_size;
160 de167e41 bellard
    from=array->pointer+index_from*is;
161 de167e41 bellard
    to=array->pointer+index_to*is;
162 de167e41 bellard
    buf=malloc(is*count);
163 de167e41 bellard
    memcpy(buf,from,is*count);
164 de167e41 bellard
165 de167e41 bellard
    if(index_to<index_from)
166 de167e41 bellard
        memmove(to+is*count,to,from-to);
167 de167e41 bellard
    else
168 de167e41 bellard
        memmove(from,from+is*count,to-from);
169 3b46e624 ths
170 de167e41 bellard
    memcpy(to,buf,is*count);
171 de167e41 bellard
172 de167e41 bellard
    free(buf);
173 de167e41 bellard
174 de167e41 bellard
    return 0;
175 de167e41 bellard
}
176 de167e41 bellard
177 9596ebb7 pbrook
static inline int array_remove_slice(array_t* array,int index, int count)
178 de167e41 bellard
{
179 a046433a bellard
    assert(index >=0);
180 a046433a bellard
    assert(count > 0);
181 a046433a bellard
    assert(index + count <= array->next);
182 a046433a bellard
    if(array_roll(array,array->next-1,index,count))
183 de167e41 bellard
        return -1;
184 a046433a bellard
    array->next -= count;
185 de167e41 bellard
    return 0;
186 de167e41 bellard
}
187 de167e41 bellard
188 9596ebb7 pbrook
static int array_remove(array_t* array,int index)
189 a046433a bellard
{
190 a046433a bellard
    return array_remove_slice(array, index, 1);
191 a046433a bellard
}
192 a046433a bellard
193 a046433a bellard
/* return the index for a given member */
194 9596ebb7 pbrook
static int array_index(array_t* array, void* pointer)
195 a046433a bellard
{
196 a046433a bellard
    size_t offset = (char*)pointer - array->pointer;
197 a046433a bellard
    assert((offset % array->item_size) == 0);
198 a046433a bellard
    assert(offset/array->item_size < array->next);
199 a046433a bellard
    return offset/array->item_size;
200 a046433a bellard
}
201 a046433a bellard
202 de167e41 bellard
/* These structures are used to fake a disk and the VFAT filesystem.
203 de167e41 bellard
 * For this reason we need to use __attribute__((packed)). */
204 de167e41 bellard
205 de167e41 bellard
typedef struct bootsector_t {
206 de167e41 bellard
    uint8_t jump[3];
207 de167e41 bellard
    uint8_t name[8];
208 de167e41 bellard
    uint16_t sector_size;
209 de167e41 bellard
    uint8_t sectors_per_cluster;
210 de167e41 bellard
    uint16_t reserved_sectors;
211 de167e41 bellard
    uint8_t number_of_fats;
212 de167e41 bellard
    uint16_t root_entries;
213 a046433a bellard
    uint16_t total_sectors16;
214 de167e41 bellard
    uint8_t media_type;
215 de167e41 bellard
    uint16_t sectors_per_fat;
216 de167e41 bellard
    uint16_t sectors_per_track;
217 de167e41 bellard
    uint16_t number_of_heads;
218 de167e41 bellard
    uint32_t hidden_sectors;
219 de167e41 bellard
    uint32_t total_sectors;
220 de167e41 bellard
    union {
221 de167e41 bellard
        struct {
222 de167e41 bellard
            uint8_t drive_number;
223 de167e41 bellard
            uint8_t current_head;
224 de167e41 bellard
            uint8_t signature;
225 de167e41 bellard
            uint32_t id;
226 de167e41 bellard
            uint8_t volume_label[11];
227 de167e41 bellard
        } __attribute__((packed)) fat16;
228 de167e41 bellard
        struct {
229 de167e41 bellard
            uint32_t sectors_per_fat;
230 de167e41 bellard
            uint16_t flags;
231 de167e41 bellard
            uint8_t major,minor;
232 de167e41 bellard
            uint32_t first_cluster_of_root_directory;
233 de167e41 bellard
            uint16_t info_sector;
234 de167e41 bellard
            uint16_t backup_boot_sector;
235 de167e41 bellard
            uint16_t ignored;
236 de167e41 bellard
        } __attribute__((packed)) fat32;
237 de167e41 bellard
    } u;
238 de167e41 bellard
    uint8_t fat_type[8];
239 de167e41 bellard
    uint8_t ignored[0x1c0];
240 de167e41 bellard
    uint8_t magic[2];
241 de167e41 bellard
} __attribute__((packed)) bootsector_t;
242 de167e41 bellard
243 b570094d ths
typedef struct {
244 b570094d ths
    uint8_t head;
245 b570094d ths
    uint8_t sector;
246 b570094d ths
    uint8_t cylinder;
247 b570094d ths
} mbr_chs_t;
248 b570094d ths
249 de167e41 bellard
typedef struct partition_t {
250 de167e41 bellard
    uint8_t attributes; /* 0x80 = bootable */
251 b570094d ths
    mbr_chs_t start_CHS;
252 b570094d ths
    uint8_t   fs_type; /* 0x1 = FAT12, 0x6 = FAT16, 0xe = FAT16_LBA, 0xb = FAT32, 0xc = FAT32_LBA */
253 b570094d ths
    mbr_chs_t end_CHS;
254 de167e41 bellard
    uint32_t start_sector_long;
255 b570094d ths
    uint32_t length_sector_long;
256 de167e41 bellard
} __attribute__((packed)) partition_t;
257 de167e41 bellard
258 de167e41 bellard
typedef struct mbr_t {
259 b570094d ths
    uint8_t ignored[0x1b8];
260 b570094d ths
    uint32_t nt_id;
261 b570094d ths
    uint8_t ignored2[2];
262 de167e41 bellard
    partition_t partition[4];
263 de167e41 bellard
    uint8_t magic[2];
264 de167e41 bellard
} __attribute__((packed)) mbr_t;
265 de167e41 bellard
266 de167e41 bellard
typedef struct direntry_t {
267 de167e41 bellard
    uint8_t name[8];
268 de167e41 bellard
    uint8_t extension[3];
269 de167e41 bellard
    uint8_t attributes;
270 de167e41 bellard
    uint8_t reserved[2];
271 de167e41 bellard
    uint16_t ctime;
272 de167e41 bellard
    uint16_t cdate;
273 de167e41 bellard
    uint16_t adate;
274 de167e41 bellard
    uint16_t begin_hi;
275 de167e41 bellard
    uint16_t mtime;
276 de167e41 bellard
    uint16_t mdate;
277 de167e41 bellard
    uint16_t begin;
278 de167e41 bellard
    uint32_t size;
279 de167e41 bellard
} __attribute__((packed)) direntry_t;
280 de167e41 bellard
281 de167e41 bellard
/* this structure are used to transparently access the files */
282 de167e41 bellard
283 de167e41 bellard
typedef struct mapping_t {
284 a046433a bellard
    /* begin is the first cluster, end is the last+1 */
285 a046433a bellard
    uint32_t begin,end;
286 de167e41 bellard
    /* as s->directory is growable, no pointer may be used here */
287 de167e41 bellard
    unsigned int dir_index;
288 a046433a bellard
    /* the clusters of a file may be in any order; this points to the first */
289 a046433a bellard
    int first_mapping_index;
290 a046433a bellard
    union {
291 a046433a bellard
        /* offset is
292 a046433a bellard
         * - the offset in the file (in clusters) for a file, or
293 a046433a bellard
         * - the next cluster of the directory for a directory, and
294 a046433a bellard
         * - the address of the buffer for a faked entry
295 a046433a bellard
         */
296 a046433a bellard
        struct {
297 a046433a bellard
            uint32_t offset;
298 a046433a bellard
        } file;
299 a046433a bellard
        struct {
300 a046433a bellard
            int parent_mapping_index;
301 a046433a bellard
            int first_dir_index;
302 a046433a bellard
        } dir;
303 a046433a bellard
    } info;
304 a046433a bellard
    /* path contains the full path, i.e. it always starts with s->path */
305 a046433a bellard
    char* path;
306 a046433a bellard
307 a046433a bellard
    enum { MODE_UNDEFINED = 0, MODE_NORMAL = 1, MODE_MODIFIED = 2,
308 a046433a bellard
        MODE_DIRECTORY = 4, MODE_FAKED = 8,
309 a046433a bellard
        MODE_DELETED = 16, MODE_RENAMED = 32 } mode;
310 a046433a bellard
    int read_only;
311 de167e41 bellard
} mapping_t;
312 de167e41 bellard
313 a046433a bellard
#ifdef DEBUG
314 a046433a bellard
static void print_direntry(const struct direntry_t*);
315 a046433a bellard
static void print_mapping(const struct mapping_t* mapping);
316 a046433a bellard
#endif
317 de167e41 bellard
318 de167e41 bellard
/* here begins the real VVFAT driver */
319 de167e41 bellard
320 de167e41 bellard
typedef struct BDRVVVFATState {
321 a046433a bellard
    BlockDriverState* bs; /* pointer to parent */
322 de167e41 bellard
    unsigned int first_sectors_number; /* 1 for a single partition, 0x40 for a disk with partition table */
323 de167e41 bellard
    unsigned char first_sectors[0x40*0x200];
324 3b46e624 ths
325 de167e41 bellard
    int fat_type; /* 16 or 32 */
326 de167e41 bellard
    array_t fat,directory,mapping;
327 3b46e624 ths
328 de167e41 bellard
    unsigned int cluster_size;
329 de167e41 bellard
    unsigned int sectors_per_cluster;
330 de167e41 bellard
    unsigned int sectors_per_fat;
331 de167e41 bellard
    unsigned int sectors_of_root_directory;
332 a046433a bellard
    uint32_t last_cluster_of_root_directory;
333 de167e41 bellard
    unsigned int faked_sectors; /* how many sectors are faked before file data */
334 de167e41 bellard
    uint32_t sector_count; /* total number of sectors of the partition */
335 de167e41 bellard
    uint32_t cluster_count; /* total number of clusters of this partition */
336 de167e41 bellard
    uint32_t max_fat_value;
337 3b46e624 ths
338 de167e41 bellard
    int current_fd;
339 de167e41 bellard
    mapping_t* current_mapping;
340 a046433a bellard
    unsigned char* cluster; /* points to current cluster */
341 a046433a bellard
    unsigned char* cluster_buffer; /* points to a buffer to hold temp data */
342 de167e41 bellard
    unsigned int current_cluster;
343 de167e41 bellard
344 de167e41 bellard
    /* write support */
345 a046433a bellard
    BlockDriverState* write_target;
346 a046433a bellard
    char* qcow_filename;
347 a046433a bellard
    BlockDriverState* qcow;
348 a046433a bellard
    void* fat2;
349 a046433a bellard
    char* used_clusters;
350 a046433a bellard
    array_t commits;
351 a046433a bellard
    const char* path;
352 a046433a bellard
    int downcase_short_names;
353 de167e41 bellard
} BDRVVVFATState;
354 de167e41 bellard
355 b570094d ths
/* take the sector position spos and convert it to Cylinder/Head/Sector position
356 b570094d ths
 * if the position is outside the specified geometry, fill maximum value for CHS
357 b570094d ths
 * and return 1 to signal overflow.
358 b570094d ths
 */
359 b570094d ths
static int sector2CHS(BlockDriverState* bs, mbr_chs_t * chs, int spos){
360 b570094d ths
    int head,sector;
361 b570094d ths
    sector   = spos % (bs->secs);  spos/= bs->secs;
362 b570094d ths
    head     = spos % (bs->heads); spos/= bs->heads;
363 b570094d ths
    if(spos >= bs->cyls){
364 b570094d ths
        /* Overflow,
365 b570094d ths
        it happens if 32bit sector positions are used, while CHS is only 24bit.
366 b570094d ths
        Windows/Dos is said to take 1023/255/63 as nonrepresentable CHS */
367 b570094d ths
        chs->head     = 0xFF;
368 b570094d ths
        chs->sector   = 0xFF;
369 b570094d ths
        chs->cylinder = 0xFF;
370 b570094d ths
        return 1;
371 b570094d ths
    }
372 b570094d ths
    chs->head     = (uint8_t)head;
373 b570094d ths
    chs->sector   = (uint8_t)( (sector+1) | ((spos>>8)<<6) );
374 b570094d ths
    chs->cylinder = (uint8_t)spos;
375 b570094d ths
    return 0;
376 b570094d ths
}
377 de167e41 bellard
378 de167e41 bellard
static void init_mbr(BDRVVVFATState* s)
379 de167e41 bellard
{
380 de167e41 bellard
    /* TODO: if the files mbr.img and bootsect.img exist, use them */
381 de167e41 bellard
    mbr_t* real_mbr=(mbr_t*)s->first_sectors;
382 de167e41 bellard
    partition_t* partition=&(real_mbr->partition[0]);
383 b570094d ths
    int lba;
384 de167e41 bellard
385 de167e41 bellard
    memset(s->first_sectors,0,512);
386 3b46e624 ths
387 b570094d ths
    /* Win NT Disk Signature */
388 b570094d ths
    real_mbr->nt_id= cpu_to_le32(0xbe1afdfa);
389 b570094d ths
390 de167e41 bellard
    partition->attributes=0x80; /* bootable */
391 b570094d ths
392 b570094d ths
    /* LBA is used when partition is outside the CHS geometry */
393 b570094d ths
    lba = sector2CHS(s->bs, &partition->start_CHS, s->first_sectors_number-1);
394 b570094d ths
    lba|= sector2CHS(s->bs, &partition->end_CHS,   s->sector_count);
395 b570094d ths
396 b570094d ths
    /*LBA partitions are identified only by start/length_sector_long not by CHS*/
397 b570094d ths
    partition->start_sector_long =cpu_to_le32(s->first_sectors_number-1);
398 b570094d ths
    partition->length_sector_long=cpu_to_le32(s->sector_count - s->first_sectors_number+1);
399 b570094d ths
400 a046433a bellard
    /* FAT12/FAT16/FAT32 */
401 b570094d ths
    /* DOS uses different types when partition is LBA,
402 b570094d ths
       probably to prevent older versions from using CHS on them */
403 b570094d ths
    partition->fs_type= s->fat_type==12 ? 0x1:
404 b570094d ths
                        s->fat_type==16 ? (lba?0xe:0x06):
405 b570094d ths
                         /*fat_tyoe==32*/ (lba?0xc:0x0b);
406 de167e41 bellard
407 de167e41 bellard
    real_mbr->magic[0]=0x55; real_mbr->magic[1]=0xaa;
408 de167e41 bellard
}
409 de167e41 bellard
410 a046433a bellard
/* direntry functions */
411 a046433a bellard
412 de167e41 bellard
/* dest is assumed to hold 258 bytes, and pads with 0xffff up to next multiple of 26 */
413 60fe76f3 ths
static inline int short2long_name(char* dest,const char* src)
414 de167e41 bellard
{
415 de167e41 bellard
    int i;
416 1e080d5d balrog
    int len;
417 de167e41 bellard
    for(i=0;i<129 && src[i];i++) {
418 de167e41 bellard
        dest[2*i]=src[i];
419 de167e41 bellard
        dest[2*i+1]=0;
420 de167e41 bellard
    }
421 1e080d5d balrog
    len=2*i;
422 de167e41 bellard
    dest[2*i]=dest[2*i+1]=0;
423 de167e41 bellard
    for(i=2*i+2;(i%26);i++)
424 de167e41 bellard
        dest[i]=0xff;
425 1e080d5d balrog
    return len;
426 de167e41 bellard
}
427 de167e41 bellard
428 de167e41 bellard
static inline direntry_t* create_long_filename(BDRVVVFATState* s,const char* filename)
429 de167e41 bellard
{
430 de167e41 bellard
    char buffer[258];
431 de167e41 bellard
    int length=short2long_name(buffer,filename),
432 de167e41 bellard
        number_of_entries=(length+25)/26,i;
433 de167e41 bellard
    direntry_t* entry;
434 de167e41 bellard
435 de167e41 bellard
    for(i=0;i<number_of_entries;i++) {
436 de167e41 bellard
        entry=array_get_next(&(s->directory));
437 de167e41 bellard
        entry->attributes=0xf;
438 de167e41 bellard
        entry->reserved[0]=0;
439 de167e41 bellard
        entry->begin=0;
440 de167e41 bellard
        entry->name[0]=(number_of_entries-i)|(i==0?0x40:0);
441 de167e41 bellard
    }
442 1e080d5d balrog
    for(i=0;i<26*number_of_entries;i++) {
443 de167e41 bellard
        int offset=(i%26);
444 de167e41 bellard
        if(offset<10) offset=1+offset;
445 de167e41 bellard
        else if(offset<22) offset=14+offset-10;
446 de167e41 bellard
        else offset=28+offset-22;
447 de167e41 bellard
        entry=array_get(&(s->directory),s->directory.next-1-(i/26));
448 de167e41 bellard
        entry->name[offset]=buffer[i];
449 de167e41 bellard
    }
450 de167e41 bellard
    return array_get(&(s->directory),s->directory.next-number_of_entries);
451 de167e41 bellard
}
452 de167e41 bellard
453 a046433a bellard
static char is_free(const direntry_t* direntry)
454 a046433a bellard
{
455 ad1a897e ths
    return direntry->name[0]==0xe5 || direntry->name[0]==0x00;
456 a046433a bellard
}
457 a046433a bellard
458 a046433a bellard
static char is_volume_label(const direntry_t* direntry)
459 a046433a bellard
{
460 a046433a bellard
    return direntry->attributes == 0x28;
461 a046433a bellard
}
462 a046433a bellard
463 a046433a bellard
static char is_long_name(const direntry_t* direntry)
464 a046433a bellard
{
465 a046433a bellard
    return direntry->attributes == 0xf;
466 a046433a bellard
}
467 a046433a bellard
468 a046433a bellard
static char is_short_name(const direntry_t* direntry)
469 a046433a bellard
{
470 a046433a bellard
    return !is_volume_label(direntry) && !is_long_name(direntry)
471 a046433a bellard
        && !is_free(direntry);
472 a046433a bellard
}
473 a046433a bellard
474 a046433a bellard
static char is_directory(const direntry_t* direntry)
475 a046433a bellard
{
476 a046433a bellard
    return direntry->attributes & 0x10 && direntry->name[0] != 0xe5;
477 a046433a bellard
}
478 a046433a bellard
479 a046433a bellard
static inline char is_dot(const direntry_t* direntry)
480 a046433a bellard
{
481 a046433a bellard
    return is_short_name(direntry) && direntry->name[0] == '.';
482 a046433a bellard
}
483 a046433a bellard
484 a046433a bellard
static char is_file(const direntry_t* direntry)
485 a046433a bellard
{
486 a046433a bellard
    return is_short_name(direntry) && !is_directory(direntry);
487 a046433a bellard
}
488 a046433a bellard
489 a046433a bellard
static inline uint32_t begin_of_direntry(const direntry_t* direntry)
490 a046433a bellard
{
491 a046433a bellard
    return le16_to_cpu(direntry->begin)|(le16_to_cpu(direntry->begin_hi)<<16);
492 a046433a bellard
}
493 a046433a bellard
494 a046433a bellard
static inline uint32_t filesize_of_direntry(const direntry_t* direntry)
495 a046433a bellard
{
496 a046433a bellard
    return le32_to_cpu(direntry->size);
497 a046433a bellard
}
498 a046433a bellard
499 a046433a bellard
static void set_begin_of_direntry(direntry_t* direntry, uint32_t begin)
500 a046433a bellard
{
501 a046433a bellard
    direntry->begin = cpu_to_le16(begin & 0xffff);
502 a046433a bellard
    direntry->begin_hi = cpu_to_le16((begin >> 16) & 0xffff);
503 a046433a bellard
}
504 a046433a bellard
505 de167e41 bellard
/* fat functions */
506 de167e41 bellard
507 a046433a bellard
static inline uint8_t fat_chksum(const direntry_t* entry)
508 de167e41 bellard
{
509 de167e41 bellard
    uint8_t chksum=0;
510 de167e41 bellard
    int i;
511 de167e41 bellard
512 de167e41 bellard
    for(i=0;i<11;i++)
513 de167e41 bellard
        chksum=(((chksum&0xfe)>>1)|((chksum&0x01)?0x80:0))
514 de167e41 bellard
            +(unsigned char)entry->name[i];
515 3b46e624 ths
516 de167e41 bellard
    return chksum;
517 de167e41 bellard
}
518 de167e41 bellard
519 de167e41 bellard
/* if return_time==0, this returns the fat_date, else the fat_time */
520 de167e41 bellard
static uint16_t fat_datetime(time_t time,int return_time) {
521 de167e41 bellard
    struct tm* t;
522 de167e41 bellard
#ifdef _WIN32
523 de167e41 bellard
    t=localtime(&time); /* this is not thread safe */
524 de167e41 bellard
#else
525 de167e41 bellard
    struct tm t1;
526 de167e41 bellard
    t=&t1;
527 de167e41 bellard
    localtime_r(&time,t);
528 de167e41 bellard
#endif
529 de167e41 bellard
    if(return_time)
530 de167e41 bellard
        return cpu_to_le16((t->tm_sec/2)|(t->tm_min<<5)|(t->tm_hour<<11));
531 de167e41 bellard
    return cpu_to_le16((t->tm_mday)|((t->tm_mon+1)<<5)|((t->tm_year-80)<<9));
532 de167e41 bellard
}
533 de167e41 bellard
534 de167e41 bellard
static inline void fat_set(BDRVVVFATState* s,unsigned int cluster,uint32_t value)
535 de167e41 bellard
{
536 a046433a bellard
    if(s->fat_type==32) {
537 a046433a bellard
        uint32_t* entry=array_get(&(s->fat),cluster);
538 a046433a bellard
        *entry=cpu_to_le32(value);
539 de167e41 bellard
    } else if(s->fat_type==16) {
540 de167e41 bellard
        uint16_t* entry=array_get(&(s->fat),cluster);
541 de167e41 bellard
        *entry=cpu_to_le16(value&0xffff);
542 de167e41 bellard
    } else {
543 a046433a bellard
        int offset = (cluster*3/2);
544 a046433a bellard
        unsigned char* p = array_get(&(s->fat), offset);
545 a046433a bellard
        switch (cluster&1) {
546 a046433a bellard
        case 0:
547 a046433a bellard
                p[0] = value&0xff;
548 a046433a bellard
                p[1] = (p[1]&0xf0) | ((value>>8)&0xf);
549 a046433a bellard
                break;
550 a046433a bellard
        case 1:
551 a046433a bellard
                p[0] = (p[0]&0xf) | ((value&0xf)<<4);
552 a046433a bellard
                p[1] = (value>>4);
553 a046433a bellard
                break;
554 a046433a bellard
        }
555 de167e41 bellard
    }
556 de167e41 bellard
}
557 de167e41 bellard
558 de167e41 bellard
static inline uint32_t fat_get(BDRVVVFATState* s,unsigned int cluster)
559 de167e41 bellard
{
560 a046433a bellard
    if(s->fat_type==32) {
561 a046433a bellard
        uint32_t* entry=array_get(&(s->fat),cluster);
562 a046433a bellard
        return le32_to_cpu(*entry);
563 de167e41 bellard
    } else if(s->fat_type==16) {
564 de167e41 bellard
        uint16_t* entry=array_get(&(s->fat),cluster);
565 de167e41 bellard
        return le16_to_cpu(*entry);
566 de167e41 bellard
    } else {
567 ffe8ab83 ths
        const uint8_t* x=(uint8_t*)(s->fat.pointer)+cluster*3/2;
568 a046433a bellard
        return ((x[0]|(x[1]<<8))>>(cluster&1?4:0))&0x0fff;
569 de167e41 bellard
    }
570 de167e41 bellard
}
571 de167e41 bellard
572 de167e41 bellard
static inline int fat_eof(BDRVVVFATState* s,uint32_t fat_entry)
573 de167e41 bellard
{
574 de167e41 bellard
    if(fat_entry>s->max_fat_value-8)
575 de167e41 bellard
        return -1;
576 de167e41 bellard
    return 0;
577 de167e41 bellard
}
578 de167e41 bellard
579 de167e41 bellard
static inline void init_fat(BDRVVVFATState* s)
580 de167e41 bellard
{
581 a046433a bellard
    if (s->fat_type == 12) {
582 a046433a bellard
        array_init(&(s->fat),1);
583 a046433a bellard
        array_ensure_allocated(&(s->fat),
584 a046433a bellard
                s->sectors_per_fat * 0x200 * 3 / 2 - 1);
585 a046433a bellard
    } else {
586 a046433a bellard
        array_init(&(s->fat),(s->fat_type==32?4:2));
587 a046433a bellard
        array_ensure_allocated(&(s->fat),
588 a046433a bellard
                s->sectors_per_fat * 0x200 / s->fat.item_size - 1);
589 a046433a bellard
    }
590 de167e41 bellard
    memset(s->fat.pointer,0,s->fat.size);
591 3b46e624 ths
592 de167e41 bellard
    switch(s->fat_type) {
593 de167e41 bellard
        case 12: s->max_fat_value=0xfff; break;
594 de167e41 bellard
        case 16: s->max_fat_value=0xffff; break;
595 a046433a bellard
        case 32: s->max_fat_value=0x0fffffff; break;
596 de167e41 bellard
        default: s->max_fat_value=0; /* error... */
597 de167e41 bellard
    }
598 de167e41 bellard
599 de167e41 bellard
}
600 de167e41 bellard
601 a046433a bellard
/* TODO: in create_short_filename, 0xe5->0x05 is not yet handled! */
602 a046433a bellard
/* TODO: in parse_short_filename, 0x05->0xe5 is not yet handled! */
603 a046433a bellard
static inline direntry_t* create_short_and_long_name(BDRVVVFATState* s,
604 a046433a bellard
        unsigned int directory_start, const char* filename, int is_dot)
605 de167e41 bellard
{
606 a046433a bellard
    int i,j,long_index=s->directory.next;
607 de167e41 bellard
    direntry_t* entry=0;
608 de167e41 bellard
    direntry_t* entry_long=0;
609 de167e41 bellard
610 de167e41 bellard
    if(is_dot) {
611 de167e41 bellard
        entry=array_get_next(&(s->directory));
612 de167e41 bellard
        memset(entry->name,0x20,11);
613 de167e41 bellard
        memcpy(entry->name,filename,strlen(filename));
614 de167e41 bellard
        return entry;
615 de167e41 bellard
    }
616 3b46e624 ths
617 de167e41 bellard
    entry_long=create_long_filename(s,filename);
618 3b46e624 ths
619 5fafdf24 ths
    i = strlen(filename);
620 a046433a bellard
    for(j = i - 1; j>0  && filename[j]!='.';j--);
621 a046433a bellard
    if (j > 0)
622 a046433a bellard
        i = (j > 8 ? 8 : j);
623 a046433a bellard
    else if (i > 8)
624 a046433a bellard
        i = 8;
625 a046433a bellard
626 de167e41 bellard
    entry=array_get_next(&(s->directory));
627 de167e41 bellard
    memset(entry->name,0x20,11);
628 ffe8ab83 ths
    strncpy((char*)entry->name,filename,i);
629 3b46e624 ths
630 a046433a bellard
    if(j > 0)
631 a046433a bellard
        for (i = 0; i < 3 && filename[j+1+i]; i++)
632 a046433a bellard
            entry->extension[i] = filename[j+1+i];
633 de167e41 bellard
634 de167e41 bellard
    /* upcase & remove unwanted characters */
635 de167e41 bellard
    for(i=10;i>=0;i--) {
636 a046433a bellard
        if(i==10 || i==7) for(;i>0 && entry->name[i]==' ';i--);
637 de167e41 bellard
        if(entry->name[i]<=' ' || entry->name[i]>0x7f
638 a046433a bellard
                || strchr(".*?<>|\":/\\[];,+='",entry->name[i]))
639 de167e41 bellard
            entry->name[i]='_';
640 de167e41 bellard
        else if(entry->name[i]>='a' && entry->name[i]<='z')
641 de167e41 bellard
            entry->name[i]+='A'-'a';
642 de167e41 bellard
    }
643 de167e41 bellard
644 de167e41 bellard
    /* mangle duplicates */
645 de167e41 bellard
    while(1) {
646 de167e41 bellard
        direntry_t* entry1=array_get(&(s->directory),directory_start);
647 de167e41 bellard
        int j;
648 de167e41 bellard
649 de167e41 bellard
        for(;entry1<entry;entry1++)
650 a046433a bellard
            if(!is_long_name(entry1) && !memcmp(entry1->name,entry->name,11))
651 de167e41 bellard
                break; /* found dupe */
652 de167e41 bellard
        if(entry1==entry) /* no dupe found */
653 de167e41 bellard
            break;
654 de167e41 bellard
655 5fafdf24 ths
        /* use all 8 characters of name */
656 de167e41 bellard
        if(entry->name[7]==' ') {
657 de167e41 bellard
            int j;
658 de167e41 bellard
            for(j=6;j>0 && entry->name[j]==' ';j--)
659 de167e41 bellard
                entry->name[j]='~';
660 de167e41 bellard
        }
661 de167e41 bellard
662 de167e41 bellard
        /* increment number */
663 de167e41 bellard
        for(j=7;j>0 && entry->name[j]=='9';j--)
664 de167e41 bellard
            entry->name[j]='0';
665 de167e41 bellard
        if(j>0) {
666 de167e41 bellard
            if(entry->name[j]<'0' || entry->name[j]>'9')
667 de167e41 bellard
                entry->name[j]='0';
668 de167e41 bellard
            else
669 de167e41 bellard
                entry->name[j]++;
670 de167e41 bellard
        }
671 de167e41 bellard
    }
672 de167e41 bellard
673 de167e41 bellard
    /* calculate checksum; propagate to long name */
674 de167e41 bellard
    if(entry_long) {
675 de167e41 bellard
        uint8_t chksum=fat_chksum(entry);
676 de167e41 bellard
677 de167e41 bellard
        /* calculate anew, because realloc could have taken place */
678 de167e41 bellard
        entry_long=array_get(&(s->directory),long_index);
679 a046433a bellard
        while(entry_long<entry && is_long_name(entry_long)) {
680 de167e41 bellard
            entry_long->reserved[1]=chksum;
681 de167e41 bellard
            entry_long++;
682 de167e41 bellard
        }
683 de167e41 bellard
    }
684 de167e41 bellard
685 de167e41 bellard
    return entry;
686 de167e41 bellard
}
687 de167e41 bellard
688 a046433a bellard
/*
689 a046433a bellard
 * Read a directory. (the index of the corresponding mapping must be passed).
690 a046433a bellard
 */
691 a046433a bellard
static int read_directory(BDRVVVFATState* s, int mapping_index)
692 de167e41 bellard
{
693 a046433a bellard
    mapping_t* mapping = array_get(&(s->mapping), mapping_index);
694 a046433a bellard
    direntry_t* direntry;
695 a046433a bellard
    const char* dirname = mapping->path;
696 a046433a bellard
    int first_cluster = mapping->begin;
697 a046433a bellard
    int parent_index = mapping->info.dir.parent_mapping_index;
698 a046433a bellard
    mapping_t* parent_mapping = (mapping_t*)
699 a046433a bellard
        (parent_index >= 0 ? array_get(&(s->mapping), parent_index) : 0);
700 a046433a bellard
    int first_cluster_of_parent = parent_mapping ? parent_mapping->begin : -1;
701 de167e41 bellard
702 de167e41 bellard
    DIR* dir=opendir(dirname);
703 de167e41 bellard
    struct dirent* entry;
704 de167e41 bellard
    int i;
705 de167e41 bellard
706 a046433a bellard
    assert(mapping->mode & MODE_DIRECTORY);
707 a046433a bellard
708 a046433a bellard
    if(!dir) {
709 a046433a bellard
        mapping->end = mapping->begin;
710 de167e41 bellard
        return -1;
711 a046433a bellard
    }
712 3b46e624 ths
713 a046433a bellard
    i = mapping->info.dir.first_dir_index =
714 a046433a bellard
            first_cluster == 0 ? 0 : s->directory.next;
715 a046433a bellard
716 5fafdf24 ths
    /* actually read the directory, and allocate the mappings */
717 de167e41 bellard
    while((entry=readdir(dir))) {
718 de167e41 bellard
        unsigned int length=strlen(dirname)+2+strlen(entry->d_name);
719 de167e41 bellard
        char* buffer;
720 de167e41 bellard
        direntry_t* direntry;
721 a046433a bellard
        struct stat st;
722 de167e41 bellard
        int is_dot=!strcmp(entry->d_name,".");
723 de167e41 bellard
        int is_dotdot=!strcmp(entry->d_name,"..");
724 de167e41 bellard
725 a046433a bellard
        if(first_cluster == 0 && (is_dotdot || is_dot))
726 de167e41 bellard
            continue;
727 5fafdf24 ths
728 de167e41 bellard
        buffer=(char*)malloc(length);
729 a046433a bellard
        assert(buffer);
730 de167e41 bellard
        snprintf(buffer,length,"%s/%s",dirname,entry->d_name);
731 de167e41 bellard
732 de167e41 bellard
        if(stat(buffer,&st)<0) {
733 de167e41 bellard
            free(buffer);
734 de167e41 bellard
            continue;
735 de167e41 bellard
        }
736 de167e41 bellard
737 de167e41 bellard
        /* create directory entry for this file */
738 a046433a bellard
        direntry=create_short_and_long_name(s, i, entry->d_name,
739 a046433a bellard
                is_dot || is_dotdot);
740 de167e41 bellard
        direntry->attributes=(S_ISDIR(st.st_mode)?0x10:0x20);
741 de167e41 bellard
        direntry->reserved[0]=direntry->reserved[1]=0;
742 de167e41 bellard
        direntry->ctime=fat_datetime(st.st_ctime,1);
743 de167e41 bellard
        direntry->cdate=fat_datetime(st.st_ctime,0);
744 de167e41 bellard
        direntry->adate=fat_datetime(st.st_atime,0);
745 de167e41 bellard
        direntry->begin_hi=0;
746 de167e41 bellard
        direntry->mtime=fat_datetime(st.st_mtime,1);
747 de167e41 bellard
        direntry->mdate=fat_datetime(st.st_mtime,0);
748 de167e41 bellard
        if(is_dotdot)
749 a046433a bellard
            set_begin_of_direntry(direntry, first_cluster_of_parent);
750 de167e41 bellard
        else if(is_dot)
751 a046433a bellard
            set_begin_of_direntry(direntry, first_cluster);
752 de167e41 bellard
        else
753 a046433a bellard
            direntry->begin=0; /* do that later */
754 a046433a bellard
        if (st.st_size > 0x7fffffff) {
755 a046433a bellard
            fprintf(stderr, "File %s is larger than 2GB\n", buffer);
756 a046433a bellard
            free(buffer);
757 a046433a bellard
            return -2;
758 a046433a bellard
        }
759 a046433a bellard
        direntry->size=cpu_to_le32(S_ISDIR(st.st_mode)?0:st.st_size);
760 de167e41 bellard
761 de167e41 bellard
        /* create mapping for this file */
762 a046433a bellard
        if(!is_dot && !is_dotdot && (S_ISDIR(st.st_mode) || st.st_size)) {
763 a046433a bellard
            s->current_mapping=(mapping_t*)array_get_next(&(s->mapping));
764 de167e41 bellard
            s->current_mapping->begin=0;
765 de167e41 bellard
            s->current_mapping->end=st.st_size;
766 a046433a bellard
            /*
767 a046433a bellard
             * we get the direntry of the most recent direntry, which
768 a046433a bellard
             * contains the short name and all the relevant information.
769 a046433a bellard
             */
770 de167e41 bellard
            s->current_mapping->dir_index=s->directory.next-1;
771 a046433a bellard
            s->current_mapping->first_mapping_index = -1;
772 a046433a bellard
            if (S_ISDIR(st.st_mode)) {
773 a046433a bellard
                s->current_mapping->mode = MODE_DIRECTORY;
774 a046433a bellard
                s->current_mapping->info.dir.parent_mapping_index =
775 a046433a bellard
                    mapping_index;
776 a046433a bellard
            } else {
777 a046433a bellard
                s->current_mapping->mode = MODE_UNDEFINED;
778 a046433a bellard
                s->current_mapping->info.file.offset = 0;
779 a046433a bellard
            }
780 a046433a bellard
            s->current_mapping->path=buffer;
781 a046433a bellard
            s->current_mapping->read_only =
782 a046433a bellard
                (st.st_mode & (S_IWUSR | S_IWGRP | S_IWOTH)) == 0;
783 de167e41 bellard
        }
784 de167e41 bellard
    }
785 de167e41 bellard
    closedir(dir);
786 de167e41 bellard
787 de167e41 bellard
    /* fill with zeroes up to the end of the cluster */
788 de167e41 bellard
    while(s->directory.next%(0x10*s->sectors_per_cluster)) {
789 de167e41 bellard
        direntry_t* direntry=array_get_next(&(s->directory));
790 de167e41 bellard
        memset(direntry,0,sizeof(direntry_t));
791 de167e41 bellard
    }
792 de167e41 bellard
793 a046433a bellard
/* TODO: if there are more entries, bootsector has to be adjusted! */
794 a046433a bellard
#define ROOT_ENTRIES (0x02 * 0x10 * s->sectors_per_cluster)
795 a046433a bellard
    if (mapping_index == 0 && s->directory.next < ROOT_ENTRIES) {
796 a046433a bellard
        /* root directory */
797 a046433a bellard
        int cur = s->directory.next;
798 a046433a bellard
        array_ensure_allocated(&(s->directory), ROOT_ENTRIES - 1);
799 a046433a bellard
        memset(array_get(&(s->directory), cur), 0,
800 a046433a bellard
                (ROOT_ENTRIES - cur) * sizeof(direntry_t));
801 de167e41 bellard
    }
802 5fafdf24 ths
803 a046433a bellard
     /* reget the mapping, since s->mapping was possibly realloc()ed */
804 a046433a bellard
    mapping = (mapping_t*)array_get(&(s->mapping), mapping_index);
805 a046433a bellard
    first_cluster += (s->directory.next - mapping->info.dir.first_dir_index)
806 a046433a bellard
        * 0x20 / s->cluster_size;
807 a046433a bellard
    mapping->end = first_cluster;
808 a046433a bellard
809 a046433a bellard
    direntry = (direntry_t*)array_get(&(s->directory), mapping->dir_index);
810 a046433a bellard
    set_begin_of_direntry(direntry, mapping->begin);
811 3b46e624 ths
812 a046433a bellard
    return 0;
813 a046433a bellard
}
814 de167e41 bellard
815 a046433a bellard
static inline uint32_t sector2cluster(BDRVVVFATState* s,off_t sector_num)
816 a046433a bellard
{
817 a046433a bellard
    return (sector_num-s->faked_sectors)/s->sectors_per_cluster;
818 a046433a bellard
}
819 de167e41 bellard
820 a046433a bellard
static inline off_t cluster2sector(BDRVVVFATState* s, uint32_t cluster_num)
821 a046433a bellard
{
822 a046433a bellard
    return s->faked_sectors + s->sectors_per_cluster * cluster_num;
823 a046433a bellard
}
824 de167e41 bellard
825 a046433a bellard
static inline uint32_t sector_offset_in_cluster(BDRVVVFATState* s,off_t sector_num)
826 a046433a bellard
{
827 a046433a bellard
    return (sector_num-s->first_sectors_number-2*s->sectors_per_fat)%s->sectors_per_cluster;
828 a046433a bellard
}
829 de167e41 bellard
830 a046433a bellard
#ifdef DBG
831 a046433a bellard
static direntry_t* get_direntry_for_mapping(BDRVVVFATState* s,mapping_t* mapping)
832 a046433a bellard
{
833 a046433a bellard
    if(mapping->mode==MODE_UNDEFINED)
834 a046433a bellard
        return 0;
835 a046433a bellard
    return (direntry_t*)(s->directory.pointer+sizeof(direntry_t)*mapping->dir_index);
836 de167e41 bellard
}
837 a046433a bellard
#endif
838 de167e41 bellard
839 a046433a bellard
static int init_directories(BDRVVVFATState* s,
840 a046433a bellard
        const char* dirname)
841 de167e41 bellard
{
842 a046433a bellard
    bootsector_t* bootsector;
843 a046433a bellard
    mapping_t* mapping;
844 de167e41 bellard
    unsigned int i;
845 de167e41 bellard
    unsigned int cluster;
846 de167e41 bellard
847 de167e41 bellard
    memset(&(s->first_sectors[0]),0,0x40*0x200);
848 de167e41 bellard
849 de167e41 bellard
    s->cluster_size=s->sectors_per_cluster*0x200;
850 a046433a bellard
    s->cluster_buffer=malloc(s->cluster_size);
851 a046433a bellard
    assert(s->cluster_buffer);
852 a046433a bellard
853 a046433a bellard
    /*
854 a046433a bellard
     * The formula: sc = spf+1+spf*spc*(512*8/fat_type),
855 a046433a bellard
     * where sc is sector_count,
856 a046433a bellard
     * spf is sectors_per_fat,
857 a046433a bellard
     * spc is sectors_per_clusters, and
858 a046433a bellard
     * fat_type = 12, 16 or 32.
859 a046433a bellard
     */
860 a046433a bellard
    i = 1+s->sectors_per_cluster*0x200*8/s->fat_type;
861 a046433a bellard
    s->sectors_per_fat=(s->sector_count+i)/i; /* round up */
862 3b46e624 ths
863 de167e41 bellard
    array_init(&(s->mapping),sizeof(mapping_t));
864 de167e41 bellard
    array_init(&(s->directory),sizeof(direntry_t));
865 de167e41 bellard
866 de167e41 bellard
    /* add volume label */
867 de167e41 bellard
    {
868 de167e41 bellard
        direntry_t* entry=array_get_next(&(s->directory));
869 de167e41 bellard
        entry->attributes=0x28; /* archive | volume label */
870 ffe8ab83 ths
        snprintf((char*)entry->name,11,"QEMU VVFAT");
871 de167e41 bellard
    }
872 de167e41 bellard
873 de167e41 bellard
    /* Now build FAT, and write back information into directory */
874 de167e41 bellard
    init_fat(s);
875 de167e41 bellard
876 a046433a bellard
    s->faked_sectors=s->first_sectors_number+s->sectors_per_fat*2;
877 a046433a bellard
    s->cluster_count=sector2cluster(s, s->sector_count);
878 a046433a bellard
879 a046433a bellard
    mapping = array_get_next(&(s->mapping));
880 a046433a bellard
    mapping->begin = 0;
881 a046433a bellard
    mapping->dir_index = 0;
882 a046433a bellard
    mapping->info.dir.parent_mapping_index = -1;
883 a046433a bellard
    mapping->first_mapping_index = -1;
884 a046433a bellard
    mapping->path = strdup(dirname);
885 a046433a bellard
    i = strlen(mapping->path);
886 a046433a bellard
    if (i > 0 && mapping->path[i - 1] == '/')
887 a046433a bellard
        mapping->path[i - 1] = '\0';
888 a046433a bellard
    mapping->mode = MODE_DIRECTORY;
889 a046433a bellard
    mapping->read_only = 0;
890 a046433a bellard
    s->path = mapping->path;
891 a046433a bellard
892 a046433a bellard
    for (i = 0, cluster = 0; i < s->mapping.next; i++) {
893 a046433a bellard
        int j;
894 5fafdf24 ths
        /* MS-DOS expects the FAT to be 0 for the root directory
895 a046433a bellard
         * (except for the media byte). */
896 a046433a bellard
        /* LATER TODO: still true for FAT32? */
897 a046433a bellard
        int fix_fat = (i != 0);
898 a046433a bellard
        mapping = array_get(&(s->mapping), i);
899 a046433a bellard
900 a046433a bellard
        if (mapping->mode & MODE_DIRECTORY) {
901 a046433a bellard
            mapping->begin = cluster;
902 a046433a bellard
            if(read_directory(s, i)) {
903 a046433a bellard
                fprintf(stderr, "Could not read directory %s\n",
904 a046433a bellard
                        mapping->path);
905 de167e41 bellard
                return -1;
906 de167e41 bellard
            }
907 a046433a bellard
            mapping = array_get(&(s->mapping), i);
908 a046433a bellard
        } else {
909 a046433a bellard
            assert(mapping->mode == MODE_UNDEFINED);
910 de167e41 bellard
            mapping->mode=MODE_NORMAL;
911 a046433a bellard
            mapping->begin = cluster;
912 a046433a bellard
            if (mapping->end > 0) {
913 a046433a bellard
                direntry_t* direntry = array_get(&(s->directory),
914 a046433a bellard
                        mapping->dir_index);
915 a046433a bellard
916 a046433a bellard
                mapping->end = cluster + 1 + (mapping->end-1)/s->cluster_size;
917 a046433a bellard
                set_begin_of_direntry(direntry, mapping->begin);
918 a046433a bellard
            } else {
919 a046433a bellard
                mapping->end = cluster + 1;
920 a046433a bellard
                fix_fat = 0;
921 de167e41 bellard
            }
922 a046433a bellard
        }
923 a046433a bellard
924 a046433a bellard
        assert(mapping->begin < mapping->end);
925 a046433a bellard
926 a046433a bellard
        /* fix fat for entry */
927 a046433a bellard
        if (fix_fat) {
928 a046433a bellard
            for(j = mapping->begin; j < mapping->end - 1; j++)
929 a046433a bellard
                fat_set(s, j, j+1);
930 a046433a bellard
            fat_set(s, mapping->end - 1, s->max_fat_value);
931 a046433a bellard
        }
932 a046433a bellard
933 a046433a bellard
        /* next free cluster */
934 a046433a bellard
        cluster = mapping->end;
935 de167e41 bellard
936 a046433a bellard
        if(cluster > s->cluster_count) {
937 a046433a bellard
            fprintf(stderr,"Directory does not fit in FAT%d\n",s->fat_type);
938 a046433a bellard
            return -1;
939 de167e41 bellard
        }
940 de167e41 bellard
    }
941 de167e41 bellard
942 a046433a bellard
    mapping = array_get(&(s->mapping), 0);
943 a046433a bellard
    s->sectors_of_root_directory = mapping->end * s->sectors_per_cluster;
944 a046433a bellard
    s->last_cluster_of_root_directory = mapping->end;
945 a046433a bellard
946 a046433a bellard
    /* the FAT signature */
947 a046433a bellard
    fat_set(s,0,s->max_fat_value);
948 a046433a bellard
    fat_set(s,1,s->max_fat_value);
949 de167e41 bellard
950 a046433a bellard
    s->current_mapping = NULL;
951 a046433a bellard
952 a046433a bellard
    bootsector=(bootsector_t*)(s->first_sectors+(s->first_sectors_number-1)*0x200);
953 de167e41 bellard
    bootsector->jump[0]=0xeb;
954 de167e41 bellard
    bootsector->jump[1]=0x3e;
955 de167e41 bellard
    bootsector->jump[2]=0x90;
956 de167e41 bellard
    memcpy(bootsector->name,"QEMU    ",8);
957 de167e41 bellard
    bootsector->sector_size=cpu_to_le16(0x200);
958 de167e41 bellard
    bootsector->sectors_per_cluster=s->sectors_per_cluster;
959 de167e41 bellard
    bootsector->reserved_sectors=cpu_to_le16(1);
960 de167e41 bellard
    bootsector->number_of_fats=0x2; /* number of FATs */
961 de167e41 bellard
    bootsector->root_entries=cpu_to_le16(s->sectors_of_root_directory*0x10);
962 a046433a bellard
    bootsector->total_sectors16=s->sector_count>0xffff?0:cpu_to_le16(s->sector_count);
963 a046433a bellard
    bootsector->media_type=(s->fat_type!=12?0xf8:s->sector_count==5760?0xf9:0xf8); /* media descriptor */
964 a046433a bellard
    s->fat.pointer[0] = bootsector->media_type;
965 de167e41 bellard
    bootsector->sectors_per_fat=cpu_to_le16(s->sectors_per_fat);
966 a046433a bellard
    bootsector->sectors_per_track=cpu_to_le16(s->bs->secs);
967 a046433a bellard
    bootsector->number_of_heads=cpu_to_le16(s->bs->heads);
968 de167e41 bellard
    bootsector->hidden_sectors=cpu_to_le32(s->first_sectors_number==1?0:0x3f);
969 a046433a bellard
    bootsector->total_sectors=cpu_to_le32(s->sector_count>0xffff?s->sector_count:0);
970 de167e41 bellard
971 a046433a bellard
    /* LATER TODO: if FAT32, this is wrong */
972 a046433a bellard
    bootsector->u.fat16.drive_number=s->fat_type==12?0:0x80; /* assume this is hda (TODO) */
973 de167e41 bellard
    bootsector->u.fat16.current_head=0;
974 de167e41 bellard
    bootsector->u.fat16.signature=0x29;
975 de167e41 bellard
    bootsector->u.fat16.id=cpu_to_le32(0xfabe1afd);
976 de167e41 bellard
977 de167e41 bellard
    memcpy(bootsector->u.fat16.volume_label,"QEMU VVFAT ",11);
978 de167e41 bellard
    memcpy(bootsector->fat_type,(s->fat_type==12?"FAT12   ":s->fat_type==16?"FAT16   ":"FAT32   "),8);
979 de167e41 bellard
    bootsector->magic[0]=0x55; bootsector->magic[1]=0xaa;
980 de167e41 bellard
981 de167e41 bellard
    return 0;
982 de167e41 bellard
}
983 de167e41 bellard
984 83f64091 bellard
#ifdef DEBUG
985 a046433a bellard
static BDRVVVFATState *vvv = NULL;
986 83f64091 bellard
#endif
987 a046433a bellard
988 a046433a bellard
static int enable_write_target(BDRVVVFATState *s);
989 a046433a bellard
static int is_consistent(BDRVVVFATState *s);
990 a046433a bellard
991 83f64091 bellard
static int vvfat_open(BlockDriverState *bs, const char* dirname, int flags)
992 de167e41 bellard
{
993 de167e41 bellard
    BDRVVVFATState *s = bs->opaque;
994 a046433a bellard
    int floppy = 0;
995 de167e41 bellard
    int i;
996 de167e41 bellard
997 83f64091 bellard
#ifdef DEBUG
998 a046433a bellard
    vvv = s;
999 83f64091 bellard
#endif
1000 a046433a bellard
1001 a046433a bellard
DLOG(if (stderr == NULL) {
1002 a046433a bellard
    stderr = fopen("vvfat.log", "a");
1003 a046433a bellard
    setbuf(stderr, NULL);
1004 a046433a bellard
})
1005 a046433a bellard
1006 a046433a bellard
    s->bs = bs;
1007 a046433a bellard
1008 de167e41 bellard
    s->fat_type=16;
1009 a046433a bellard
    /* LATER TODO: if FAT32, adjust */
1010 a046433a bellard
    s->sectors_per_cluster=0x10;
1011 b570094d ths
    /* 504MB disk*/
1012 b570094d ths
    bs->cyls=1024; bs->heads=16; bs->secs=63;
1013 de167e41 bellard
1014 de167e41 bellard
    s->current_cluster=0xffffffff;
1015 de167e41 bellard
1016 de167e41 bellard
    s->first_sectors_number=0x40;
1017 a046433a bellard
    /* read only is the default for safety */
1018 a046433a bellard
    bs->read_only = 1;
1019 a046433a bellard
    s->qcow = s->write_target = NULL;
1020 a046433a bellard
    s->qcow_filename = NULL;
1021 a046433a bellard
    s->fat2 = NULL;
1022 a046433a bellard
    s->downcase_short_names = 1;
1023 3b46e624 ths
1024 a046433a bellard
    if (!strstart(dirname, "fat:", NULL))
1025 a046433a bellard
        return -1;
1026 a046433a bellard
1027 a046433a bellard
    if (strstr(dirname, ":floppy:")) {
1028 a046433a bellard
        floppy = 1;
1029 a046433a bellard
        s->fat_type = 12;
1030 a046433a bellard
        s->first_sectors_number = 1;
1031 a046433a bellard
        s->sectors_per_cluster=2;
1032 a046433a bellard
        bs->cyls = 80; bs->heads = 2; bs->secs = 36;
1033 a046433a bellard
    }
1034 a046433a bellard
1035 b570094d ths
    s->sector_count=bs->cyls*bs->heads*bs->secs;
1036 b570094d ths
1037 a046433a bellard
    if (strstr(dirname, ":32:")) {
1038 a046433a bellard
        fprintf(stderr, "Big fat greek warning: FAT32 has not been tested. You are welcome to do so!\n");
1039 a046433a bellard
        s->fat_type = 32;
1040 a046433a bellard
    } else if (strstr(dirname, ":16:")) {
1041 a046433a bellard
        s->fat_type = 16;
1042 a046433a bellard
    } else if (strstr(dirname, ":12:")) {
1043 a046433a bellard
        s->fat_type = 12;
1044 a046433a bellard
        s->sector_count=2880;
1045 de167e41 bellard
    }
1046 a046433a bellard
1047 b570094d ths
    if (strstr(dirname, ":rw:")) {
1048 b570094d ths
        if (enable_write_target(s))
1049 b570094d ths
            return -1;
1050 b570094d ths
        bs->read_only = 0;
1051 b570094d ths
    }
1052 b570094d ths
1053 a046433a bellard
    i = strrchr(dirname, ':') - dirname;
1054 a046433a bellard
    assert(i >= 3);
1055 a046433a bellard
    if (dirname[i-2] == ':' && isalpha(dirname[i-1]))
1056 a046433a bellard
        /* workaround for DOS drive names */
1057 a046433a bellard
        dirname += i-1;
1058 a046433a bellard
    else
1059 a046433a bellard
        dirname += i+1;
1060 a046433a bellard
1061 a046433a bellard
    bs->total_sectors=bs->cyls*bs->heads*bs->secs;
1062 b570094d ths
1063 a046433a bellard
    if(init_directories(s, dirname))
1064 de167e41 bellard
        return -1;
1065 de167e41 bellard
1066 b570094d ths
    s->sector_count = s->faked_sectors + s->sectors_per_cluster*s->cluster_count;
1067 b570094d ths
1068 de167e41 bellard
    if(s->first_sectors_number==0x40)
1069 de167e41 bellard
        init_mbr(s);
1070 de167e41 bellard
1071 a046433a bellard
    /* for some reason or other, MS-DOS does not like to know about CHS... */
1072 a046433a bellard
    if (floppy)
1073 a046433a bellard
        bs->heads = bs->cyls = bs->secs = 0;
1074 a046433a bellard
1075 a046433a bellard
    //    assert(is_consistent(s));
1076 de167e41 bellard
    return 0;
1077 de167e41 bellard
}
1078 de167e41 bellard
1079 de167e41 bellard
static inline void vvfat_close_current_file(BDRVVVFATState *s)
1080 de167e41 bellard
{
1081 de167e41 bellard
    if(s->current_mapping) {
1082 a046433a bellard
        s->current_mapping = NULL;
1083 a046433a bellard
        if (s->current_fd) {
1084 a046433a bellard
                close(s->current_fd);
1085 a046433a bellard
                s->current_fd = 0;
1086 a046433a bellard
        }
1087 de167e41 bellard
    }
1088 a046433a bellard
    s->current_cluster = -1;
1089 de167e41 bellard
}
1090 de167e41 bellard
1091 de167e41 bellard
/* mappings between index1 and index2-1 are supposed to be ordered
1092 de167e41 bellard
 * return value is the index of the last mapping for which end>cluster_num
1093 de167e41 bellard
 */
1094 de167e41 bellard
static inline int find_mapping_for_cluster_aux(BDRVVVFATState* s,int cluster_num,int index1,int index2)
1095 de167e41 bellard
{
1096 de167e41 bellard
    int index3=index1+1;
1097 de167e41 bellard
    while(1) {
1098 de167e41 bellard
        mapping_t* mapping;
1099 de167e41 bellard
        index3=(index1+index2)/2;
1100 de167e41 bellard
        mapping=array_get(&(s->mapping),index3);
1101 a046433a bellard
        assert(mapping->begin < mapping->end);
1102 a046433a bellard
        if(mapping->begin>=cluster_num) {
1103 de167e41 bellard
            assert(index2!=index3 || index2==0);
1104 de167e41 bellard
            if(index2==index3)
1105 a046433a bellard
                return index1;
1106 de167e41 bellard
            index2=index3;
1107 de167e41 bellard
        } else {
1108 de167e41 bellard
            if(index1==index3)
1109 a046433a bellard
                return mapping->end<=cluster_num ? index2 : index1;
1110 de167e41 bellard
            index1=index3;
1111 de167e41 bellard
        }
1112 de167e41 bellard
        assert(index1<=index2);
1113 a046433a bellard
        DLOG(mapping=array_get(&(s->mapping),index1);
1114 a046433a bellard
        assert(mapping->begin<=cluster_num);
1115 5fafdf24 ths
        assert(index2 >= s->mapping.next ||
1116 a046433a bellard
                ((mapping = array_get(&(s->mapping),index2)) &&
1117 a046433a bellard
                mapping->end>cluster_num)));
1118 de167e41 bellard
    }
1119 de167e41 bellard
}
1120 de167e41 bellard
1121 de167e41 bellard
static inline mapping_t* find_mapping_for_cluster(BDRVVVFATState* s,int cluster_num)
1122 de167e41 bellard
{
1123 de167e41 bellard
    int index=find_mapping_for_cluster_aux(s,cluster_num,0,s->mapping.next);
1124 de167e41 bellard
    mapping_t* mapping;
1125 de167e41 bellard
    if(index>=s->mapping.next)
1126 de167e41 bellard
        return 0;
1127 de167e41 bellard
    mapping=array_get(&(s->mapping),index);
1128 de167e41 bellard
    if(mapping->begin>cluster_num)
1129 de167e41 bellard
        return 0;
1130 a046433a bellard
    assert(mapping->begin<=cluster_num && mapping->end>cluster_num);
1131 de167e41 bellard
    return mapping;
1132 de167e41 bellard
}
1133 de167e41 bellard
1134 a046433a bellard
/*
1135 a046433a bellard
 * This function simply compares path == mapping->path. Since the mappings
1136 a046433a bellard
 * are sorted by cluster, this is expensive: O(n).
1137 a046433a bellard
 */
1138 a046433a bellard
static inline mapping_t* find_mapping_for_path(BDRVVVFATState* s,
1139 a046433a bellard
        const char* path)
1140 a046433a bellard
{
1141 a046433a bellard
    int i;
1142 a046433a bellard
1143 a046433a bellard
    for (i = 0; i < s->mapping.next; i++) {
1144 a046433a bellard
        mapping_t* mapping = array_get(&(s->mapping), i);
1145 a046433a bellard
        if (mapping->first_mapping_index < 0 &&
1146 a046433a bellard
                !strcmp(path, mapping->path))
1147 a046433a bellard
            return mapping;
1148 a046433a bellard
    }
1149 a046433a bellard
1150 a046433a bellard
    return NULL;
1151 a046433a bellard
}
1152 a046433a bellard
1153 a046433a bellard
static int open_file(BDRVVVFATState* s,mapping_t* mapping)
1154 de167e41 bellard
{
1155 de167e41 bellard
    if(!mapping)
1156 de167e41 bellard
        return -1;
1157 de167e41 bellard
    if(!s->current_mapping ||
1158 a046433a bellard
            strcmp(s->current_mapping->path,mapping->path)) {
1159 de167e41 bellard
        /* open file */
1160 a046433a bellard
        int fd = open(mapping->path, O_RDONLY | O_BINARY | O_LARGEFILE);
1161 de167e41 bellard
        if(fd<0)
1162 de167e41 bellard
            return -1;
1163 de167e41 bellard
        vvfat_close_current_file(s);
1164 de167e41 bellard
        s->current_fd = fd;
1165 de167e41 bellard
        s->current_mapping = mapping;
1166 de167e41 bellard
    }
1167 de167e41 bellard
    return 0;
1168 de167e41 bellard
}
1169 de167e41 bellard
1170 de167e41 bellard
static inline int read_cluster(BDRVVVFATState *s,int cluster_num)
1171 de167e41 bellard
{
1172 de167e41 bellard
    if(s->current_cluster != cluster_num) {
1173 de167e41 bellard
        int result=0;
1174 de167e41 bellard
        off_t offset;
1175 a046433a bellard
        assert(!s->current_mapping || s->current_fd || (s->current_mapping->mode & MODE_DIRECTORY));
1176 de167e41 bellard
        if(!s->current_mapping
1177 de167e41 bellard
                || s->current_mapping->begin>cluster_num
1178 de167e41 bellard
                || s->current_mapping->end<=cluster_num) {
1179 de167e41 bellard
            /* binary search of mappings for file */
1180 de167e41 bellard
            mapping_t* mapping=find_mapping_for_cluster(s,cluster_num);
1181 a046433a bellard
1182 a046433a bellard
            assert(!mapping || (cluster_num>=mapping->begin && cluster_num<mapping->end));
1183 a046433a bellard
1184 a046433a bellard
            if (mapping && mapping->mode & MODE_DIRECTORY) {
1185 a046433a bellard
                vvfat_close_current_file(s);
1186 a046433a bellard
                s->current_mapping = mapping;
1187 a046433a bellard
read_cluster_directory:
1188 a046433a bellard
                offset = s->cluster_size*(cluster_num-s->current_mapping->begin);
1189 ffe8ab83 ths
                s->cluster = (unsigned char*)s->directory.pointer+offset
1190 a046433a bellard
                        + 0x20*s->current_mapping->info.dir.first_dir_index;
1191 a046433a bellard
                assert(((s->cluster-(unsigned char*)s->directory.pointer)%s->cluster_size)==0);
1192 a046433a bellard
                assert((char*)s->cluster+s->cluster_size <= s->directory.pointer+s->directory.next*s->directory.item_size);
1193 a046433a bellard
                s->current_cluster = cluster_num;
1194 a046433a bellard
                return 0;
1195 a046433a bellard
            }
1196 a046433a bellard
1197 a046433a bellard
            if(open_file(s,mapping))
1198 de167e41 bellard
                return -2;
1199 a046433a bellard
        } else if (s->current_mapping->mode & MODE_DIRECTORY)
1200 a046433a bellard
            goto read_cluster_directory;
1201 de167e41 bellard
1202 a046433a bellard
        assert(s->current_fd);
1203 a046433a bellard
1204 a046433a bellard
        offset=s->cluster_size*(cluster_num-s->current_mapping->begin)+s->current_mapping->info.file.offset;
1205 de167e41 bellard
        if(lseek(s->current_fd, offset, SEEK_SET)!=offset)
1206 de167e41 bellard
            return -3;
1207 a046433a bellard
        s->cluster=s->cluster_buffer;
1208 de167e41 bellard
        result=read(s->current_fd,s->cluster,s->cluster_size);
1209 de167e41 bellard
        if(result<0) {
1210 de167e41 bellard
            s->current_cluster = -1;
1211 de167e41 bellard
            return -1;
1212 de167e41 bellard
        }
1213 de167e41 bellard
        s->current_cluster = cluster_num;
1214 de167e41 bellard
    }
1215 de167e41 bellard
    return 0;
1216 de167e41 bellard
}
1217 de167e41 bellard
1218 a046433a bellard
#ifdef DEBUG
1219 a046433a bellard
static void hexdump(const void* address, uint32_t len)
1220 de167e41 bellard
{
1221 a046433a bellard
    const unsigned char* p = address;
1222 a046433a bellard
    int i, j;
1223 a046433a bellard
1224 a046433a bellard
    for (i = 0; i < len; i += 16) {
1225 a046433a bellard
        for (j = 0; j < 16 && i + j < len; j++)
1226 a046433a bellard
            fprintf(stderr, "%02x ", p[i + j]);
1227 a046433a bellard
        for (; j < 16; j++)
1228 a046433a bellard
            fprintf(stderr, "   ");
1229 a046433a bellard
        fprintf(stderr, " ");
1230 a046433a bellard
        for (j = 0; j < 16 && i + j < len; j++)
1231 a046433a bellard
            fprintf(stderr, "%c", (p[i + j] < ' ' || p[i + j] > 0x7f) ? '.' : p[i + j]);
1232 a046433a bellard
        fprintf(stderr, "\n");
1233 de167e41 bellard
    }
1234 de167e41 bellard
}
1235 de167e41 bellard
1236 a046433a bellard
static void print_direntry(const direntry_t* direntry)
1237 de167e41 bellard
{
1238 a046433a bellard
    int j = 0;
1239 a046433a bellard
    char buffer[1024];
1240 a046433a bellard
1241 a046433a bellard
    fprintf(stderr, "direntry 0x%x: ", (int)direntry);
1242 de167e41 bellard
    if(!direntry)
1243 de167e41 bellard
        return;
1244 a046433a bellard
    if(is_long_name(direntry)) {
1245 de167e41 bellard
        unsigned char* c=(unsigned char*)direntry;
1246 de167e41 bellard
        int i;
1247 de167e41 bellard
        for(i=1;i<11 && c[i] && c[i]!=0xff;i+=2)
1248 a046433a bellard
#define ADD_CHAR(c) {buffer[j] = (c); if (buffer[j] < ' ') buffer[j] = '?'; j++;}
1249 a046433a bellard
            ADD_CHAR(c[i]);
1250 de167e41 bellard
        for(i=14;i<26 && c[i] && c[i]!=0xff;i+=2)
1251 a046433a bellard
            ADD_CHAR(c[i]);
1252 de167e41 bellard
        for(i=28;i<32 && c[i] && c[i]!=0xff;i+=2)
1253 a046433a bellard
            ADD_CHAR(c[i]);
1254 a046433a bellard
        buffer[j] = 0;
1255 a046433a bellard
        fprintf(stderr, "%s\n", buffer);
1256 de167e41 bellard
    } else {
1257 de167e41 bellard
        int i;
1258 de167e41 bellard
        for(i=0;i<11;i++)
1259 a046433a bellard
            ADD_CHAR(direntry->name[i]);
1260 a046433a bellard
        buffer[j] = 0;
1261 a046433a bellard
        fprintf(stderr,"%s attributes=0x%02x begin=%d size=%d\n",
1262 a046433a bellard
                buffer,
1263 de167e41 bellard
                direntry->attributes,
1264 a046433a bellard
                begin_of_direntry(direntry),le32_to_cpu(direntry->size));
1265 de167e41 bellard
    }
1266 de167e41 bellard
}
1267 de167e41 bellard
1268 a046433a bellard
static void print_mapping(const mapping_t* mapping)
1269 de167e41 bellard
{
1270 a046433a bellard
    fprintf(stderr, "mapping (0x%x): begin, end = %d, %d, dir_index = %d, first_mapping_index = %d, name = %s, mode = 0x%x, " , (int)mapping, mapping->begin, mapping->end, mapping->dir_index, mapping->first_mapping_index, mapping->path, mapping->mode);
1271 a046433a bellard
    if (mapping->mode & MODE_DIRECTORY)
1272 a046433a bellard
        fprintf(stderr, "parent_mapping_index = %d, first_dir_index = %d\n", mapping->info.dir.parent_mapping_index, mapping->info.dir.first_dir_index);
1273 a046433a bellard
    else
1274 a046433a bellard
        fprintf(stderr, "offset = %d\n", mapping->info.file.offset);
1275 de167e41 bellard
}
1276 a046433a bellard
#endif
1277 de167e41 bellard
1278 5fafdf24 ths
static int vvfat_read(BlockDriverState *bs, int64_t sector_num,
1279 a046433a bellard
                    uint8_t *buf, int nb_sectors)
1280 de167e41 bellard
{
1281 a046433a bellard
    BDRVVVFATState *s = bs->opaque;
1282 de167e41 bellard
    int i;
1283 de167e41 bellard
1284 a046433a bellard
    for(i=0;i<nb_sectors;i++,sector_num++) {
1285 a046433a bellard
        if (sector_num >= s->sector_count)
1286 a046433a bellard
           return -1;
1287 a046433a bellard
        if (s->qcow) {
1288 a046433a bellard
            int n;
1289 a046433a bellard
            if (s->qcow->drv->bdrv_is_allocated(s->qcow,
1290 a046433a bellard
                        sector_num, nb_sectors-i, &n)) {
1291 a046433a bellard
DLOG(fprintf(stderr, "sectors %d+%d allocated\n", (int)sector_num, n));
1292 a046433a bellard
                if (s->qcow->drv->bdrv_read(s->qcow, sector_num, buf+i*0x200, n))
1293 a046433a bellard
                    return -1;
1294 a046433a bellard
                i += n - 1;
1295 a046433a bellard
                sector_num += n - 1;
1296 a046433a bellard
                continue;
1297 a046433a bellard
            }
1298 a046433a bellard
DLOG(fprintf(stderr, "sector %d not allocated\n", (int)sector_num));
1299 de167e41 bellard
        }
1300 a046433a bellard
        if(sector_num<s->faked_sectors) {
1301 a046433a bellard
            if(sector_num<s->first_sectors_number)
1302 a046433a bellard
                memcpy(buf+i*0x200,&(s->first_sectors[sector_num*0x200]),0x200);
1303 a046433a bellard
            else if(sector_num-s->first_sectors_number<s->sectors_per_fat)
1304 a046433a bellard
                memcpy(buf+i*0x200,&(s->fat.pointer[(sector_num-s->first_sectors_number)*0x200]),0x200);
1305 a046433a bellard
            else if(sector_num-s->first_sectors_number-s->sectors_per_fat<s->sectors_per_fat)
1306 a046433a bellard
                memcpy(buf+i*0x200,&(s->fat.pointer[(sector_num-s->first_sectors_number-s->sectors_per_fat)*0x200]),0x200);
1307 a046433a bellard
        } else {
1308 a046433a bellard
            uint32_t sector=sector_num-s->faked_sectors,
1309 a046433a bellard
            sector_offset_in_cluster=(sector%s->sectors_per_cluster),
1310 a046433a bellard
            cluster_num=sector/s->sectors_per_cluster;
1311 a046433a bellard
            if(read_cluster(s, cluster_num) != 0) {
1312 a046433a bellard
                /* LATER TODO: strict: return -1; */
1313 a046433a bellard
                memset(buf+i*0x200,0,0x200);
1314 a046433a bellard
                continue;
1315 de167e41 bellard
            }
1316 a046433a bellard
            memcpy(buf+i*0x200,s->cluster+sector_offset_in_cluster*0x200,0x200);
1317 de167e41 bellard
        }
1318 de167e41 bellard
    }
1319 de167e41 bellard
    return 0;
1320 de167e41 bellard
}
1321 de167e41 bellard
1322 a046433a bellard
/* LATER TODO: statify all functions */
1323 de167e41 bellard
1324 a046433a bellard
/*
1325 a046433a bellard
 * Idea of the write support (use snapshot):
1326 de167e41 bellard
 *
1327 a046433a bellard
 * 1. check if all data is consistent, recording renames, modifications,
1328 a046433a bellard
 *    new files and directories (in s->commits).
1329 de167e41 bellard
 *
1330 a046433a bellard
 * 2. if the data is not consistent, stop committing
1331 de167e41 bellard
 *
1332 a046433a bellard
 * 3. handle renames, and create new files and directories (do not yet
1333 a046433a bellard
 *    write their contents)
1334 de167e41 bellard
 *
1335 a046433a bellard
 * 4. walk the directories, fixing the mapping and direntries, and marking
1336 a046433a bellard
 *    the handled mappings as not deleted
1337 de167e41 bellard
 *
1338 a046433a bellard
 * 5. commit the contents of the files
1339 de167e41 bellard
 *
1340 a046433a bellard
 * 6. handle deleted files and directories
1341 de167e41 bellard
 *
1342 de167e41 bellard
 */
1343 de167e41 bellard
1344 a046433a bellard
typedef struct commit_t {
1345 a046433a bellard
    char* path;
1346 a046433a bellard
    union {
1347 a046433a bellard
        struct { uint32_t cluster; } rename;
1348 a046433a bellard
        struct { int dir_index; uint32_t modified_offset; } writeout;
1349 a046433a bellard
        struct { uint32_t first_cluster; } new_file;
1350 a046433a bellard
        struct { uint32_t cluster; } mkdir;
1351 a046433a bellard
    } param;
1352 a046433a bellard
    /* DELETEs and RMDIRs are handled differently: see handle_deletes() */
1353 a046433a bellard
    enum {
1354 a046433a bellard
        ACTION_RENAME, ACTION_WRITEOUT, ACTION_NEW_FILE, ACTION_MKDIR
1355 a046433a bellard
    } action;
1356 a046433a bellard
} commit_t;
1357 de167e41 bellard
1358 a046433a bellard
static void clear_commits(BDRVVVFATState* s)
1359 de167e41 bellard
{
1360 de167e41 bellard
    int i;
1361 a046433a bellard
DLOG(fprintf(stderr, "clear_commits (%d commits)\n", s->commits.next));
1362 a046433a bellard
    for (i = 0; i < s->commits.next; i++) {
1363 a046433a bellard
        commit_t* commit = array_get(&(s->commits), i);
1364 a046433a bellard
        assert(commit->path || commit->action == ACTION_WRITEOUT);
1365 a046433a bellard
        if (commit->action != ACTION_WRITEOUT) {
1366 a046433a bellard
            assert(commit->path);
1367 a046433a bellard
            free(commit->path);
1368 a046433a bellard
        } else
1369 a046433a bellard
            assert(commit->path == NULL);
1370 de167e41 bellard
    }
1371 a046433a bellard
    s->commits.next = 0;
1372 de167e41 bellard
}
1373 de167e41 bellard
1374 a046433a bellard
static void schedule_rename(BDRVVVFATState* s,
1375 a046433a bellard
        uint32_t cluster, char* new_path)
1376 de167e41 bellard
{
1377 a046433a bellard
    commit_t* commit = array_get_next(&(s->commits));
1378 a046433a bellard
    commit->path = new_path;
1379 a046433a bellard
    commit->param.rename.cluster = cluster;
1380 a046433a bellard
    commit->action = ACTION_RENAME;
1381 de167e41 bellard
}
1382 de167e41 bellard
1383 a046433a bellard
static void schedule_writeout(BDRVVVFATState* s,
1384 a046433a bellard
        int dir_index, uint32_t modified_offset)
1385 de167e41 bellard
{
1386 a046433a bellard
    commit_t* commit = array_get_next(&(s->commits));
1387 a046433a bellard
    commit->path = NULL;
1388 a046433a bellard
    commit->param.writeout.dir_index = dir_index;
1389 a046433a bellard
    commit->param.writeout.modified_offset = modified_offset;
1390 a046433a bellard
    commit->action = ACTION_WRITEOUT;
1391 de167e41 bellard
}
1392 de167e41 bellard
1393 a046433a bellard
static void schedule_new_file(BDRVVVFATState* s,
1394 a046433a bellard
        char* path, uint32_t first_cluster)
1395 de167e41 bellard
{
1396 a046433a bellard
    commit_t* commit = array_get_next(&(s->commits));
1397 a046433a bellard
    commit->path = path;
1398 a046433a bellard
    commit->param.new_file.first_cluster = first_cluster;
1399 a046433a bellard
    commit->action = ACTION_NEW_FILE;
1400 a046433a bellard
}
1401 a046433a bellard
1402 a046433a bellard
static void schedule_mkdir(BDRVVVFATState* s, uint32_t cluster, char* path)
1403 a046433a bellard
{
1404 a046433a bellard
    commit_t* commit = array_get_next(&(s->commits));
1405 a046433a bellard
    commit->path = path;
1406 a046433a bellard
    commit->param.mkdir.cluster = cluster;
1407 a046433a bellard
    commit->action = ACTION_MKDIR;
1408 a046433a bellard
}
1409 a046433a bellard
1410 a046433a bellard
typedef struct {
1411 64eaabda ths
    /*
1412 64eaabda ths
     * Since the sequence number is at most 0x3f, and the filename
1413 64eaabda ths
     * length is at most 13 times the sequence number, the maximal
1414 64eaabda ths
     * filename length is 0x3f * 13 bytes.
1415 64eaabda ths
     */
1416 64eaabda ths
    unsigned char name[0x3f * 13 + 1];
1417 a046433a bellard
    int checksum, len;
1418 a046433a bellard
    int sequence_number;
1419 a046433a bellard
} long_file_name;
1420 a046433a bellard
1421 a046433a bellard
static void lfn_init(long_file_name* lfn)
1422 a046433a bellard
{
1423 a046433a bellard
   lfn->sequence_number = lfn->len = 0;
1424 a046433a bellard
   lfn->checksum = 0x100;
1425 a046433a bellard
}
1426 a046433a bellard
1427 a046433a bellard
/* return 0 if parsed successfully, > 0 if no long name, < 0 if error */
1428 a046433a bellard
static int parse_long_name(long_file_name* lfn,
1429 a046433a bellard
        const direntry_t* direntry)
1430 a046433a bellard
{
1431 a046433a bellard
    int i, j, offset;
1432 a046433a bellard
    const unsigned char* pointer = (const unsigned char*)direntry;
1433 a046433a bellard
1434 a046433a bellard
    if (!is_long_name(direntry))
1435 a046433a bellard
        return 1;
1436 a046433a bellard
1437 a046433a bellard
    if (pointer[0] & 0x40) {
1438 a046433a bellard
        lfn->sequence_number = pointer[0] & 0x3f;
1439 a046433a bellard
        lfn->checksum = pointer[13];
1440 a046433a bellard
        lfn->name[0] = 0;
1441 59fdb018 ths
        lfn->name[lfn->sequence_number * 13] = 0;
1442 a046433a bellard
    } else if ((pointer[0] & 0x3f) != --lfn->sequence_number)
1443 a046433a bellard
        return -1;
1444 a046433a bellard
    else if (pointer[13] != lfn->checksum)
1445 a046433a bellard
        return -2;
1446 a046433a bellard
    else if (pointer[12] || pointer[26] || pointer[27])
1447 a046433a bellard
        return -3;
1448 a046433a bellard
1449 a046433a bellard
    offset = 13 * (lfn->sequence_number - 1);
1450 a046433a bellard
    for (i = 0, j = 1; i < 13; i++, j+=2) {
1451 a046433a bellard
        if (j == 11)
1452 a046433a bellard
            j = 14;
1453 a046433a bellard
        else if (j == 26)
1454 a046433a bellard
            j = 28;
1455 a046433a bellard
1456 a046433a bellard
        if (pointer[j+1] == 0)
1457 a046433a bellard
            lfn->name[offset + i] = pointer[j];
1458 a046433a bellard
        else if (pointer[j+1] != 0xff || (pointer[0] & 0x40) == 0)
1459 a046433a bellard
            return -4;
1460 a046433a bellard
        else
1461 a046433a bellard
            lfn->name[offset + i] = 0;
1462 de167e41 bellard
    }
1463 a046433a bellard
1464 a046433a bellard
    if (pointer[0] & 0x40)
1465 ffe8ab83 ths
        lfn->len = offset + strlen((char*)lfn->name + offset);
1466 a046433a bellard
1467 de167e41 bellard
    return 0;
1468 de167e41 bellard
}
1469 de167e41 bellard
1470 a046433a bellard
/* returns 0 if successful, >0 if no short_name, and <0 on error */
1471 a046433a bellard
static int parse_short_name(BDRVVVFATState* s,
1472 a046433a bellard
        long_file_name* lfn, direntry_t* direntry)
1473 de167e41 bellard
{
1474 a046433a bellard
    int i, j;
1475 de167e41 bellard
1476 a046433a bellard
    if (!is_short_name(direntry))
1477 a046433a bellard
        return 1;
1478 a046433a bellard
1479 a046433a bellard
    for (j = 7; j >= 0 && direntry->name[j] == ' '; j--);
1480 a046433a bellard
    for (i = 0; i <= j; i++) {
1481 a046433a bellard
        if (direntry->name[i] <= ' ' || direntry->name[i] > 0x7f)
1482 a046433a bellard
            return -1;
1483 a046433a bellard
        else if (s->downcase_short_names)
1484 a046433a bellard
            lfn->name[i] = tolower(direntry->name[i]);
1485 a046433a bellard
        else
1486 a046433a bellard
            lfn->name[i] = direntry->name[i];
1487 de167e41 bellard
    }
1488 de167e41 bellard
1489 a046433a bellard
    for (j = 2; j >= 0 && direntry->extension[j] == ' '; j--);
1490 a046433a bellard
    if (j >= 0) {
1491 a046433a bellard
        lfn->name[i++] = '.';
1492 a046433a bellard
        lfn->name[i + j + 1] = '\0';
1493 a046433a bellard
        for (;j >= 0; j--) {
1494 a046433a bellard
            if (direntry->extension[j] <= ' ' || direntry->extension[j] > 0x7f)
1495 a046433a bellard
                return -2;
1496 a046433a bellard
            else if (s->downcase_short_names)
1497 a046433a bellard
                lfn->name[i + j] = tolower(direntry->extension[j]);
1498 a046433a bellard
            else
1499 a046433a bellard
                lfn->name[i + j] = direntry->extension[j];
1500 a046433a bellard
        }
1501 a046433a bellard
    } else
1502 a046433a bellard
        lfn->name[i + j + 1] = '\0';
1503 a046433a bellard
1504 ffe8ab83 ths
    lfn->len = strlen((char*)lfn->name);
1505 a046433a bellard
1506 a046433a bellard
    return 0;
1507 de167e41 bellard
}
1508 de167e41 bellard
1509 a046433a bellard
static inline uint32_t modified_fat_get(BDRVVVFATState* s,
1510 a046433a bellard
        unsigned int cluster)
1511 de167e41 bellard
{
1512 a046433a bellard
    if (cluster < s->last_cluster_of_root_directory) {
1513 a046433a bellard
        if (cluster + 1 == s->last_cluster_of_root_directory)
1514 a046433a bellard
            return s->max_fat_value;
1515 a046433a bellard
        else
1516 a046433a bellard
            return cluster + 1;
1517 a046433a bellard
    }
1518 a046433a bellard
1519 a046433a bellard
    if (s->fat_type==32) {
1520 a046433a bellard
        uint32_t* entry=((uint32_t*)s->fat2)+cluster;
1521 a046433a bellard
        return le32_to_cpu(*entry);
1522 a046433a bellard
    } else if (s->fat_type==16) {
1523 a046433a bellard
        uint16_t* entry=((uint16_t*)s->fat2)+cluster;
1524 a046433a bellard
        return le16_to_cpu(*entry);
1525 a046433a bellard
    } else {
1526 a046433a bellard
        const uint8_t* x=s->fat2+cluster*3/2;
1527 a046433a bellard
        return ((x[0]|(x[1]<<8))>>(cluster&1?4:0))&0x0fff;
1528 a046433a bellard
    }
1529 a046433a bellard
}
1530 a046433a bellard
1531 a046433a bellard
static inline int cluster_was_modified(BDRVVVFATState* s, uint32_t cluster_num)
1532 a046433a bellard
{
1533 a046433a bellard
    int was_modified = 0;
1534 a046433a bellard
    int i, dummy;
1535 a046433a bellard
1536 a046433a bellard
    if (s->qcow == NULL)
1537 de167e41 bellard
        return 0;
1538 a046433a bellard
1539 a046433a bellard
    for (i = 0; !was_modified && i < s->sectors_per_cluster; i++)
1540 a046433a bellard
        was_modified = s->qcow->drv->bdrv_is_allocated(s->qcow,
1541 a046433a bellard
                cluster2sector(s, cluster_num) + i, 1, &dummy);
1542 a046433a bellard
1543 a046433a bellard
    return was_modified;
1544 de167e41 bellard
}
1545 de167e41 bellard
1546 a046433a bellard
static const char* get_basename(const char* path)
1547 de167e41 bellard
{
1548 a046433a bellard
    char* basename = strrchr(path, '/');
1549 a046433a bellard
    if (basename == NULL)
1550 a046433a bellard
        return path;
1551 a046433a bellard
    else
1552 a046433a bellard
        return basename + 1; /* strip '/' */
1553 de167e41 bellard
}
1554 de167e41 bellard
1555 a046433a bellard
/*
1556 a046433a bellard
 * The array s->used_clusters holds the states of the clusters. If it is
1557 a046433a bellard
 * part of a file, it has bit 2 set, in case of a directory, bit 1. If it
1558 a046433a bellard
 * was modified, bit 3 is set.
1559 a046433a bellard
 * If any cluster is allocated, but not part of a file or directory, this
1560 a046433a bellard
 * driver refuses to commit.
1561 a046433a bellard
 */
1562 a046433a bellard
typedef enum {
1563 a046433a bellard
     USED_DIRECTORY = 1, USED_FILE = 2, USED_ANY = 3, USED_ALLOCATED = 4
1564 a046433a bellard
} used_t;
1565 de167e41 bellard
1566 a046433a bellard
/*
1567 a046433a bellard
 * get_cluster_count_for_direntry() not only determines how many clusters
1568 a046433a bellard
 * are occupied by direntry, but also if it was renamed or modified.
1569 a046433a bellard
 *
1570 a046433a bellard
 * A file is thought to be renamed *only* if there already was a file with
1571 a046433a bellard
 * exactly the same first cluster, but a different name.
1572 a046433a bellard
 *
1573 a046433a bellard
 * Further, the files/directories handled by this function are
1574 a046433a bellard
 * assumed to be *not* deleted (and *only* those).
1575 a046433a bellard
 */
1576 a046433a bellard
static uint32_t get_cluster_count_for_direntry(BDRVVVFATState* s,
1577 a046433a bellard
        direntry_t* direntry, const char* path)
1578 de167e41 bellard
{
1579 a046433a bellard
    /*
1580 a046433a bellard
     * This is a little bit tricky:
1581 a046433a bellard
     * IF the guest OS just inserts a cluster into the file chain,
1582 a046433a bellard
     * and leaves the rest alone, (i.e. the original file had clusters
1583 a046433a bellard
     * 15 -> 16, but now has 15 -> 32 -> 16), then the following happens:
1584 a046433a bellard
     *
1585 a046433a bellard
     * - do_commit will write the cluster into the file at the given
1586 a046433a bellard
     *   offset, but
1587 a046433a bellard
     *
1588 a046433a bellard
     * - the cluster which is overwritten should be moved to a later
1589 a046433a bellard
     *   position in the file.
1590 a046433a bellard
     *
1591 a046433a bellard
     * I am not aware that any OS does something as braindead, but this
1592 a046433a bellard
     * situation could happen anyway when not committing for a long time.
1593 a046433a bellard
     * Just to be sure that this does not bite us, detect it, and copy the
1594 a046433a bellard
     * contents of the clusters to-be-overwritten into the qcow.
1595 a046433a bellard
     */
1596 a046433a bellard
    int copy_it = 0;
1597 a046433a bellard
    int was_modified = 0;
1598 a046433a bellard
    int32_t ret = 0;
1599 a046433a bellard
1600 a046433a bellard
    uint32_t cluster_num = begin_of_direntry(direntry);
1601 a046433a bellard
    uint32_t offset = 0;
1602 a046433a bellard
    int first_mapping_index = -1;
1603 a046433a bellard
    mapping_t* mapping = NULL;
1604 a046433a bellard
    const char* basename2 = NULL;
1605 de167e41 bellard
1606 a046433a bellard
    vvfat_close_current_file(s);
1607 de167e41 bellard
1608 a046433a bellard
    /* the root directory */
1609 a046433a bellard
    if (cluster_num == 0)
1610 de167e41 bellard
        return 0;
1611 de167e41 bellard
1612 a046433a bellard
    /* write support */
1613 a046433a bellard
    if (s->qcow) {
1614 a046433a bellard
        basename2 = get_basename(path);
1615 de167e41 bellard
1616 a046433a bellard
        mapping = find_mapping_for_cluster(s, cluster_num);
1617 a046433a bellard
1618 a046433a bellard
        if (mapping) {
1619 da2414e9 bellard
            const char* basename;
1620 da2414e9 bellard
1621 a046433a bellard
            assert(mapping->mode & MODE_DELETED);
1622 a046433a bellard
            mapping->mode &= ~MODE_DELETED;
1623 a046433a bellard
1624 da2414e9 bellard
            basename = get_basename(mapping->path);
1625 a046433a bellard
1626 a046433a bellard
            assert(mapping->mode & MODE_NORMAL);
1627 a046433a bellard
1628 a046433a bellard
            /* rename */
1629 a046433a bellard
            if (strcmp(basename, basename2))
1630 a046433a bellard
                schedule_rename(s, cluster_num, strdup(path));
1631 a046433a bellard
        } else if (is_file(direntry))
1632 a046433a bellard
            /* new file */
1633 a046433a bellard
            schedule_new_file(s, strdup(path), cluster_num);
1634 a046433a bellard
        else {
1635 a046433a bellard
            assert(0);
1636 a046433a bellard
            return 0;
1637 a046433a bellard
        }
1638 de167e41 bellard
    }
1639 de167e41 bellard
1640 a046433a bellard
    while(1) {
1641 a046433a bellard
        if (s->qcow) {
1642 a046433a bellard
            if (!copy_it && cluster_was_modified(s, cluster_num)) {
1643 a046433a bellard
                if (mapping == NULL ||
1644 a046433a bellard
                        mapping->begin > cluster_num ||
1645 a046433a bellard
                        mapping->end <= cluster_num)
1646 a046433a bellard
                mapping = find_mapping_for_cluster(s, cluster_num);
1647 de167e41 bellard
1648 a046433a bellard
1649 a046433a bellard
                if (mapping &&
1650 a046433a bellard
                        (mapping->mode & MODE_DIRECTORY) == 0) {
1651 a046433a bellard
1652 a046433a bellard
                    /* was modified in qcow */
1653 a046433a bellard
                    if (offset != mapping->info.file.offset + s->cluster_size
1654 a046433a bellard
                            * (cluster_num - mapping->begin)) {
1655 a046433a bellard
                        /* offset of this cluster in file chain has changed */
1656 a046433a bellard
                        assert(0);
1657 a046433a bellard
                        copy_it = 1;
1658 a046433a bellard
                    } else if (offset == 0) {
1659 a046433a bellard
                        const char* basename = get_basename(mapping->path);
1660 a046433a bellard
1661 a046433a bellard
                        if (strcmp(basename, basename2))
1662 a046433a bellard
                            copy_it = 1;
1663 a046433a bellard
                        first_mapping_index = array_index(&(s->mapping), mapping);
1664 a046433a bellard
                    }
1665 a046433a bellard
1666 a046433a bellard
                    if (mapping->first_mapping_index != first_mapping_index
1667 a046433a bellard
                            && mapping->info.file.offset > 0) {
1668 a046433a bellard
                        assert(0);
1669 a046433a bellard
                        copy_it = 1;
1670 a046433a bellard
                    }
1671 a046433a bellard
1672 a046433a bellard
                    /* need to write out? */
1673 a046433a bellard
                    if (!was_modified && is_file(direntry)) {
1674 a046433a bellard
                        was_modified = 1;
1675 a046433a bellard
                        schedule_writeout(s, mapping->dir_index, offset);
1676 a046433a bellard
                    }
1677 a046433a bellard
                }
1678 a046433a bellard
            }
1679 a046433a bellard
1680 a046433a bellard
            if (copy_it) {
1681 a046433a bellard
                int i, dummy;
1682 a046433a bellard
                /*
1683 a046433a bellard
                 * This is horribly inefficient, but that is okay, since
1684 a046433a bellard
                 * it is rarely executed, if at all.
1685 a046433a bellard
                 */
1686 a046433a bellard
                int64_t offset = cluster2sector(s, cluster_num);
1687 a046433a bellard
1688 a046433a bellard
                vvfat_close_current_file(s);
1689 a046433a bellard
                for (i = 0; i < s->sectors_per_cluster; i++)
1690 a046433a bellard
                    if (!s->qcow->drv->bdrv_is_allocated(s->qcow,
1691 a046433a bellard
                                offset + i, 1, &dummy)) {
1692 a046433a bellard
                        if (vvfat_read(s->bs,
1693 a046433a bellard
                                    offset, s->cluster_buffer, 1))
1694 a046433a bellard
                            return -1;
1695 a046433a bellard
                        if (s->qcow->drv->bdrv_write(s->qcow,
1696 a046433a bellard
                                    offset, s->cluster_buffer, 1))
1697 a046433a bellard
                            return -2;
1698 a046433a bellard
                    }
1699 a046433a bellard
            }
1700 a046433a bellard
        }
1701 a046433a bellard
1702 a046433a bellard
        ret++;
1703 a046433a bellard
        if (s->used_clusters[cluster_num] & USED_ANY)
1704 a046433a bellard
            return 0;
1705 a046433a bellard
        s->used_clusters[cluster_num] = USED_FILE;
1706 a046433a bellard
1707 a046433a bellard
        cluster_num = modified_fat_get(s, cluster_num);
1708 a046433a bellard
1709 a046433a bellard
        if (fat_eof(s, cluster_num))
1710 a046433a bellard
            return ret;
1711 a046433a bellard
        else if (cluster_num < 2 || cluster_num > s->max_fat_value - 16)
1712 a046433a bellard
            return -1;
1713 a046433a bellard
1714 a046433a bellard
        offset += s->cluster_size;
1715 a046433a bellard
    }
1716 de167e41 bellard
}
1717 de167e41 bellard
1718 a046433a bellard
/*
1719 5fafdf24 ths
 * This function looks at the modified data (qcow).
1720 a046433a bellard
 * It returns 0 upon inconsistency or error, and the number of clusters
1721 a046433a bellard
 * used by the directory, its subdirectories and their files.
1722 a046433a bellard
 */
1723 a046433a bellard
static int check_directory_consistency(BDRVVVFATState *s,
1724 a046433a bellard
        int cluster_num, const char* path)
1725 de167e41 bellard
{
1726 a046433a bellard
    int ret = 0;
1727 a046433a bellard
    unsigned char* cluster = malloc(s->cluster_size);
1728 a046433a bellard
    direntry_t* direntries = (direntry_t*)cluster;
1729 a046433a bellard
    mapping_t* mapping = find_mapping_for_cluster(s, cluster_num);
1730 a046433a bellard
1731 a046433a bellard
    long_file_name lfn;
1732 a046433a bellard
    int path_len = strlen(path);
1733 a046433a bellard
    char path2[PATH_MAX];
1734 a046433a bellard
1735 a046433a bellard
    assert(path_len < PATH_MAX); /* len was tested before! */
1736 363a37d5 blueswir1
    pstrcpy(path2, sizeof(path2), path);
1737 a046433a bellard
    path2[path_len] = '/';
1738 a046433a bellard
    path2[path_len + 1] = '\0';
1739 a046433a bellard
1740 a046433a bellard
    if (mapping) {
1741 a046433a bellard
        const char* basename = get_basename(mapping->path);
1742 a046433a bellard
        const char* basename2 = get_basename(path);
1743 a046433a bellard
1744 a046433a bellard
        assert(mapping->mode & MODE_DIRECTORY);
1745 a046433a bellard
1746 a046433a bellard
        assert(mapping->mode & MODE_DELETED);
1747 a046433a bellard
        mapping->mode &= ~MODE_DELETED;
1748 a046433a bellard
1749 a046433a bellard
        if (strcmp(basename, basename2))
1750 a046433a bellard
            schedule_rename(s, cluster_num, strdup(path));
1751 a046433a bellard
    } else
1752 a046433a bellard
        /* new directory */
1753 a046433a bellard
        schedule_mkdir(s, cluster_num, strdup(path));
1754 3b46e624 ths
1755 a046433a bellard
    lfn_init(&lfn);
1756 a046433a bellard
    do {
1757 de167e41 bellard
        int i;
1758 a046433a bellard
        int subret = 0;
1759 a046433a bellard
1760 a046433a bellard
        ret++;
1761 a046433a bellard
1762 a046433a bellard
        if (s->used_clusters[cluster_num] & USED_ANY) {
1763 a046433a bellard
            fprintf(stderr, "cluster %d used more than once\n", (int)cluster_num);
1764 a046433a bellard
            return 0;
1765 a046433a bellard
        }
1766 a046433a bellard
        s->used_clusters[cluster_num] = USED_DIRECTORY;
1767 a046433a bellard
1768 a046433a bellard
DLOG(fprintf(stderr, "read cluster %d (sector %d)\n", (int)cluster_num, (int)cluster2sector(s, cluster_num)));
1769 a046433a bellard
        subret = vvfat_read(s->bs, cluster2sector(s, cluster_num), cluster,
1770 a046433a bellard
                s->sectors_per_cluster);
1771 a046433a bellard
        if (subret) {
1772 a046433a bellard
            fprintf(stderr, "Error fetching direntries\n");
1773 a046433a bellard
        fail:
1774 a046433a bellard
            free(cluster);
1775 a046433a bellard
            return 0;
1776 a046433a bellard
        }
1777 a046433a bellard
1778 a046433a bellard
        for (i = 0; i < 0x10 * s->sectors_per_cluster; i++) {
1779 a046433a bellard
            int cluster_count;
1780 a046433a bellard
1781 a046433a bellard
DLOG(fprintf(stderr, "check direntry %d: \n", i); print_direntry(direntries + i));
1782 a046433a bellard
            if (is_volume_label(direntries + i) || is_dot(direntries + i) ||
1783 a046433a bellard
                    is_free(direntries + i))
1784 a046433a bellard
                continue;
1785 a046433a bellard
1786 a046433a bellard
            subret = parse_long_name(&lfn, direntries + i);
1787 a046433a bellard
            if (subret < 0) {
1788 a046433a bellard
                fprintf(stderr, "Error in long name\n");
1789 a046433a bellard
                goto fail;
1790 de167e41 bellard
            }
1791 a046433a bellard
            if (subret == 0 || is_free(direntries + i))
1792 a046433a bellard
                continue;
1793 a046433a bellard
1794 a046433a bellard
            if (fat_chksum(direntries+i) != lfn.checksum) {
1795 a046433a bellard
                subret = parse_short_name(s, &lfn, direntries + i);
1796 a046433a bellard
                if (subret < 0) {
1797 a046433a bellard
                    fprintf(stderr, "Error in short name (%d)\n", subret);
1798 a046433a bellard
                    goto fail;
1799 a046433a bellard
                }
1800 ffe8ab83 ths
                if (subret > 0 || !strcmp((char*)lfn.name, ".")
1801 ffe8ab83 ths
                        || !strcmp((char*)lfn.name, ".."))
1802 a046433a bellard
                    continue;
1803 a046433a bellard
            }
1804 a046433a bellard
            lfn.checksum = 0x100; /* cannot use long name twice */
1805 a046433a bellard
1806 a046433a bellard
            if (path_len + 1 + lfn.len >= PATH_MAX) {
1807 a046433a bellard
                fprintf(stderr, "Name too long: %s/%s\n", path, lfn.name);
1808 a046433a bellard
                goto fail;
1809 a046433a bellard
            }
1810 363a37d5 blueswir1
            pstrcpy(path2 + path_len + 1, sizeof(path2) - path_len - 1,
1811 363a37d5 blueswir1
                    (char*)lfn.name);
1812 a046433a bellard
1813 a046433a bellard
            if (is_directory(direntries + i)) {
1814 a046433a bellard
                if (begin_of_direntry(direntries + i) == 0) {
1815 a046433a bellard
                    DLOG(fprintf(stderr, "invalid begin for directory: %s\n", path2); print_direntry(direntries + i));
1816 a046433a bellard
                    goto fail;
1817 a046433a bellard
                }
1818 a046433a bellard
                cluster_count = check_directory_consistency(s,
1819 a046433a bellard
                        begin_of_direntry(direntries + i), path2);
1820 a046433a bellard
                if (cluster_count == 0) {
1821 a046433a bellard
                    DLOG(fprintf(stderr, "problem in directory %s:\n", path2); print_direntry(direntries + i));
1822 a046433a bellard
                    goto fail;
1823 a046433a bellard
                }
1824 a046433a bellard
            } else if (is_file(direntries + i)) {
1825 a046433a bellard
                /* check file size with FAT */
1826 a046433a bellard
                cluster_count = get_cluster_count_for_direntry(s, direntries + i, path2);
1827 a046433a bellard
                if (cluster_count !=
1828 a046433a bellard
                        (le32_to_cpu(direntries[i].size) + s->cluster_size
1829 a046433a bellard
                         - 1) / s->cluster_size) {
1830 a046433a bellard
                    DLOG(fprintf(stderr, "Cluster count mismatch\n"));
1831 a046433a bellard
                    goto fail;
1832 a046433a bellard
                }
1833 a046433a bellard
            } else
1834 a046433a bellard
                assert(0); /* cluster_count = 0; */
1835 a046433a bellard
1836 a046433a bellard
            ret += cluster_count;
1837 de167e41 bellard
        }
1838 de167e41 bellard
1839 a046433a bellard
        cluster_num = modified_fat_get(s, cluster_num);
1840 a046433a bellard
    } while(!fat_eof(s, cluster_num));
1841 de167e41 bellard
1842 a046433a bellard
    free(cluster);
1843 a046433a bellard
    return ret;
1844 a046433a bellard
}
1845 a046433a bellard
1846 a046433a bellard
/* returns 1 on success */
1847 a046433a bellard
static int is_consistent(BDRVVVFATState* s)
1848 a046433a bellard
{
1849 a046433a bellard
    int i, check;
1850 a046433a bellard
    int used_clusters_count = 0;
1851 a046433a bellard
1852 a046433a bellard
DLOG(checkpoint());
1853 a046433a bellard
    /*
1854 a046433a bellard
     * - get modified FAT
1855 a046433a bellard
     * - compare the two FATs (TODO)
1856 a046433a bellard
     * - get buffer for marking used clusters
1857 a046433a bellard
     * - recurse direntries from root (using bs->bdrv_read to make
1858 a046433a bellard
     *    sure to get the new data)
1859 a046433a bellard
     *   - check that the FAT agrees with the size
1860 a046433a bellard
     *   - count the number of clusters occupied by this directory and
1861 a046433a bellard
     *     its files
1862 a046433a bellard
     * - check that the cumulative used cluster count agrees with the
1863 a046433a bellard
     *   FAT
1864 a046433a bellard
     * - if all is fine, return number of used clusters
1865 a046433a bellard
     */
1866 a046433a bellard
    if (s->fat2 == NULL) {
1867 a046433a bellard
        int size = 0x200 * s->sectors_per_fat;
1868 a046433a bellard
        s->fat2 = malloc(size);
1869 a046433a bellard
        memcpy(s->fat2, s->fat.pointer, size);
1870 a046433a bellard
    }
1871 a046433a bellard
    check = vvfat_read(s->bs,
1872 a046433a bellard
            s->first_sectors_number, s->fat2, s->sectors_per_fat);
1873 a046433a bellard
    if (check) {
1874 a046433a bellard
        fprintf(stderr, "Could not copy fat\n");
1875 a046433a bellard
        return 0;
1876 a046433a bellard
    }
1877 a046433a bellard
    assert (s->used_clusters);
1878 a046433a bellard
    for (i = 0; i < sector2cluster(s, s->sector_count); i++)
1879 a046433a bellard
        s->used_clusters[i] &= ~USED_ANY;
1880 a046433a bellard
1881 a046433a bellard
    clear_commits(s);
1882 a046433a bellard
1883 a046433a bellard
    /* mark every mapped file/directory as deleted.
1884 a046433a bellard
     * (check_directory_consistency() will unmark those still present). */
1885 a046433a bellard
    if (s->qcow)
1886 a046433a bellard
        for (i = 0; i < s->mapping.next; i++) {
1887 a046433a bellard
            mapping_t* mapping = array_get(&(s->mapping), i);
1888 a046433a bellard
            if (mapping->first_mapping_index < 0)
1889 a046433a bellard
                mapping->mode |= MODE_DELETED;
1890 de167e41 bellard
        }
1891 a046433a bellard
1892 a046433a bellard
    used_clusters_count = check_directory_consistency(s, 0, s->path);
1893 a046433a bellard
    if (used_clusters_count <= 0) {
1894 a046433a bellard
        DLOG(fprintf(stderr, "problem in directory\n"));
1895 a046433a bellard
        return 0;
1896 de167e41 bellard
    }
1897 de167e41 bellard
1898 a046433a bellard
    check = s->last_cluster_of_root_directory;
1899 a046433a bellard
    for (i = check; i < sector2cluster(s, s->sector_count); i++) {
1900 a046433a bellard
        if (modified_fat_get(s, i)) {
1901 a046433a bellard
            if(!s->used_clusters[i]) {
1902 a046433a bellard
                DLOG(fprintf(stderr, "FAT was modified (%d), but cluster is not used?\n", i));
1903 a046433a bellard
                return 0;
1904 a046433a bellard
            }
1905 a046433a bellard
            check++;
1906 a046433a bellard
        }
1907 a046433a bellard
1908 a046433a bellard
        if (s->used_clusters[i] == USED_ALLOCATED) {
1909 a046433a bellard
            /* allocated, but not used... */
1910 a046433a bellard
            DLOG(fprintf(stderr, "unused, modified cluster: %d\n", i));
1911 a046433a bellard
            return 0;
1912 a046433a bellard
        }
1913 a046433a bellard
    }
1914 a046433a bellard
1915 a046433a bellard
    if (check != used_clusters_count)
1916 a046433a bellard
        return 0;
1917 a046433a bellard
1918 a046433a bellard
    return used_clusters_count;
1919 a046433a bellard
}
1920 a046433a bellard
1921 a046433a bellard
static inline void adjust_mapping_indices(BDRVVVFATState* s,
1922 a046433a bellard
        int offset, int adjust)
1923 a046433a bellard
{
1924 a046433a bellard
    int i;
1925 a046433a bellard
1926 a046433a bellard
    for (i = 0; i < s->mapping.next; i++) {
1927 a046433a bellard
        mapping_t* mapping = array_get(&(s->mapping), i);
1928 a046433a bellard
1929 a046433a bellard
#define ADJUST_MAPPING_INDEX(name) \
1930 a046433a bellard
        if (mapping->name >= offset) \
1931 a046433a bellard
            mapping->name += adjust
1932 a046433a bellard
1933 a046433a bellard
        ADJUST_MAPPING_INDEX(first_mapping_index);
1934 a046433a bellard
        if (mapping->mode & MODE_DIRECTORY)
1935 a046433a bellard
            ADJUST_MAPPING_INDEX(info.dir.parent_mapping_index);
1936 de167e41 bellard
    }
1937 a046433a bellard
}
1938 a046433a bellard
1939 a046433a bellard
/* insert or update mapping */
1940 a046433a bellard
static mapping_t* insert_mapping(BDRVVVFATState* s,
1941 a046433a bellard
        uint32_t begin, uint32_t end)
1942 a046433a bellard
{
1943 a046433a bellard
    /*
1944 a046433a bellard
     * - find mapping where mapping->begin >= begin,
1945 a046433a bellard
     * - if mapping->begin > begin: insert
1946 a046433a bellard
     *   - adjust all references to mappings!
1947 a046433a bellard
     * - else: adjust
1948 a046433a bellard
     * - replace name
1949 a046433a bellard
     */
1950 a046433a bellard
    int index = find_mapping_for_cluster_aux(s, begin, 0, s->mapping.next);
1951 a046433a bellard
    mapping_t* mapping = NULL;
1952 a046433a bellard
    mapping_t* first_mapping = array_get(&(s->mapping), 0);
1953 a046433a bellard
1954 a046433a bellard
    if (index < s->mapping.next && (mapping = array_get(&(s->mapping), index))
1955 a046433a bellard
            && mapping->begin < begin) {
1956 a046433a bellard
        mapping->end = begin;
1957 a046433a bellard
        index++;
1958 a046433a bellard
        mapping = array_get(&(s->mapping), index);
1959 a046433a bellard
    }
1960 a046433a bellard
    if (index >= s->mapping.next || mapping->begin > begin) {
1961 a046433a bellard
        mapping = array_insert(&(s->mapping), index, 1);
1962 a046433a bellard
        mapping->path = NULL;
1963 a046433a bellard
        adjust_mapping_indices(s, index, +1);
1964 a046433a bellard
    }
1965 a046433a bellard
1966 a046433a bellard
    mapping->begin = begin;
1967 a046433a bellard
    mapping->end = end;
1968 de167e41 bellard
1969 a046433a bellard
DLOG(mapping_t* next_mapping;
1970 a046433a bellard
assert(index + 1 >= s->mapping.next ||
1971 a046433a bellard
((next_mapping = array_get(&(s->mapping), index + 1)) &&
1972 a046433a bellard
 next_mapping->begin >= end)));
1973 a046433a bellard
1974 a046433a bellard
    if (s->current_mapping && first_mapping != (mapping_t*)s->mapping.pointer)
1975 a046433a bellard
        s->current_mapping = array_get(&(s->mapping),
1976 a046433a bellard
                s->current_mapping - first_mapping);
1977 a046433a bellard
1978 a046433a bellard
    return mapping;
1979 a046433a bellard
}
1980 a046433a bellard
1981 a046433a bellard
static int remove_mapping(BDRVVVFATState* s, int mapping_index)
1982 a046433a bellard
{
1983 a046433a bellard
    mapping_t* mapping = array_get(&(s->mapping), mapping_index);
1984 a046433a bellard
    mapping_t* first_mapping = array_get(&(s->mapping), 0);
1985 a046433a bellard
1986 a046433a bellard
    /* free mapping */
1987 a046433a bellard
    if (mapping->first_mapping_index < 0)
1988 a046433a bellard
        free(mapping->path);
1989 a046433a bellard
1990 a046433a bellard
    /* remove from s->mapping */
1991 a046433a bellard
    array_remove(&(s->mapping), mapping_index);
1992 a046433a bellard
1993 a046433a bellard
    /* adjust all references to mappings */
1994 a046433a bellard
    adjust_mapping_indices(s, mapping_index, -1);
1995 a046433a bellard
1996 a046433a bellard
    if (s->current_mapping && first_mapping != (mapping_t*)s->mapping.pointer)
1997 a046433a bellard
        s->current_mapping = array_get(&(s->mapping),
1998 a046433a bellard
                s->current_mapping - first_mapping);
1999 de167e41 bellard
2000 de167e41 bellard
    return 0;
2001 de167e41 bellard
}
2002 de167e41 bellard
2003 a046433a bellard
static void adjust_dirindices(BDRVVVFATState* s, int offset, int adjust)
2004 a046433a bellard
{
2005 a046433a bellard
    int i;
2006 a046433a bellard
    for (i = 0; i < s->mapping.next; i++) {
2007 a046433a bellard
        mapping_t* mapping = array_get(&(s->mapping), i);
2008 a046433a bellard
        if (mapping->dir_index >= offset)
2009 a046433a bellard
            mapping->dir_index += adjust;
2010 a046433a bellard
        if ((mapping->mode & MODE_DIRECTORY) &&
2011 a046433a bellard
                mapping->info.dir.first_dir_index >= offset)
2012 a046433a bellard
            mapping->info.dir.first_dir_index += adjust;
2013 a046433a bellard
    }
2014 a046433a bellard
}
2015 de167e41 bellard
2016 a046433a bellard
static direntry_t* insert_direntries(BDRVVVFATState* s,
2017 a046433a bellard
        int dir_index, int count)
2018 de167e41 bellard
{
2019 a046433a bellard
    /*
2020 a046433a bellard
     * make room in s->directory,
2021 a046433a bellard
     * adjust_dirindices
2022 a046433a bellard
     */
2023 a046433a bellard
    direntry_t* result = array_insert(&(s->directory), dir_index, count);
2024 a046433a bellard
    if (result == NULL)
2025 a046433a bellard
        return NULL;
2026 a046433a bellard
    adjust_dirindices(s, dir_index, count);
2027 de167e41 bellard
    return result;
2028 de167e41 bellard
}
2029 de167e41 bellard
2030 a046433a bellard
static int remove_direntries(BDRVVVFATState* s, int dir_index, int count)
2031 a046433a bellard
{
2032 a046433a bellard
    int ret = array_remove_slice(&(s->directory), dir_index, count);
2033 a046433a bellard
    if (ret)
2034 a046433a bellard
        return ret;
2035 a046433a bellard
    adjust_dirindices(s, dir_index, -count);
2036 a046433a bellard
    return 0;
2037 a046433a bellard
}
2038 de167e41 bellard
2039 a046433a bellard
/*
2040 a046433a bellard
 * Adapt the mappings of the cluster chain starting at first cluster
2041 a046433a bellard
 * (i.e. if a file starts at first_cluster, the chain is followed according
2042 a046433a bellard
 * to the modified fat, and the corresponding entries in s->mapping are
2043 a046433a bellard
 * adjusted)
2044 a046433a bellard
 */
2045 a046433a bellard
static int commit_mappings(BDRVVVFATState* s,
2046 a046433a bellard
        uint32_t first_cluster, int dir_index)
2047 de167e41 bellard
{
2048 a046433a bellard
    mapping_t* mapping = find_mapping_for_cluster(s, first_cluster);
2049 a046433a bellard
    direntry_t* direntry = array_get(&(s->directory), dir_index);
2050 a046433a bellard
    uint32_t cluster = first_cluster;
2051 a046433a bellard
2052 a046433a bellard
    vvfat_close_current_file(s);
2053 a046433a bellard
2054 a046433a bellard
    assert(mapping);
2055 a046433a bellard
    assert(mapping->begin == first_cluster);
2056 a046433a bellard
    mapping->first_mapping_index = -1;
2057 a046433a bellard
    mapping->dir_index = dir_index;
2058 a046433a bellard
    mapping->mode = (dir_index <= 0 || is_directory(direntry)) ?
2059 a046433a bellard
        MODE_DIRECTORY : MODE_NORMAL;
2060 a046433a bellard
2061 a046433a bellard
    while (!fat_eof(s, cluster)) {
2062 a046433a bellard
        uint32_t c, c1;
2063 a046433a bellard
2064 a046433a bellard
        for (c = cluster, c1 = modified_fat_get(s, c); c + 1 == c1;
2065 a046433a bellard
                c = c1, c1 = modified_fat_get(s, c1));
2066 a046433a bellard
2067 a046433a bellard
        c++;
2068 a046433a bellard
        if (c > mapping->end) {
2069 a046433a bellard
            int index = array_index(&(s->mapping), mapping);
2070 a046433a bellard
            int i, max_i = s->mapping.next - index;
2071 a046433a bellard
            for (i = 1; i < max_i && mapping[i].begin < c; i++);
2072 a046433a bellard
            while (--i > 0)
2073 a046433a bellard
                remove_mapping(s, index + 1);
2074 a046433a bellard
        }
2075 a046433a bellard
        assert(mapping == array_get(&(s->mapping), s->mapping.next - 1)
2076 a046433a bellard
                || mapping[1].begin >= c);
2077 a046433a bellard
        mapping->end = c;
2078 a046433a bellard
2079 a046433a bellard
        if (!fat_eof(s, c1)) {
2080 a046433a bellard
            int i = find_mapping_for_cluster_aux(s, c1, 0, s->mapping.next);
2081 a046433a bellard
            mapping_t* next_mapping = i >= s->mapping.next ? NULL :
2082 a046433a bellard
                array_get(&(s->mapping), i);
2083 a046433a bellard
2084 a046433a bellard
            if (next_mapping == NULL || next_mapping->begin > c1) {
2085 a046433a bellard
                int i1 = array_index(&(s->mapping), mapping);
2086 a046433a bellard
2087 a046433a bellard
                next_mapping = insert_mapping(s, c1, c1+1);
2088 a046433a bellard
2089 a046433a bellard
                if (c1 < c)
2090 a046433a bellard
                    i1++;
2091 a046433a bellard
                mapping = array_get(&(s->mapping), i1);
2092 a046433a bellard
            }
2093 a046433a bellard
2094 a046433a bellard
            next_mapping->dir_index = mapping->dir_index;
2095 5fafdf24 ths
            next_mapping->first_mapping_index =
2096 a046433a bellard
                mapping->first_mapping_index < 0 ?
2097 a046433a bellard
                array_index(&(s->mapping), mapping) :
2098 a046433a bellard
                mapping->first_mapping_index;
2099 a046433a bellard
            next_mapping->path = mapping->path;
2100 a046433a bellard
            next_mapping->mode = mapping->mode;
2101 a046433a bellard
            next_mapping->read_only = mapping->read_only;
2102 a046433a bellard
            if (mapping->mode & MODE_DIRECTORY) {
2103 a046433a bellard
                next_mapping->info.dir.parent_mapping_index =
2104 a046433a bellard
                        mapping->info.dir.parent_mapping_index;
2105 a046433a bellard
                next_mapping->info.dir.first_dir_index =
2106 a046433a bellard
                        mapping->info.dir.first_dir_index +
2107 a046433a bellard
                        0x10 * s->sectors_per_cluster *
2108 a046433a bellard
                        (mapping->end - mapping->begin);
2109 a046433a bellard
            } else
2110 a046433a bellard
                next_mapping->info.file.offset = mapping->info.file.offset +
2111 a046433a bellard
                        mapping->end - mapping->begin;
2112 a046433a bellard
2113 a046433a bellard
            mapping = next_mapping;
2114 a046433a bellard
        }
2115 3b46e624 ths
2116 a046433a bellard
        cluster = c1;
2117 a046433a bellard
    }
2118 de167e41 bellard
2119 de167e41 bellard
    return 0;
2120 de167e41 bellard
}
2121 de167e41 bellard
2122 a046433a bellard
static int commit_direntries(BDRVVVFATState* s,
2123 a046433a bellard
        int dir_index, int parent_mapping_index)
2124 de167e41 bellard
{
2125 a046433a bellard
    direntry_t* direntry = array_get(&(s->directory), dir_index);
2126 a046433a bellard
    uint32_t first_cluster = dir_index == 0 ? 0 : begin_of_direntry(direntry);
2127 a046433a bellard
    mapping_t* mapping = find_mapping_for_cluster(s, first_cluster);
2128 a046433a bellard
2129 a046433a bellard
    int factor = 0x10 * s->sectors_per_cluster;
2130 a046433a bellard
    int old_cluster_count, new_cluster_count;
2131 a046433a bellard
    int current_dir_index = mapping->info.dir.first_dir_index;
2132 a046433a bellard
    int first_dir_index = current_dir_index;
2133 a046433a bellard
    int ret, i;
2134 a046433a bellard
    uint32_t c;
2135 a046433a bellard
2136 a046433a bellard
DLOG(fprintf(stderr, "commit_direntries for %s, parent_mapping_index %d\n", mapping->path, parent_mapping_index));
2137 a046433a bellard
2138 a046433a bellard
    assert(direntry);
2139 a046433a bellard
    assert(mapping);
2140 a046433a bellard
    assert(mapping->begin == first_cluster);
2141 a046433a bellard
    assert(mapping->info.dir.first_dir_index < s->directory.next);
2142 a046433a bellard
    assert(mapping->mode & MODE_DIRECTORY);
2143 a046433a bellard
    assert(dir_index == 0 || is_directory(direntry));
2144 a046433a bellard
2145 a046433a bellard
    mapping->info.dir.parent_mapping_index = parent_mapping_index;
2146 a046433a bellard
2147 a046433a bellard
    if (first_cluster == 0) {
2148 a046433a bellard
        old_cluster_count = new_cluster_count =
2149 a046433a bellard
            s->last_cluster_of_root_directory;
2150 a046433a bellard
    } else {
2151 a046433a bellard
        for (old_cluster_count = 0, c = first_cluster; !fat_eof(s, c);
2152 a046433a bellard
                c = fat_get(s, c))
2153 a046433a bellard
            old_cluster_count++;
2154 de167e41 bellard
2155 a046433a bellard
        for (new_cluster_count = 0, c = first_cluster; !fat_eof(s, c);
2156 a046433a bellard
                c = modified_fat_get(s, c))
2157 a046433a bellard
            new_cluster_count++;
2158 a046433a bellard
    }
2159 de167e41 bellard
2160 a046433a bellard
    if (new_cluster_count > old_cluster_count) {
2161 a046433a bellard
        if (insert_direntries(s,
2162 a046433a bellard
                current_dir_index + factor * old_cluster_count,
2163 a046433a bellard
                factor * (new_cluster_count - old_cluster_count)) == NULL)
2164 a046433a bellard
            return -1;
2165 a046433a bellard
    } else if (new_cluster_count < old_cluster_count)
2166 a046433a bellard
        remove_direntries(s,
2167 a046433a bellard
                current_dir_index + factor * new_cluster_count,
2168 a046433a bellard
                factor * (old_cluster_count - new_cluster_count));
2169 a046433a bellard
2170 a046433a bellard
    for (c = first_cluster; !fat_eof(s, c); c = modified_fat_get(s, c)) {
2171 a046433a bellard
        void* direntry = array_get(&(s->directory), current_dir_index);
2172 a046433a bellard
        int ret = vvfat_read(s->bs, cluster2sector(s, c), direntry,
2173 a046433a bellard
                s->sectors_per_cluster);
2174 a046433a bellard
        if (ret)
2175 a046433a bellard
            return ret;
2176 a046433a bellard
        assert(!strncmp(s->directory.pointer, "QEMU", 4));
2177 a046433a bellard
        current_dir_index += factor;
2178 a046433a bellard
    }
2179 de167e41 bellard
2180 a046433a bellard
    ret = commit_mappings(s, first_cluster, dir_index);
2181 a046433a bellard
    if (ret)
2182 a046433a bellard
        return ret;
2183 a046433a bellard
2184 a046433a bellard
    /* recurse */
2185 a046433a bellard
    for (i = 0; i < factor * new_cluster_count; i++) {
2186 a046433a bellard
        direntry = array_get(&(s->directory), first_dir_index + i);
2187 a046433a bellard
        if (is_directory(direntry) && !is_dot(direntry)) {
2188 a046433a bellard
            mapping = find_mapping_for_cluster(s, first_cluster);
2189 a046433a bellard
            assert(mapping->mode & MODE_DIRECTORY);
2190 a046433a bellard
            ret = commit_direntries(s, first_dir_index + i,
2191 a046433a bellard
                array_index(&(s->mapping), mapping));
2192 a046433a bellard
            if (ret)
2193 a046433a bellard
                return ret;
2194 a046433a bellard
        }
2195 a046433a bellard
    }
2196 de167e41 bellard
2197 a046433a bellard
    return 0;
2198 a046433a bellard
}
2199 de167e41 bellard
2200 a046433a bellard
/* commit one file (adjust contents, adjust mapping),
2201 a046433a bellard
   return first_mapping_index */
2202 a046433a bellard
static int commit_one_file(BDRVVVFATState* s,
2203 a046433a bellard
        int dir_index, uint32_t offset)
2204 a046433a bellard
{
2205 a046433a bellard
    direntry_t* direntry = array_get(&(s->directory), dir_index);
2206 a046433a bellard
    uint32_t c = begin_of_direntry(direntry);
2207 a046433a bellard
    uint32_t first_cluster = c;
2208 a046433a bellard
    mapping_t* mapping = find_mapping_for_cluster(s, c);
2209 a046433a bellard
    uint32_t size = filesize_of_direntry(direntry);
2210 a046433a bellard
    char* cluster = malloc(s->cluster_size);
2211 a046433a bellard
    uint32_t i;
2212 a046433a bellard
    int fd = 0;
2213 a046433a bellard
2214 a046433a bellard
    assert(offset < size);
2215 a046433a bellard
    assert((offset % s->cluster_size) == 0);
2216 a046433a bellard
2217 a046433a bellard
    for (i = s->cluster_size; i < offset; i += s->cluster_size)
2218 a046433a bellard
        c = modified_fat_get(s, c);
2219 a046433a bellard
2220 6bcb76c3 bellard
    fd = open(mapping->path, O_RDWR | O_CREAT | O_BINARY, 0666);
2221 a046433a bellard
    if (fd < 0) {
2222 a046433a bellard
        fprintf(stderr, "Could not open %s... (%s, %d)\n", mapping->path,
2223 a046433a bellard
                strerror(errno), errno);
2224 a046433a bellard
        return fd;
2225 de167e41 bellard
    }
2226 a046433a bellard
    if (offset > 0)
2227 a046433a bellard
        if (lseek(fd, offset, SEEK_SET) != offset)
2228 a046433a bellard
            return -3;
2229 a046433a bellard
2230 a046433a bellard
    while (offset < size) {
2231 a046433a bellard
        uint32_t c1;
2232 a046433a bellard
        int rest_size = (size - offset > s->cluster_size ?
2233 a046433a bellard
                s->cluster_size : size - offset);
2234 a046433a bellard
        int ret;
2235 a046433a bellard
2236 a046433a bellard
        c1 = modified_fat_get(s, c);
2237 a046433a bellard
2238 a046433a bellard
        assert((size - offset == 0 && fat_eof(s, c)) ||
2239 a046433a bellard
                (size > offset && c >=2 && !fat_eof(s, c)));
2240 a046433a bellard
2241 a046433a bellard
        ret = vvfat_read(s->bs, cluster2sector(s, c),
2242 ffe8ab83 ths
            (uint8_t*)cluster, (rest_size + 0x1ff) / 0x200);
2243 a046433a bellard
2244 a046433a bellard
        if (ret < 0)
2245 a046433a bellard
            return ret;
2246 a046433a bellard
2247 a046433a bellard
        if (write(fd, cluster, rest_size) < 0)
2248 a046433a bellard
            return -2;
2249 a046433a bellard
2250 a046433a bellard
        offset += rest_size;
2251 a046433a bellard
        c = c1;
2252 a046433a bellard
    }
2253 a046433a bellard
2254 a046433a bellard
    ftruncate(fd, size);
2255 a046433a bellard
    close(fd);
2256 a046433a bellard
2257 a046433a bellard
    return commit_mappings(s, first_cluster, dir_index);
2258 a046433a bellard
}
2259 a046433a bellard
2260 a046433a bellard
#ifdef DEBUG
2261 a046433a bellard
/* test, if all mappings point to valid direntries */
2262 a046433a bellard
static void check1(BDRVVVFATState* s)
2263 a046433a bellard
{
2264 a046433a bellard
    int i;
2265 a046433a bellard
    for (i = 0; i < s->mapping.next; i++) {
2266 a046433a bellard
        mapping_t* mapping = array_get(&(s->mapping), i);
2267 a046433a bellard
        if (mapping->mode & MODE_DELETED) {
2268 a046433a bellard
            fprintf(stderr, "deleted\n");
2269 a046433a bellard
            continue;
2270 a046433a bellard
        }
2271 a046433a bellard
        assert(mapping->dir_index >= 0);
2272 a046433a bellard
        assert(mapping->dir_index < s->directory.next);
2273 a046433a bellard
        direntry_t* direntry = array_get(&(s->directory), mapping->dir_index);
2274 a046433a bellard
        assert(mapping->begin == begin_of_direntry(direntry) || mapping->first_mapping_index >= 0);
2275 a046433a bellard
        if (mapping->mode & MODE_DIRECTORY) {
2276 a046433a bellard
            assert(mapping->info.dir.first_dir_index + 0x10 * s->sectors_per_cluster * (mapping->end - mapping->begin) <= s->directory.next);
2277 a046433a bellard
            assert((mapping->info.dir.first_dir_index % (0x10 * s->sectors_per_cluster)) == 0);
2278 de167e41 bellard
        }
2279 de167e41 bellard
    }
2280 de167e41 bellard
}
2281 de167e41 bellard
2282 a046433a bellard
/* test, if all direntries have mappings */
2283 a046433a bellard
static void check2(BDRVVVFATState* s)
2284 de167e41 bellard
{
2285 de167e41 bellard
    int i;
2286 a046433a bellard
    int first_mapping = -1;
2287 de167e41 bellard
2288 a046433a bellard
    for (i = 0; i < s->directory.next; i++) {
2289 a046433a bellard
        direntry_t* direntry = array_get(&(s->directory), i);
2290 de167e41 bellard
2291 a046433a bellard
        if (is_short_name(direntry) && begin_of_direntry(direntry)) {
2292 a046433a bellard
            mapping_t* mapping = find_mapping_for_cluster(s, begin_of_direntry(direntry));
2293 a046433a bellard
            assert(mapping);
2294 a046433a bellard
            assert(mapping->dir_index == i || is_dot(direntry));
2295 a046433a bellard
            assert(mapping->begin == begin_of_direntry(direntry) || is_dot(direntry));
2296 a046433a bellard
        }
2297 de167e41 bellard
2298 a046433a bellard
        if ((i % (0x10 * s->sectors_per_cluster)) == 0) {
2299 a046433a bellard
            /* cluster start */
2300 a046433a bellard
            int j, count = 0;
2301 de167e41 bellard
2302 a046433a bellard
            for (j = 0; j < s->mapping.next; j++) {
2303 a046433a bellard
                mapping_t* mapping = array_get(&(s->mapping), j);
2304 a046433a bellard
                if (mapping->mode & MODE_DELETED)
2305 de167e41 bellard
                    continue;
2306 a046433a bellard
                if (mapping->mode & MODE_DIRECTORY) {
2307 a046433a bellard
                    if (mapping->info.dir.first_dir_index <= i && mapping->info.dir.first_dir_index + 0x10 * s->sectors_per_cluster > i) {
2308 a046433a bellard
                        assert(++count == 1);
2309 a046433a bellard
                        if (mapping->first_mapping_index == -1)
2310 a046433a bellard
                            first_mapping = array_index(&(s->mapping), mapping);
2311 a046433a bellard
                        else
2312 a046433a bellard
                            assert(first_mapping == mapping->first_mapping_index);
2313 a046433a bellard
                        if (mapping->info.dir.parent_mapping_index < 0)
2314 a046433a bellard
                            assert(j == 0);
2315 a046433a bellard
                        else {
2316 a046433a bellard
                            mapping_t* parent = array_get(&(s->mapping), mapping->info.dir.parent_mapping_index);
2317 a046433a bellard
                            assert(parent->mode & MODE_DIRECTORY);
2318 a046433a bellard
                            assert(parent->info.dir.first_dir_index < mapping->info.dir.first_dir_index);
2319 a046433a bellard
                        }
2320 a046433a bellard
                    }
2321 de167e41 bellard
                }
2322 a046433a bellard
            }
2323 a046433a bellard
            if (count == 0)
2324 a046433a bellard
                first_mapping = -1;
2325 a046433a bellard
        }
2326 a046433a bellard
    }
2327 a046433a bellard
}
2328 a046433a bellard
#endif
2329 de167e41 bellard
2330 a046433a bellard
static int handle_renames_and_mkdirs(BDRVVVFATState* s)
2331 a046433a bellard
{
2332 a046433a bellard
    int i;
2333 de167e41 bellard
2334 a046433a bellard
#ifdef DEBUG
2335 a046433a bellard
    fprintf(stderr, "handle_renames\n");
2336 a046433a bellard
    for (i = 0; i < s->commits.next; i++) {
2337 a046433a bellard
        commit_t* commit = array_get(&(s->commits), i);
2338 a046433a bellard
        fprintf(stderr, "%d, %s (%d, %d)\n", i, commit->path ? commit->path : "(null)", commit->param.rename.cluster, commit->action);
2339 a046433a bellard
    }
2340 a046433a bellard
#endif
2341 a046433a bellard
2342 a046433a bellard
    for (i = 0; i < s->commits.next;) {
2343 a046433a bellard
        commit_t* commit = array_get(&(s->commits), i);
2344 a046433a bellard
        if (commit->action == ACTION_RENAME) {
2345 a046433a bellard
            mapping_t* mapping = find_mapping_for_cluster(s,
2346 a046433a bellard
                    commit->param.rename.cluster);
2347 a046433a bellard
            char* old_path = mapping->path;
2348 a046433a bellard
2349 a046433a bellard
            assert(commit->path);
2350 a046433a bellard
            mapping->path = commit->path;
2351 a046433a bellard
            if (rename(old_path, mapping->path))
2352 a046433a bellard
                return -2;
2353 a046433a bellard
2354 a046433a bellard
            if (mapping->mode & MODE_DIRECTORY) {
2355 a046433a bellard
                int l1 = strlen(mapping->path);
2356 a046433a bellard
                int l2 = strlen(old_path);
2357 a046433a bellard
                int diff = l1 - l2;
2358 a046433a bellard
                direntry_t* direntry = array_get(&(s->directory),
2359 a046433a bellard
                        mapping->info.dir.first_dir_index);
2360 a046433a bellard
                uint32_t c = mapping->begin;
2361 a046433a bellard
                int i = 0;
2362 a046433a bellard
2363 a046433a bellard
                /* recurse */
2364 a046433a bellard
                while (!fat_eof(s, c)) {
2365 a046433a bellard
                    do {
2366 a046433a bellard
                        direntry_t* d = direntry + i;
2367 a046433a bellard
2368 a046433a bellard
                        if (is_file(d) || (is_directory(d) && !is_dot(d))) {
2369 a046433a bellard
                            mapping_t* m = find_mapping_for_cluster(s,
2370 a046433a bellard
                                    begin_of_direntry(d));
2371 a046433a bellard
                            int l = strlen(m->path);
2372 a046433a bellard
                            char* new_path = malloc(l + diff + 1);
2373 a046433a bellard
2374 a046433a bellard
                            assert(!strncmp(m->path, mapping->path, l2));
2375 a046433a bellard
2376 363a37d5 blueswir1
                            pstrcpy(new_path, l + diff + 1, mapping->path);
2377 363a37d5 blueswir1
                            pstrcpy(new_path + l1, l + diff + 1 - l1,
2378 363a37d5 blueswir1
                                    m->path + l2);
2379 a046433a bellard
2380 a046433a bellard
                            schedule_rename(s, m->begin, new_path);
2381 de167e41 bellard
                        }
2382 a046433a bellard
                        i++;
2383 a046433a bellard
                    } while((i % (0x10 * s->sectors_per_cluster)) != 0);
2384 a046433a bellard
                    c = fat_get(s, c);
2385 de167e41 bellard
                }
2386 de167e41 bellard
            }
2387 de167e41 bellard
2388 a046433a bellard
            free(old_path);
2389 a046433a bellard
            array_remove(&(s->commits), i);
2390 a046433a bellard
            continue;
2391 a046433a bellard
        } else if (commit->action == ACTION_MKDIR) {
2392 a046433a bellard
            mapping_t* mapping;
2393 a046433a bellard
            int j, parent_path_len;
2394 a046433a bellard
2395 48c2f068 bellard
#ifdef __MINGW32__
2396 48c2f068 bellard
            if (mkdir(commit->path))
2397 48c2f068 bellard
                return -5;
2398 48c2f068 bellard
#else
2399 48c2f068 bellard
            if (mkdir(commit->path, 0755))
2400 48c2f068 bellard
                return -5;
2401 48c2f068 bellard
#endif
2402 a046433a bellard
2403 a046433a bellard
            mapping = insert_mapping(s, commit->param.mkdir.cluster,
2404 a046433a bellard
                    commit->param.mkdir.cluster + 1);
2405 a046433a bellard
            if (mapping == NULL)
2406 a046433a bellard
                return -6;
2407 a046433a bellard
2408 a046433a bellard
            mapping->mode = MODE_DIRECTORY;
2409 a046433a bellard
            mapping->read_only = 0;
2410 a046433a bellard
            mapping->path = commit->path;
2411 a046433a bellard
            j = s->directory.next;
2412 a046433a bellard
            assert(j);
2413 a046433a bellard
            insert_direntries(s, s->directory.next,
2414 a046433a bellard
                    0x10 * s->sectors_per_cluster);
2415 a046433a bellard
            mapping->info.dir.first_dir_index = j;
2416 a046433a bellard
2417 a046433a bellard
            parent_path_len = strlen(commit->path)
2418 a046433a bellard
                - strlen(get_basename(commit->path)) - 1;
2419 a046433a bellard
            for (j = 0; j < s->mapping.next; j++) {
2420 a046433a bellard
                mapping_t* m = array_get(&(s->mapping), j);
2421 a046433a bellard
                if (m->first_mapping_index < 0 && m != mapping &&
2422 a046433a bellard
                        !strncmp(m->path, mapping->path, parent_path_len) &&
2423 a046433a bellard
                        strlen(m->path) == parent_path_len)
2424 a046433a bellard
                    break;
2425 a046433a bellard
            }
2426 a046433a bellard
            assert(j < s->mapping.next);
2427 a046433a bellard
            mapping->info.dir.parent_mapping_index = j;
2428 a046433a bellard
2429 a046433a bellard
            array_remove(&(s->commits), i);
2430 a046433a bellard
            continue;
2431 a046433a bellard
        }
2432 a046433a bellard
2433 a046433a bellard
        i++;
2434 a046433a bellard
    }
2435 a046433a bellard
    return 0;
2436 a046433a bellard
}
2437 a046433a bellard
2438 a046433a bellard
/*
2439 a046433a bellard
 * TODO: make sure that the short name is not matching *another* file
2440 a046433a bellard
 */
2441 a046433a bellard
static int handle_commits(BDRVVVFATState* s)
2442 a046433a bellard
{
2443 a046433a bellard
    int i, fail = 0;
2444 a046433a bellard
2445 a046433a bellard
    vvfat_close_current_file(s);
2446 a046433a bellard
2447 a046433a bellard
    for (i = 0; !fail && i < s->commits.next; i++) {
2448 a046433a bellard
        commit_t* commit = array_get(&(s->commits), i);
2449 a046433a bellard
        switch(commit->action) {
2450 a046433a bellard
        case ACTION_RENAME: case ACTION_MKDIR:
2451 a046433a bellard
            assert(0);
2452 a046433a bellard
            fail = -2;
2453 a046433a bellard
            break;
2454 a046433a bellard
        case ACTION_WRITEOUT: {
2455 a046433a bellard
            direntry_t* entry = array_get(&(s->directory),
2456 a046433a bellard
                    commit->param.writeout.dir_index);
2457 a046433a bellard
            uint32_t begin = begin_of_direntry(entry);
2458 a046433a bellard
            mapping_t* mapping = find_mapping_for_cluster(s, begin);
2459 a046433a bellard
2460 a046433a bellard
            assert(mapping);
2461 a046433a bellard
            assert(mapping->begin == begin);
2462 a046433a bellard
            assert(commit->path == NULL);
2463 a046433a bellard
2464 a046433a bellard
            if (commit_one_file(s, commit->param.writeout.dir_index,
2465 a046433a bellard
                        commit->param.writeout.modified_offset))
2466 a046433a bellard
                fail = -3;
2467 a046433a bellard
2468 a046433a bellard
            break;
2469 a046433a bellard
        }
2470 a046433a bellard
        case ACTION_NEW_FILE: {
2471 a046433a bellard
            int begin = commit->param.new_file.first_cluster;
2472 a046433a bellard
            mapping_t* mapping = find_mapping_for_cluster(s, begin);
2473 a046433a bellard
            direntry_t* entry;
2474 a046433a bellard
            int i;
2475 de167e41 bellard
2476 a046433a bellard
            /* find direntry */
2477 a046433a bellard
            for (i = 0; i < s->directory.next; i++) {
2478 a046433a bellard
                entry = array_get(&(s->directory), i);
2479 a046433a bellard
                if (is_file(entry) && begin_of_direntry(entry) == begin)
2480 a046433a bellard
                    break;
2481 de167e41 bellard
            }
2482 de167e41 bellard
2483 a046433a bellard
            if (i >= s->directory.next) {
2484 a046433a bellard
                fail = -6;
2485 a046433a bellard
                continue;
2486 a046433a bellard
            }
2487 de167e41 bellard
2488 a046433a bellard
            /* make sure there exists an initial mapping */
2489 a046433a bellard
            if (mapping && mapping->begin != begin) {
2490 a046433a bellard
                mapping->end = begin;
2491 a046433a bellard
                mapping = NULL;
2492 a046433a bellard
            }
2493 a046433a bellard
            if (mapping == NULL) {
2494 a046433a bellard
                mapping = insert_mapping(s, begin, begin+1);
2495 a046433a bellard
            }
2496 a046433a bellard
            /* most members will be fixed in commit_mappings() */
2497 a046433a bellard
            assert(commit->path);
2498 a046433a bellard
            mapping->path = commit->path;
2499 a046433a bellard
            mapping->read_only = 0;
2500 a046433a bellard
            mapping->mode = MODE_NORMAL;
2501 a046433a bellard
            mapping->info.file.offset = 0;
2502 a046433a bellard
2503 a046433a bellard
            if (commit_one_file(s, i, 0))
2504 a046433a bellard
                fail = -7;
2505 a046433a bellard
2506 a046433a bellard
            break;
2507 a046433a bellard
        }
2508 a046433a bellard
        default:
2509 a046433a bellard
            assert(0);
2510 a046433a bellard
        }
2511 a046433a bellard
    }
2512 a046433a bellard
    if (i > 0 && array_remove_slice(&(s->commits), 0, i))
2513 a046433a bellard
        return -1;
2514 a046433a bellard
    return fail;
2515 a046433a bellard
}
2516 a046433a bellard
2517 a046433a bellard
static int handle_deletes(BDRVVVFATState* s)
2518 a046433a bellard
{
2519 a046433a bellard
    int i, deferred = 1, deleted = 1;
2520 a046433a bellard
2521 a046433a bellard
    /* delete files corresponding to mappings marked as deleted */
2522 a046433a bellard
    /* handle DELETEs and unused mappings (modified_fat_get(s, mapping->begin) == 0) */
2523 a046433a bellard
    while (deferred && deleted) {
2524 a046433a bellard
        deferred = 0;
2525 a046433a bellard
        deleted = 0;
2526 a046433a bellard
2527 a046433a bellard
        for (i = 1; i < s->mapping.next; i++) {
2528 a046433a bellard
            mapping_t* mapping = array_get(&(s->mapping), i);
2529 a046433a bellard
            if (mapping->mode & MODE_DELETED) {
2530 a046433a bellard
                direntry_t* entry = array_get(&(s->directory),
2531 a046433a bellard
                        mapping->dir_index);
2532 a046433a bellard
2533 a046433a bellard
                if (is_free(entry)) {
2534 a046433a bellard
                    /* remove file/directory */
2535 a046433a bellard
                    if (mapping->mode & MODE_DIRECTORY) {
2536 a046433a bellard
                        int j, next_dir_index = s->directory.next,
2537 a046433a bellard
                        first_dir_index = mapping->info.dir.first_dir_index;
2538 a046433a bellard
2539 a046433a bellard
                        if (rmdir(mapping->path) < 0) {
2540 a046433a bellard
                            if (errno == ENOTEMPTY) {
2541 a046433a bellard
                                deferred++;
2542 a046433a bellard
                                continue;
2543 a046433a bellard
                            } else
2544 a046433a bellard
                                return -5;
2545 de167e41 bellard
                        }
2546 a046433a bellard
2547 a046433a bellard
                        for (j = 1; j < s->mapping.next; j++) {
2548 a046433a bellard
                            mapping_t* m = array_get(&(s->mapping), j);
2549 a046433a bellard
                            if (m->mode & MODE_DIRECTORY &&
2550 a046433a bellard
                                    m->info.dir.first_dir_index >
2551 a046433a bellard
                                    first_dir_index &&
2552 a046433a bellard
                                    m->info.dir.first_dir_index <
2553 a046433a bellard
                                    next_dir_index)
2554 a046433a bellard
                                next_dir_index =
2555 a046433a bellard
                                    m->info.dir.first_dir_index;
2556 de167e41 bellard
                        }
2557 a046433a bellard
                        remove_direntries(s, first_dir_index,
2558 a046433a bellard
                                next_dir_index - first_dir_index);
2559 de167e41 bellard
2560 a046433a bellard
                        deleted++;
2561 de167e41 bellard
                    }
2562 a046433a bellard
                } else {
2563 a046433a bellard
                    if (unlink(mapping->path))
2564 a046433a bellard
                        return -4;
2565 a046433a bellard
                    deleted++;
2566 de167e41 bellard
                }
2567 a046433a bellard
                DLOG(fprintf(stderr, "DELETE (%d)\n", i); print_mapping(mapping); print_direntry(entry));
2568 a046433a bellard
                remove_mapping(s, i);
2569 de167e41 bellard
            }
2570 de167e41 bellard
        }
2571 de167e41 bellard
    }
2572 a046433a bellard
2573 a046433a bellard
    return 0;
2574 a046433a bellard
}
2575 a046433a bellard
2576 a046433a bellard
/*
2577 a046433a bellard
 * synchronize mapping with new state:
2578 a046433a bellard
 *
2579 a046433a bellard
 * - copy FAT (with bdrv_read)
2580 a046433a bellard
 * - mark all filenames corresponding to mappings as deleted
2581 a046433a bellard
 * - recurse direntries from root (using bs->bdrv_read)
2582 a046433a bellard
 * - delete files corresponding to mappings marked as deleted
2583 a046433a bellard
 */
2584 a046433a bellard
static int do_commit(BDRVVVFATState* s)
2585 a046433a bellard
{
2586 a046433a bellard
    int ret = 0;
2587 a046433a bellard
2588 a046433a bellard
    /* the real meat are the commits. Nothing to do? Move along! */
2589 a046433a bellard
    if (s->commits.next == 0)
2590 a046433a bellard
        return 0;
2591 a046433a bellard
2592 a046433a bellard
    vvfat_close_current_file(s);
2593 a046433a bellard
2594 a046433a bellard
    ret = handle_renames_and_mkdirs(s);
2595 a046433a bellard
    if (ret) {
2596 a046433a bellard
        fprintf(stderr, "Error handling renames (%d)\n", ret);
2597 a046433a bellard
        assert(0);
2598 a046433a bellard
        return ret;
2599 a046433a bellard
    }
2600 a046433a bellard
2601 5fafdf24 ths
    /* copy FAT (with bdrv_read) */
2602 a046433a bellard
    memcpy(s->fat.pointer, s->fat2, 0x200 * s->sectors_per_fat);
2603 a046433a bellard
2604 a046433a bellard
    /* recurse direntries from root (using bs->bdrv_read) */
2605 a046433a bellard
    ret = commit_direntries(s, 0, -1);
2606 a046433a bellard
    if (ret) {
2607 a046433a bellard
        fprintf(stderr, "Fatal: error while committing (%d)\n", ret);
2608 a046433a bellard
        assert(0);
2609 a046433a bellard
        return ret;
2610 a046433a bellard
    }
2611 a046433a bellard
2612 a046433a bellard
    ret = handle_commits(s);
2613 a046433a bellard
    if (ret) {
2614 a046433a bellard
        fprintf(stderr, "Error handling commits (%d)\n", ret);
2615 a046433a bellard
        assert(0);
2616 a046433a bellard
        return ret;
2617 a046433a bellard
    }
2618 a046433a bellard
2619 a046433a bellard
    ret = handle_deletes(s);
2620 a046433a bellard
    if (ret) {
2621 a046433a bellard
        fprintf(stderr, "Error deleting\n");
2622 a046433a bellard
        assert(0);
2623 a046433a bellard
        return ret;
2624 a046433a bellard
    }
2625 a046433a bellard
2626 a046433a bellard
    s->qcow->drv->bdrv_make_empty(s->qcow);
2627 a046433a bellard
2628 a046433a bellard
    memset(s->used_clusters, 0, sector2cluster(s, s->sector_count));
2629 a046433a bellard
2630 a046433a bellard
DLOG(checkpoint());
2631 a046433a bellard
    return 0;
2632 a046433a bellard
}
2633 a046433a bellard
2634 a046433a bellard
static int try_commit(BDRVVVFATState* s)
2635 a046433a bellard
{
2636 a046433a bellard
    vvfat_close_current_file(s);
2637 a046433a bellard
DLOG(checkpoint());
2638 a046433a bellard
    if(!is_consistent(s))
2639 a046433a bellard
        return -1;
2640 a046433a bellard
    return do_commit(s);
2641 a046433a bellard
}
2642 a046433a bellard
2643 5fafdf24 ths
static int vvfat_write(BlockDriverState *bs, int64_t sector_num,
2644 a046433a bellard
                    const uint8_t *buf, int nb_sectors)
2645 a046433a bellard
{
2646 5fafdf24 ths
    BDRVVVFATState *s = bs->opaque;
2647 a046433a bellard
    int i, ret;
2648 a046433a bellard
2649 a046433a bellard
DLOG(checkpoint());
2650 a046433a bellard
2651 a046433a bellard
    vvfat_close_current_file(s);
2652 a046433a bellard
2653 a046433a bellard
    /*
2654 a046433a bellard
     * Some sanity checks:
2655 a046433a bellard
     * - do not allow writing to the boot sector
2656 a046433a bellard
     * - do not allow to write non-ASCII filenames
2657 a046433a bellard
     */
2658 a046433a bellard
2659 a046433a bellard
    if (sector_num < s->first_sectors_number)
2660 a046433a bellard
        return -1;
2661 a046433a bellard
2662 a046433a bellard
    for (i = sector2cluster(s, sector_num);
2663 a046433a bellard
            i <= sector2cluster(s, sector_num + nb_sectors - 1);) {
2664 a046433a bellard
        mapping_t* mapping = find_mapping_for_cluster(s, i);
2665 a046433a bellard
        if (mapping) {
2666 a046433a bellard
            if (mapping->read_only) {
2667 a046433a bellard
                fprintf(stderr, "Tried to write to write-protected file %s\n",
2668 a046433a bellard
                        mapping->path);
2669 a046433a bellard
                return -1;
2670 a046433a bellard
            }
2671 a046433a bellard
2672 a046433a bellard
            if (mapping->mode & MODE_DIRECTORY) {
2673 a046433a bellard
                int begin = cluster2sector(s, i);
2674 a046433a bellard
                int end = begin + s->sectors_per_cluster, k;
2675 a046433a bellard
                int dir_index;
2676 a046433a bellard
                const direntry_t* direntries;
2677 a046433a bellard
                long_file_name lfn;
2678 a046433a bellard
2679 a046433a bellard
                lfn_init(&lfn);
2680 a046433a bellard
2681 a046433a bellard
                if (begin < sector_num)
2682 a046433a bellard
                    begin = sector_num;
2683 a046433a bellard
                if (end > sector_num + nb_sectors)
2684 a046433a bellard
                    end = sector_num + nb_sectors;
2685 5fafdf24 ths
                dir_index  = mapping->dir_index +
2686 a046433a bellard
                    0x10 * (begin - mapping->begin * s->sectors_per_cluster);
2687 a046433a bellard
                direntries = (direntry_t*)(buf + 0x200 * (begin - sector_num));
2688 a046433a bellard
2689 a046433a bellard
                for (k = 0; k < (end - begin) * 0x10; k++) {
2690 a046433a bellard
                    /* do not allow non-ASCII filenames */
2691 a046433a bellard
                    if (parse_long_name(&lfn, direntries + k) < 0) {
2692 a046433a bellard
                        fprintf(stderr, "Warning: non-ASCII filename\n");
2693 a046433a bellard
                        return -1;
2694 a046433a bellard
                    }
2695 a046433a bellard
                    /* no access to the direntry of a read-only file */
2696 a046433a bellard
                    else if (is_short_name(direntries+k) &&
2697 a046433a bellard
                            (direntries[k].attributes & 1)) {
2698 a046433a bellard
                        if (memcmp(direntries + k,
2699 a046433a bellard
                                    array_get(&(s->directory), dir_index + k),
2700 a046433a bellard
                                    sizeof(direntry_t))) {
2701 a046433a bellard
                            fprintf(stderr, "Warning: tried to write to write-protected file\n");
2702 a046433a bellard
                            return -1;
2703 a046433a bellard
                        }
2704 a046433a bellard
                    }
2705 a046433a bellard
                }
2706 a046433a bellard
            }
2707 a046433a bellard
            i = mapping->end;
2708 a046433a bellard
        } else
2709 a046433a bellard
            i++;
2710 a046433a bellard
    }
2711 a046433a bellard
2712 a046433a bellard
    /*
2713 a046433a bellard
     * Use qcow backend. Commit later.
2714 a046433a bellard
     */
2715 a046433a bellard
DLOG(fprintf(stderr, "Write to qcow backend: %d + %d\n", (int)sector_num, nb_sectors));
2716 a046433a bellard
    ret = s->qcow->drv->bdrv_write(s->qcow, sector_num, buf, nb_sectors);
2717 a046433a bellard
    if (ret < 0) {
2718 a046433a bellard
        fprintf(stderr, "Error writing to qcow backend\n");
2719 a046433a bellard
        return ret;
2720 a046433a bellard
    }
2721 a046433a bellard
2722 a046433a bellard
    for (i = sector2cluster(s, sector_num);
2723 a046433a bellard
            i <= sector2cluster(s, sector_num + nb_sectors - 1); i++)
2724 a046433a bellard
        if (i >= 0)
2725 a046433a bellard
            s->used_clusters[i] |= USED_ALLOCATED;
2726 a046433a bellard
2727 a046433a bellard
DLOG(checkpoint());
2728 a046433a bellard
    /* TODO: add timeout */
2729 a046433a bellard
    try_commit(s);
2730 a046433a bellard
2731 a046433a bellard
DLOG(checkpoint());
2732 a046433a bellard
    return 0;
2733 a046433a bellard
}
2734 a046433a bellard
2735 a046433a bellard
static int vvfat_is_allocated(BlockDriverState *bs,
2736 a046433a bellard
        int64_t sector_num, int nb_sectors, int* n)
2737 a046433a bellard
{
2738 a046433a bellard
    BDRVVVFATState* s = bs->opaque;
2739 a046433a bellard
    *n = s->sector_count - sector_num;
2740 a046433a bellard
    if (*n > nb_sectors)
2741 a046433a bellard
        *n = nb_sectors;
2742 a046433a bellard
    else if (*n < 0)
2743 a046433a bellard
        return 0;
2744 5fafdf24 ths
    return 1;
2745 a046433a bellard
}
2746 a046433a bellard
2747 a046433a bellard
static int write_target_commit(BlockDriverState *bs, int64_t sector_num,
2748 a046433a bellard
        const uint8_t* buffer, int nb_sectors) {
2749 a046433a bellard
    BDRVVVFATState* s = bs->opaque;
2750 a046433a bellard
    return try_commit(s);
2751 a046433a bellard
}
2752 a046433a bellard
2753 a046433a bellard
static void write_target_close(BlockDriverState *bs) {
2754 a046433a bellard
    BDRVVVFATState* s = bs->opaque;
2755 a046433a bellard
    bdrv_delete(s->qcow);
2756 a046433a bellard
    free(s->qcow_filename);
2757 a046433a bellard
}
2758 a046433a bellard
2759 a046433a bellard
static BlockDriver vvfat_write_target = {
2760 a046433a bellard
    "vvfat_write_target", 0, NULL, NULL, NULL,
2761 a046433a bellard
    write_target_commit,
2762 a046433a bellard
    write_target_close,
2763 a046433a bellard
    NULL, NULL, NULL
2764 a046433a bellard
};
2765 a046433a bellard
2766 a046433a bellard
static int enable_write_target(BDRVVVFATState *s)
2767 a046433a bellard
{
2768 a046433a bellard
    int size = sector2cluster(s, s->sector_count);
2769 a046433a bellard
    s->used_clusters = calloc(size, 1);
2770 a046433a bellard
2771 a046433a bellard
    array_init(&(s->commits), sizeof(commit_t));
2772 a046433a bellard
2773 a046433a bellard
    s->qcow_filename = malloc(1024);
2774 83f64091 bellard
    get_tmp_filename(s->qcow_filename, 1024);
2775 a046433a bellard
    if (bdrv_create(&bdrv_qcow,
2776 a046433a bellard
                s->qcow_filename, s->sector_count, "fat:", 0) < 0)
2777 a046433a bellard
        return -1;
2778 a046433a bellard
    s->qcow = bdrv_new("");
2779 a046433a bellard
    if (s->qcow == NULL || bdrv_open(s->qcow, s->qcow_filename, 0) < 0)
2780 a046433a bellard
        return -1;
2781 a046433a bellard
2782 a046433a bellard
#ifndef _WIN32
2783 a046433a bellard
    unlink(s->qcow_filename);
2784 a046433a bellard
#endif
2785 a046433a bellard
2786 a046433a bellard
    s->bs->backing_hd = calloc(sizeof(BlockDriverState), 1);
2787 a046433a bellard
    s->bs->backing_hd->drv = &vvfat_write_target;
2788 a046433a bellard
    s->bs->backing_hd->opaque = s;
2789 a046433a bellard
2790 de167e41 bellard
    return 0;
2791 de167e41 bellard
}
2792 de167e41 bellard
2793 de167e41 bellard
static void vvfat_close(BlockDriverState *bs)
2794 de167e41 bellard
{
2795 de167e41 bellard
    BDRVVVFATState *s = bs->opaque;
2796 de167e41 bellard
2797 de167e41 bellard
    vvfat_close_current_file(s);
2798 de167e41 bellard
    array_free(&(s->fat));
2799 de167e41 bellard
    array_free(&(s->directory));
2800 de167e41 bellard
    array_free(&(s->mapping));
2801 a046433a bellard
    if(s->cluster_buffer)
2802 a046433a bellard
        free(s->cluster_buffer);
2803 de167e41 bellard
}
2804 de167e41 bellard
2805 de167e41 bellard
BlockDriver bdrv_vvfat = {
2806 de167e41 bellard
    "vvfat",
2807 de167e41 bellard
    sizeof(BDRVVVFATState),
2808 83f64091 bellard
    NULL, /* no probe for protocols */
2809 de167e41 bellard
    vvfat_open,
2810 de167e41 bellard
    vvfat_read,
2811 de167e41 bellard
    vvfat_write,
2812 de167e41 bellard
    vvfat_close,
2813 7a6cba61 pbrook
    NULL, /* ??? Not sure if we can do any meaningful flushing.  */
2814 a046433a bellard
    NULL,
2815 83f64091 bellard
    vvfat_is_allocated,
2816 83f64091 bellard
    .protocol_name = "fat",
2817 de167e41 bellard
};
2818 de167e41 bellard
2819 a046433a bellard
#ifdef DEBUG
2820 3f47aa8c blueswir1
static void checkpoint(void) {
2821 a046433a bellard
    assert(((mapping_t*)array_get(&(vvv->mapping), 0))->end == 2);
2822 a046433a bellard
    check1(vvv);
2823 a046433a bellard
    check2(vvv);
2824 a046433a bellard
    assert(!vvv->current_mapping || vvv->current_fd || (vvv->current_mapping->mode & MODE_DIRECTORY));
2825 a046433a bellard
#if 0
2826 a046433a bellard
    if (((direntry_t*)vvv->directory.pointer)[1].attributes != 0xf)
2827 a046433a bellard
        fprintf(stderr, "Nonono!\n");
2828 a046433a bellard
    mapping_t* mapping;
2829 a046433a bellard
    direntry_t* direntry;
2830 a046433a bellard
    assert(vvv->mapping.size >= vvv->mapping.item_size * vvv->mapping.next);
2831 a046433a bellard
    assert(vvv->directory.size >= vvv->directory.item_size * vvv->directory.next);
2832 a046433a bellard
    if (vvv->mapping.next<47)
2833 a046433a bellard
        return;
2834 a046433a bellard
    assert((mapping = array_get(&(vvv->mapping), 47)));
2835 a046433a bellard
    assert(mapping->dir_index < vvv->directory.next);
2836 a046433a bellard
    direntry = array_get(&(vvv->directory), mapping->dir_index);
2837 a046433a bellard
    assert(!memcmp(direntry->name, "USB     H  ", 11) || direntry->name[0]==0);
2838 a046433a bellard
#endif
2839 a046433a bellard
    return;
2840 a046433a bellard
    /* avoid compiler warnings: */
2841 a046433a bellard
    hexdump(NULL, 100);
2842 a046433a bellard
    remove_mapping(vvv, NULL);
2843 a046433a bellard
    print_mapping(NULL);
2844 a046433a bellard
    print_direntry(NULL);
2845 a046433a bellard
}
2846 a046433a bellard
#endif