1 /* $NetBSD: cd9660.c,v 1.26 2009/01/16 18:02:24 pooka Exp $ */ 2 3 /* 4 * Copyright (c) 2005 Daniel Watt, Walter Deignan, Ryan Gabrys, Alan 5 * Perez-Rathke and Ram Vedam. All rights reserved. 6 * 7 * This code was written by Daniel Watt, Walter Deignan, Ryan Gabrys, 8 * Alan Perez-Rathke and Ram Vedam. 9 * 10 * Redistribution and use in source and binary forms, with or 11 * without modification, are permitted provided that the following 12 * conditions are met: 13 * 1. Redistributions of source code must retain the above copyright 14 * notice, this list of conditions and the following disclaimer. 15 * 2. Redistributions in binary form must reproduce the above 16 * copyright notice, this list of conditions and the following 17 * disclaimer in the documentation and/or other materials provided 18 * with the distribution. 19 * 20 * THIS SOFTWARE IS PROVIDED BY DANIEL WATT, WALTER DEIGNAN, RYAN 21 * GABRYS, ALAN PEREZ-RATHKE AND RAM VEDAM ``AS IS'' AND ANY EXPRESS OR 22 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 23 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 24 * DISCLAIMED. IN NO EVENT SHALL DANIEL WATT, WALTER DEIGNAN, RYAN 25 * GABRYS, ALAN PEREZ-RATHKE AND RAM VEDAM BE LIABLE FOR ANY DIRECT, INDIRECT, 26 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 27 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF 28 * USE,DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 29 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, 30 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 31 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY 32 * OF SUCH DAMAGE. 33 */ 34 /* 35 * Copyright (c) 2001 Wasabi Systems, Inc. 36 * All rights reserved. 37 * 38 * Written by Luke Mewburn for Wasabi Systems, Inc. 39 * 40 * Redistribution and use in source and binary forms, with or without 41 * modification, are permitted provided that the following conditions 42 * are met: 43 * 1. Redistributions of source code must retain the above copyright 44 * notice, this list of conditions and the following disclaimer. 45 * 2. Redistributions in binary form must reproduce the above copyright 46 * notice, this list of conditions and the following disclaimer in the 47 * documentation and/or other materials provided with the distribution. 48 * 3. All advertising materials mentioning features or use of this software 49 * must display the following acknowledgement: 50 * This product includes software developed for the NetBSD Project by 51 * Wasabi Systems, Inc. 52 * 4. The name of Wasabi Systems, Inc. may not be used to endorse 53 * or promote products derived from this software without specific prior 54 * written permission. 55 * 56 * THIS SOFTWARE IS PROVIDED BY WASABI SYSTEMS, INC. ``AS IS'' AND 57 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 58 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 59 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL WASABI SYSTEMS, INC 60 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 61 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 62 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 63 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 64 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 65 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 66 * POSSIBILITY OF SUCH DAMAGE. 67 */ 68 /* 69 * Copyright (c) 1982, 1986, 1989, 1993 70 * The Regents of the University of California. All rights reserved. 71 * 72 * Redistribution and use in source and binary forms, with or without 73 * modification, are permitted provided that the following conditions 74 * are met: 75 * 1. Redistributions of source code must retain the above copyright 76 * notice, this list of conditions and the following disclaimer. 77 * 2. Redistributions in binary form must reproduce the above copyright 78 * notice, this list of conditions and the following disclaimer in the 79 * documentation and/or other materials provided with the distribution. 80 * 3. Neither the name of the University nor the names of its contributors 81 * may be used to endorse or promote products derived from this software 82 * without specific prior written permission. 83 * 84 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 85 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 86 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 87 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 88 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 89 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 90 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 91 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 92 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 93 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 94 * SUCH DAMAGE. 95 * 96 */ 97 98 #if HAVE_NBTOOL_CONFIG_H 99 #include "nbtool_config.h" 100 #else 101 #include <sys/mount.h> 102 #endif 103 104 #include <sys/cdefs.h> 105 #if defined(__RCSID) && !defined(__lint) 106 __RCSID("$NetBSD: cd9660.c,v 1.26 2009/01/16 18:02:24 pooka Exp $"); 107 #endif /* !__lint */ 108 109 #include <string.h> 110 #include <ctype.h> 111 #include <sys/param.h> 112 #include <sys/queue.h> 113 114 #include "makefs.h" 115 #include "cd9660.h" 116 #include "cd9660/iso9660_rrip.h" 117 #include "cd9660/cd9660_archimedes.h" 118 119 /* 120 * Global variables 121 */ 122 iso9660_disk diskStructure; 123 124 static void cd9660_finalize_PVD(void); 125 static cd9660node *cd9660_allocate_cd9660node(void); 126 static void cd9660_set_defaults(void); 127 static int cd9660_arguments_set_string(const char *, const char *, int, 128 char, char *); 129 static void cd9660_populate_iso_dir_record( 130 struct _iso_directory_record_cd9660 *, u_char, u_char, u_char, 131 const char *); 132 static void cd9660_setup_root_node(void); 133 static int cd9660_setup_volume_descriptors(void); 134 #if 0 135 static int cd9660_fill_extended_attribute_record(cd9660node *); 136 #endif 137 static void cd9660_sort_nodes(cd9660node *); 138 static int cd9960_translate_node_common(cd9660node *); 139 static int cd9660_translate_node(fsnode *, cd9660node *); 140 static int cd9660_compare_filename(const char *, const char *); 141 static void cd9660_sorted_child_insert(cd9660node *, cd9660node *); 142 static int cd9660_handle_collisions(cd9660node *, int); 143 static cd9660node *cd9660_rename_filename(cd9660node *, int, int); 144 static void cd9660_copy_filenames(cd9660node *); 145 static void cd9660_sorting_nodes(cd9660node *); 146 static int cd9660_count_collisions(cd9660node *); 147 static cd9660node *cd9660_rrip_move_directory(cd9660node *); 148 static int cd9660_add_dot_records(cd9660node *); 149 150 static void cd9660_convert_structure(fsnode *, cd9660node *, int, 151 int *, int *); 152 static void cd9660_free_structure(cd9660node *); 153 static int cd9660_generate_path_table(void); 154 static int cd9660_level1_convert_filename(const char *, char *, int); 155 static int cd9660_level2_convert_filename(const char *, char *, int); 156 #if 0 157 static int cd9660_joliet_convert_filename(const char *, char *, int); 158 #endif 159 static int cd9660_convert_filename(const char *, char *, int); 160 static void cd9660_populate_dot_records(cd9660node *); 161 static int cd9660_compute_offsets(cd9660node *, int); 162 #if 0 163 static int cd9660_copy_stat_info(cd9660node *, cd9660node *, int); 164 #endif 165 static cd9660node *cd9660_create_virtual_entry(const char *, cd9660node *, int, 166 int); 167 static cd9660node *cd9660_create_file(const char *, cd9660node *, cd9660node *); 168 static cd9660node *cd9660_create_directory(const char *, cd9660node *, 169 cd9660node *); 170 static cd9660node *cd9660_create_special_directory(u_char, cd9660node *); 171 172 173 /* 174 * Allocate and initalize a cd9660node 175 * @returns struct cd9660node * Pointer to new node, or NULL on error 176 */ 177 static cd9660node * 178 cd9660_allocate_cd9660node(void) 179 { 180 cd9660node *temp; 181 182 if ((temp = calloc(1, sizeof(cd9660node))) == NULL) 183 err(EXIT_FAILURE, "%s: calloc", __func__); 184 TAILQ_INIT(&temp->cn_children); 185 temp->parent = temp->dot_record = temp->dot_dot_record = NULL; 186 temp->ptnext = temp->ptprev = temp->ptlast = NULL; 187 temp->node = NULL; 188 temp->isoDirRecord = NULL; 189 temp->isoExtAttributes = NULL; 190 temp->rr_real_parent = temp->rr_relocated = NULL; 191 temp->su_tail_data = NULL; 192 return temp; 193 } 194 195 int cd9660_defaults_set = 0; 196 197 /** 198 * Set default values for cd9660 extension to makefs 199 */ 200 static void 201 cd9660_set_defaults(void) 202 { 203 /*Fix the sector size for now, though the spec allows for other sizes*/ 204 diskStructure.sectorSize = 2048; 205 206 /* Set up defaults in our own structure */ 207 diskStructure.verbose_level = 0; 208 diskStructure.keep_bad_images = 0; 209 diskStructure.follow_sym_links = 0; 210 diskStructure.isoLevel = 2; 211 212 diskStructure.rock_ridge_enabled = 0; 213 diskStructure.rock_ridge_renamed_dir_name = 0; 214 diskStructure.rock_ridge_move_count = 0; 215 diskStructure.rr_moved_dir = 0; 216 217 diskStructure.archimedes_enabled = 0; 218 219 diskStructure.include_padding_areas = 1; 220 221 /* Spec breaking functionality */ 222 diskStructure.allow_deep_trees = 223 diskStructure.allow_start_dot = 224 diskStructure.allow_max_name = 225 diskStructure.allow_illegal_chars = 226 diskStructure.allow_lowercase = 227 diskStructure.allow_multidot = 228 diskStructure.omit_trailing_period = 0; 229 230 /* Make sure the PVD is clear */ 231 memset(&diskStructure.primaryDescriptor, 0, 2048); 232 233 memset(diskStructure.primaryDescriptor.volume_set_id, 0x20,32); 234 memset(diskStructure.primaryDescriptor.publisher_id, 0x20,128); 235 memset(diskStructure.primaryDescriptor.preparer_id, 0x20,128); 236 memset(diskStructure.primaryDescriptor.application_id, 0x20,128); 237 memset(diskStructure.primaryDescriptor.copyright_file_id, 0x20,128); 238 memset(diskStructure.primaryDescriptor.abstract_file_id, 0x20,128); 239 memset(diskStructure.primaryDescriptor.bibliographic_file_id, 0x20,128); 240 241 strcpy(diskStructure.primaryDescriptor.system_id,"NetBSD"); 242 243 cd9660_defaults_set = 1; 244 245 /* Boot support: Initially disabled */ 246 diskStructure.has_generic_bootimage = 0; 247 diskStructure.generic_bootimage = NULL; 248 249 diskStructure.boot_image_directory = 0; 250 /*memset(diskStructure.boot_descriptor, 0, 2048);*/ 251 252 diskStructure.is_bootable = 0; 253 TAILQ_INIT(&diskStructure.boot_images); 254 LIST_INIT(&diskStructure.boot_entries); 255 } 256 257 void 258 cd9660_prep_opts(fsinfo_t *fsopts __unused) 259 { 260 cd9660_set_defaults(); 261 } 262 263 void 264 cd9660_cleanup_opts(fsinfo_t *fsopts __unused) 265 { 266 267 } 268 269 static int 270 cd9660_arguments_set_string(const char *val, const char *fieldtitle, int length, 271 char testmode, char * dest) 272 { 273 int len, test; 274 275 if (val == NULL) 276 warnx("error: The %s requires a string argument", fieldtitle); 277 else if ((len = strlen(val)) <= length) { 278 if (testmode == 'd') 279 test = cd9660_valid_d_chars(val); 280 else 281 test = cd9660_valid_a_chars(val); 282 if (test) { 283 memcpy(dest, val, len); 284 if (test == 2) 285 cd9660_uppercase_characters(dest, len); 286 return 1; 287 } else 288 warnx("error: The %s must be composed of " 289 "%c-characters", fieldtitle, testmode); 290 } else 291 warnx("error: The %s must be at most 32 characters long", 292 fieldtitle); 293 return 0; 294 } 295 296 /* 297 * Command-line parsing function 298 */ 299 300 int 301 cd9660_parse_opts(const char *option, fsinfo_t *fsopts) 302 { 303 char *var, *val; 304 int rv; 305 /* Set up allowed options - integer options ONLY */ 306 option_t cd9660_options[] = { 307 { "l", &diskStructure.isoLevel, 1, 3, "ISO Level" }, 308 { "isolevel", &diskStructure.isoLevel, 1, 3, "ISO Level" }, 309 { "verbose", &diskStructure.verbose_level, 0, 2, 310 "Turns on verbose output" }, 311 { "v", &diskStructure.verbose_level, 0 , 2, 312 "Turns on verbose output"}, 313 { .name = NULL } 314 }; 315 316 if (cd9660_defaults_set == 0) 317 cd9660_set_defaults(); 318 319 /* 320 * Todo : finish implementing this, and make a function that 321 * parses them 322 */ 323 /* 324 string_option_t cd9660_string_options[] = { 325 { "L", "Label", &diskStructure.primaryDescriptor.volume_id, 1, 32, "Disk Label", ISO_STRING_FILTER_DCHARS }, 326 { NULL } 327 } 328 */ 329 330 assert(option != NULL); 331 332 if (debug & DEBUG_FS_PARSE_OPTS) 333 printf("cd9660_parse_opts: got `%s'\n", option); 334 335 if ((var = strdup(option)) == NULL) 336 err(1, "allocating memory for copy of option string"); 337 rv = 1; 338 339 val = strchr(var, '='); 340 if (val != NULL) 341 *val++ = '\0'; 342 343 /* First handle options with no parameters */ 344 if (strcmp(var, "h") == 0) { 345 diskStructure.displayHelp = 1; 346 rv = 1; 347 } else if (CD9660_IS_COMMAND_ARG_DUAL(var, "S", "follow-symlinks")) { 348 /* this is not handled yet */ 349 diskStructure.follow_sym_links = 1; 350 rv = 1; 351 } else if (CD9660_IS_COMMAND_ARG_DUAL(var, "L", "label")) { 352 rv = cd9660_arguments_set_string(val, "Disk Label", 32, 'd', 353 diskStructure.primaryDescriptor.volume_id); 354 } else if (CD9660_IS_COMMAND_ARG_DUAL(var, "A", "applicationid")) { 355 rv = cd9660_arguments_set_string(val, "Application Identifier", 128, 'a', 356 diskStructure.primaryDescriptor.application_id); 357 } else if(CD9660_IS_COMMAND_ARG_DUAL(var, "P", "publisher")) { 358 rv = cd9660_arguments_set_string(val, "Publisher Identifier", 359 128, 'a', diskStructure.primaryDescriptor.publisher_id); 360 } else if (CD9660_IS_COMMAND_ARG_DUAL(var, "p", "preparer")) { 361 rv = cd9660_arguments_set_string(val, "Preparer Identifier", 362 128, 'a', diskStructure.primaryDescriptor.preparer_id); 363 } else if (CD9660_IS_COMMAND_ARG_DUAL(var, "V", "volumeid")) { 364 rv = cd9660_arguments_set_string(val, "Volume Set Identifier", 365 128, 'a', diskStructure.primaryDescriptor.volume_set_id); 366 /* Boot options */ 367 } else if (CD9660_IS_COMMAND_ARG_DUAL(var, "B", "bootimage")) { 368 if (val == NULL) 369 warnx("error: The Boot Image parameter requires a valid boot information string"); 370 else 371 rv = cd9660_add_boot_disk(val); 372 } else if (CD9660_IS_COMMAND_ARG(var, "bootimagedir")) { 373 /* 374 * XXXfvdl this is unused. 375 */ 376 if (val == NULL) 377 errx(1, "error: The Boot Image Directory parameter" 378 " requires a directory name\n"); 379 else { 380 if ((diskStructure.boot_image_directory = 381 malloc(strlen(val) + 1)) == NULL) { 382 CD9660_MEM_ALLOC_ERROR("cd9660_parse_opts"); 383 exit(1); 384 } 385 386 /* BIG TODO: Add the max length function here */ 387 cd9660_arguments_set_string(val, "Boot Image Directory", 388 12 , 'd', diskStructure.boot_image_directory); 389 } 390 } else if (CD9660_IS_COMMAND_ARG_DUAL(var, "G", "generic-bootimage")) { 391 if (val == NULL) 392 warnx("error: The Boot Image parameter requires a valid boot information string"); 393 else 394 rv = cd9660_add_generic_bootimage(val); 395 } else if (CD9660_IS_COMMAND_ARG(var, "no-trailing-padding")) 396 diskStructure.include_padding_areas = 0; 397 /* RRIP */ 398 else if (CD9660_IS_COMMAND_ARG_DUAL(var, "R", "rockridge")) 399 diskStructure.rock_ridge_enabled = 1; 400 else if (CD9660_IS_COMMAND_ARG_DUAL(var, "A", "archimedes")) 401 diskStructure.archimedes_enabled = 1; 402 else if (CD9660_IS_COMMAND_ARG_DUAL(var, "K", "keep-bad-images")) 403 diskStructure.keep_bad_images = 1; 404 else if (CD9660_IS_COMMAND_ARG(var, "allow-deep-trees")) 405 diskStructure.allow_deep_trees = 1; 406 else if (CD9660_IS_COMMAND_ARG(var, "allow-max-name")) 407 diskStructure.allow_max_name = 1; 408 else if (CD9660_IS_COMMAND_ARG(var, "allow-illegal-chars")) 409 diskStructure.allow_illegal_chars = 1; 410 else if (CD9660_IS_COMMAND_ARG(var, "allow-lowercase")) 411 diskStructure.allow_lowercase = 1; 412 else if (CD9660_IS_COMMAND_ARG(var,"allow-multidot")) 413 diskStructure.allow_multidot = 1; 414 else if (CD9660_IS_COMMAND_ARG(var, "omit-trailing-period")) 415 diskStructure.omit_trailing_period = 1; 416 else if (CD9660_IS_COMMAND_ARG(var, "no-emul-boot") || 417 CD9660_IS_COMMAND_ARG(var, "no-boot") || 418 CD9660_IS_COMMAND_ARG(var, "hard-disk-boot")) { 419 cd9660_eltorito_add_boot_option(var, 0); 420 421 /* End of flag variables */ 422 } else if (CD9660_IS_COMMAND_ARG(var, "boot-load-segment")) { 423 if (val == NULL) { 424 warnx("Option `%s' doesn't contain a value", var); 425 rv = 0; 426 } else { 427 cd9660_eltorito_add_boot_option(var, val); 428 } 429 } else { 430 if (val == NULL) { 431 warnx("Option `%s' doesn't contain a value", var); 432 rv = 0; 433 } else 434 rv = set_option(cd9660_options, var, val); 435 } 436 437 if (var) 438 free(var); 439 return (rv); 440 } 441 442 /* 443 * Main function for cd9660_makefs 444 * Builds the ISO image file 445 * @param const char *image The image filename to create 446 * @param const char *dir The directory that is being read 447 * @param struct fsnode *root The root node of the filesystem tree 448 * @param struct fsinfo_t *fsopts Any options 449 */ 450 void 451 cd9660_makefs(const char *image, const char *dir, fsnode *root, 452 fsinfo_t *fsopts) 453 { 454 int startoffset; 455 int numDirectories; 456 int pathTableSectors; 457 int firstAvailableSector; 458 int totalSpace; 459 int error; 460 cd9660node *real_root; 461 462 if (diskStructure.verbose_level > 0) 463 printf("cd9660_makefs: ISO level is %i\n", 464 diskStructure.isoLevel); 465 if (diskStructure.isoLevel < 2 && 466 diskStructure.allow_multidot) 467 errx(1, "allow-multidot requires iso level of 2\n"); 468 469 assert(image != NULL); 470 assert(dir != NULL); 471 assert(root != NULL); 472 473 if (diskStructure.displayHelp) { 474 /* 475 * Display help here - probably want to put it in 476 * a separate function 477 */ 478 return; 479 } 480 481 diskStructure.rootFilesystemPath = dir; 482 483 if (diskStructure.verbose_level > 0) 484 printf("cd9660_makefs: image %s directory %s root %p\n", 485 image, dir, root); 486 487 /* Set up some constants. Later, these will be defined with options */ 488 489 /* Counter needed for path tables */ 490 numDirectories = 0; 491 492 /* Convert tree to our own format */ 493 /* Actually, we now need to add the REAL root node, at level 0 */ 494 495 real_root = cd9660_allocate_cd9660node(); 496 if ((real_root->isoDirRecord = 497 malloc( sizeof(iso_directory_record_cd9660) )) == NULL) { 498 CD9660_MEM_ALLOC_ERROR("cd9660_makefs"); 499 exit(1); 500 } 501 502 /* Leave filename blank for root */ 503 memset(real_root->isoDirRecord->name, 0, 504 ISO_FILENAME_MAXLENGTH_WITH_PADDING); 505 506 real_root->level = 0; 507 diskStructure.rootNode = real_root; 508 real_root->type = CD9660_TYPE_DIR; 509 error = 0; 510 real_root->node = root; 511 cd9660_convert_structure(root, real_root, 1, &numDirectories, &error); 512 513 if (TAILQ_EMPTY(&real_root->cn_children)) { 514 errx(1, "cd9660_makefs: converted directory is empty. " 515 "Tree conversion failed\n"); 516 } else if (error != 0) { 517 errx(1, "cd9660_makefs: tree conversion failed\n"); 518 } else { 519 if (diskStructure.verbose_level > 0) 520 printf("cd9660_makefs: tree converted\n"); 521 } 522 523 /* Add the dot and dot dot records */ 524 cd9660_add_dot_records(real_root); 525 526 cd9660_setup_root_node(); 527 528 if (diskStructure.verbose_level > 0) 529 printf("cd9660_makefs: done converting tree\n"); 530 531 /* non-SUSP extensions */ 532 if (diskStructure.archimedes_enabled) 533 archimedes_convert_tree(diskStructure.rootNode); 534 535 /* Rock ridge / SUSP init pass */ 536 if (diskStructure.rock_ridge_enabled) { 537 cd9660_susp_initialize(diskStructure.rootNode, 538 diskStructure.rootNode, NULL); 539 } 540 541 /* Build path table structure */ 542 diskStructure.pathTableLength = cd9660_generate_path_table(); 543 544 pathTableSectors = CD9660_BLOCKS(diskStructure.sectorSize, 545 diskStructure.pathTableLength); 546 547 firstAvailableSector = cd9660_setup_volume_descriptors(); 548 if (diskStructure.is_bootable) { 549 firstAvailableSector = cd9660_setup_boot(firstAvailableSector); 550 if (firstAvailableSector < 0) 551 errx(1, "setup_boot failed"); 552 } 553 /* LE first, then BE */ 554 diskStructure.primaryLittleEndianTableSector = firstAvailableSector; 555 diskStructure.primaryBigEndianTableSector = 556 diskStructure.primaryLittleEndianTableSector + pathTableSectors; 557 558 /* Set the secondary ones to -1, not going to use them for now */ 559 diskStructure.secondaryBigEndianTableSector = -1; 560 diskStructure.secondaryLittleEndianTableSector = -1; 561 562 diskStructure.dataFirstSector = 563 diskStructure.primaryBigEndianTableSector + pathTableSectors; 564 if (diskStructure.verbose_level > 0) 565 printf("cd9660_makefs: Path table conversion complete. " 566 "Each table is %i bytes, or %i sectors.\n", 567 diskStructure.pathTableLength, pathTableSectors); 568 569 startoffset = diskStructure.sectorSize*diskStructure.dataFirstSector; 570 571 totalSpace = cd9660_compute_offsets(real_root, startoffset); 572 573 diskStructure.totalSectors = diskStructure.dataFirstSector + 574 CD9660_BLOCKS(diskStructure.sectorSize, totalSpace); 575 576 /* Disabled until pass 1 is done */ 577 if (diskStructure.rock_ridge_enabled) { 578 diskStructure.susp_continuation_area_start_sector = 579 diskStructure.totalSectors; 580 diskStructure.totalSectors += 581 CD9660_BLOCKS(diskStructure.sectorSize, 582 diskStructure.susp_continuation_area_size); 583 cd9660_susp_finalize(diskStructure.rootNode); 584 } 585 586 587 cd9660_finalize_PVD(); 588 589 /* Add padding sectors, just for testing purposes right now */ 590 /* diskStructure.totalSectors+=150; */ 591 592 /* Debugging output */ 593 if (diskStructure.verbose_level > 0) { 594 printf("cd9660_makefs: Sectors 0-15 reserved\n"); 595 printf("cd9660_makefs: Primary path tables starts in sector %i\n", 596 diskStructure.primaryLittleEndianTableSector); 597 printf("cd9660_makefs: File data starts in sector %i\n", 598 diskStructure.dataFirstSector); 599 printf("cd9660_makefs: Total sectors: %i\n",diskStructure.totalSectors); 600 } 601 602 /* 603 * Add padding sectors at the end 604 * TODO: Clean this up and separate padding 605 */ 606 if (diskStructure.include_padding_areas) 607 diskStructure.totalSectors += 150; 608 609 cd9660_write_image(image); 610 611 if (diskStructure.verbose_level > 1) { 612 debug_print_volume_descriptor_information(); 613 debug_print_tree(real_root,0); 614 debug_print_path_tree(real_root); 615 } 616 617 /* Clean up data structures */ 618 cd9660_free_structure(real_root); 619 620 if (diskStructure.verbose_level > 0) 621 printf("cd9660_makefs: done\n"); 622 } 623 624 /* Generic function pointer - implement later */ 625 typedef int (*cd9660node_func)(cd9660node *); 626 627 static void 628 cd9660_finalize_PVD(void) 629 { 630 time_t tim; 631 unsigned char *temp; 632 633 /* Copy the root directory record */ 634 temp = (unsigned char *) &diskStructure.primaryDescriptor; 635 636 /* root should be a fixed size of 34 bytes since it has no name */ 637 memcpy(diskStructure.primaryDescriptor.root_directory_record, 638 diskStructure.rootNode->dot_record->isoDirRecord, 34); 639 640 /* In RRIP, this might be longer than 34 */ 641 diskStructure.primaryDescriptor.root_directory_record[0] = 34; 642 643 /* Set up all the important numbers in the PVD */ 644 cd9660_bothendian_dword(diskStructure.totalSectors, 645 (unsigned char *)diskStructure.primaryDescriptor.volume_space_size); 646 cd9660_bothendian_word(1, 647 (unsigned char *)diskStructure.primaryDescriptor.volume_set_size); 648 cd9660_bothendian_word(1, 649 (unsigned char *) 650 diskStructure.primaryDescriptor.volume_sequence_number); 651 cd9660_bothendian_word(diskStructure.sectorSize, 652 (unsigned char *) 653 diskStructure.primaryDescriptor.logical_block_size); 654 cd9660_bothendian_dword(diskStructure.pathTableLength, 655 (unsigned char *)diskStructure.primaryDescriptor.path_table_size); 656 657 cd9660_731(diskStructure.primaryLittleEndianTableSector, 658 (u_char *)diskStructure.primaryDescriptor.type_l_path_table); 659 cd9660_732(diskStructure.primaryBigEndianTableSector, 660 (u_char *)diskStructure.primaryDescriptor.type_m_path_table); 661 662 diskStructure.primaryDescriptor.file_structure_version[0] = 1; 663 664 /* Pad all strings with spaces instead of nulls */ 665 cd9660_pad_string_spaces(diskStructure.primaryDescriptor.volume_id, 32); 666 cd9660_pad_string_spaces(diskStructure.primaryDescriptor.system_id, 32); 667 cd9660_pad_string_spaces(diskStructure.primaryDescriptor.volume_set_id, 668 128); 669 cd9660_pad_string_spaces(diskStructure.primaryDescriptor.publisher_id, 670 128); 671 cd9660_pad_string_spaces(diskStructure.primaryDescriptor.preparer_id, 672 128); 673 cd9660_pad_string_spaces(diskStructure.primaryDescriptor.application_id, 674 128); 675 cd9660_pad_string_spaces( 676 diskStructure.primaryDescriptor.copyright_file_id, 128); 677 cd9660_pad_string_spaces( 678 diskStructure.primaryDescriptor.abstract_file_id, 128); 679 cd9660_pad_string_spaces( 680 diskStructure.primaryDescriptor.bibliographic_file_id, 128); 681 682 /* Setup dates */ 683 time(&tim); 684 cd9660_time_8426( 685 (unsigned char *)diskStructure.primaryDescriptor.creation_date, 686 tim); 687 cd9660_time_8426( 688 (unsigned char *)diskStructure.primaryDescriptor.modification_date, 689 tim); 690 691 /* 692 cd9660_set_date(diskStructure.primaryDescriptor.expiration_date, now); 693 */ 694 695 memset(diskStructure.primaryDescriptor.expiration_date, '0' ,17); 696 cd9660_time_8426( 697 (unsigned char *)diskStructure.primaryDescriptor.effective_date, 698 tim); 699 } 700 701 static void 702 cd9660_populate_iso_dir_record(struct _iso_directory_record_cd9660 *record, 703 u_char ext_attr_length, u_char flags, 704 u_char name_len, const char * name) 705 { 706 record->ext_attr_length[0] = ext_attr_length; 707 record->flags[0] = ISO_FLAG_CLEAR | flags; 708 record->file_unit_size[0] = 0; 709 record->interleave[0] = 0; 710 cd9660_bothendian_word(1, record->volume_sequence_number); 711 record->name_len[0] = name_len; 712 memset(record->name, '\0', sizeof (record->name)); 713 memcpy(record->name, name, name_len); 714 record->length[0] = 33 + name_len; 715 716 /* Todo : better rounding */ 717 record->length[0] += (record->length[0] & 1) ? 1 : 0; 718 } 719 720 static void 721 cd9660_setup_root_node(void) 722 { 723 cd9660_populate_iso_dir_record(diskStructure.rootNode->isoDirRecord, 724 0, ISO_FLAG_DIRECTORY, 1, "\0"); 725 726 } 727 728 /*********** SUPPORT FUNCTIONS ***********/ 729 static int 730 cd9660_setup_volume_descriptors(void) 731 { 732 /* Boot volume descriptor should come second */ 733 int sector = 16; 734 /* For now, a fixed 2 : PVD and terminator */ 735 volume_descriptor *temp, *t; 736 737 /* Set up the PVD */ 738 if ((temp = malloc(sizeof(volume_descriptor))) == NULL) { 739 CD9660_MEM_ALLOC_ERROR("cd9660_setup_volume_descriptors"); 740 exit(1); 741 } 742 743 temp->volumeDescriptorData = 744 (unsigned char *)&diskStructure.primaryDescriptor; 745 temp->volumeDescriptorData[0] = ISO_VOLUME_DESCRIPTOR_PVD; 746 temp->volumeDescriptorData[6] = 1; 747 temp->sector = sector; 748 memcpy(temp->volumeDescriptorData + 1, 749 ISO_VOLUME_DESCRIPTOR_STANDARD_ID, 5); 750 diskStructure.firstVolumeDescriptor = temp; 751 752 sector++; 753 /* Set up boot support if enabled. BVD must reside in sector 17 */ 754 if (diskStructure.is_bootable) { 755 if ((t = malloc(sizeof(volume_descriptor))) == NULL) { 756 CD9660_MEM_ALLOC_ERROR( 757 "cd9660_setup_volume_descriptors"); 758 exit(1); 759 } 760 if ((t->volumeDescriptorData = malloc(2048)) == NULL) { 761 CD9660_MEM_ALLOC_ERROR( 762 "cd9660_setup_volume_descriptors"); 763 exit(1); 764 } 765 temp->next = t; 766 temp = t; 767 memset(t->volumeDescriptorData, 0, 2048); 768 t->sector = 17; 769 if (diskStructure.verbose_level > 0) 770 printf("Setting up boot volume descriptor\n"); 771 cd9660_setup_boot_volume_descriptor(t); 772 sector++; 773 } 774 775 /* Set up the terminator */ 776 if ((t = malloc(sizeof(volume_descriptor))) == NULL) { 777 CD9660_MEM_ALLOC_ERROR("cd9660_setup_volume_descriptors"); 778 exit(1); 779 } 780 if ((t->volumeDescriptorData = malloc(2048)) == NULL) { 781 CD9660_MEM_ALLOC_ERROR("cd9660_setup_volume_descriptors"); 782 exit(1); 783 } 784 785 temp->next = t; 786 memset(t->volumeDescriptorData, 0, 2048); 787 t->volumeDescriptorData[0] = ISO_VOLUME_DESCRIPTOR_TERMINATOR; 788 t->next = 0; 789 t->volumeDescriptorData[6] = 1; 790 t->sector = sector; 791 memcpy(t->volumeDescriptorData + 1, 792 ISO_VOLUME_DESCRIPTOR_STANDARD_ID, 5); 793 794 sector++; 795 return sector; 796 } 797 798 #if 0 799 /* 800 * Populate EAR at some point. Not required, but is used by NetBSD's 801 * cd9660 support 802 */ 803 static int 804 cd9660_fill_extended_attribute_record(cd9660node *node) 805 { 806 if ((node->isoExtAttributes = 807 malloc(sizeof(struct iso_extended_attributes))) == NULL) { 808 CD9660_MEM_ALLOC_ERROR("cd9660_fill_extended_attribute_record"); 809 exit(1); 810 }; 811 812 return 1; 813 } 814 #endif 815 816 static int 817 cd9960_translate_node_common(cd9660node *newnode) 818 { 819 time_t tim; 820 int test; 821 u_char flag; 822 char temp[ISO_FILENAME_MAXLENGTH_WITH_PADDING]; 823 824 /* Now populate the isoDirRecord structure */ 825 memset(temp, 0, ISO_FILENAME_MAXLENGTH_WITH_PADDING); 826 827 test = cd9660_convert_filename(newnode->node->name, 828 temp, !(S_ISDIR(newnode->node->type))); 829 830 flag = ISO_FLAG_CLEAR; 831 if (S_ISDIR(newnode->node->type)) 832 flag |= ISO_FLAG_DIRECTORY; 833 834 cd9660_populate_iso_dir_record(newnode->isoDirRecord, 0, 835 flag, strlen(temp), temp); 836 837 /* Set the various dates */ 838 839 /* If we want to use the current date and time */ 840 time(&tim); 841 842 cd9660_time_915(newnode->isoDirRecord->date, tim); 843 844 cd9660_bothendian_dword(newnode->fileDataLength, 845 newnode->isoDirRecord->size); 846 /* If the file is a link, we want to set the size to 0 */ 847 if (S_ISLNK(newnode->node->type)) 848 newnode->fileDataLength = 0; 849 850 return 1; 851 } 852 853 /* 854 * Translate fsnode to cd9960node 855 * Translate filenames and other metadata, including dates, sizes, 856 * permissions, etc 857 * @param struct fsnode * The node generated by makefs 858 * @param struct cd9660node * The intermediate node to be written to 859 * @returns int 0 on failure, 1 on success 860 */ 861 static int 862 cd9660_translate_node(fsnode *node, cd9660node *newnode) 863 { 864 if (node == NULL) { 865 if (diskStructure.verbose_level > 0) 866 printf("cd9660_translate_node: NULL node passed, " 867 "returning\n"); 868 return 0; 869 } 870 if ((newnode->isoDirRecord = 871 malloc(sizeof(iso_directory_record_cd9660))) == NULL) { 872 CD9660_MEM_ALLOC_ERROR("cd9660_translate_node"); 873 return 0; 874 } 875 876 /* Set the node pointer */ 877 newnode->node = node; 878 879 /* Set the size */ 880 if (!(S_ISDIR(node->type))) 881 newnode->fileDataLength = node->inode->st.st_size; 882 883 if (cd9960_translate_node_common(newnode) == 0) 884 return 0; 885 886 /* Finally, overwrite some of the values that are set by default */ 887 cd9660_time_915(newnode->isoDirRecord->date, node->inode->st.st_mtime); 888 889 return 1; 890 } 891 892 /* 893 * Compares two ISO filenames 894 * @param const char * The first file name 895 * @param const char * The second file name 896 * @returns : -1 if first is less than second, 0 if they are the same, 1 if 897 * the second is greater than the first 898 */ 899 static int 900 cd9660_compare_filename(const char *first, const char *second) 901 { 902 /* 903 * This can be made more optimal once it has been tested 904 * (the extra character, for example, is for testing) 905 */ 906 907 int p1 = 0; 908 int p2 = 0; 909 char c1, c2; 910 /* First, on the filename */ 911 912 while (p1 < ISO_FILENAME_MAXLENGTH_BEFORE_VERSION-1 913 && p2 < ISO_FILENAME_MAXLENGTH_BEFORE_VERSION-1) { 914 c1 = first[p1]; 915 c2 = second[p2]; 916 if (c1 == '.' && c2 =='.') 917 break; 918 else if (c1 == '.') { 919 p2++; 920 c1 = ' '; 921 } else if (c2 == '.') { 922 p1++; 923 c2 = ' '; 924 } else { 925 p1++; 926 p2++; 927 } 928 929 if (c1 < c2) 930 return -1; 931 else if (c1 > c2) { 932 return 1; 933 } 934 } 935 936 if (first[p1] == '.' && second[p2] == '.') { 937 p1++; 938 p2++; 939 while (p1 < ISO_FILENAME_MAXLENGTH_BEFORE_VERSION - 1 940 && p2 < ISO_FILENAME_MAXLENGTH_BEFORE_VERSION - 1) { 941 c1 = first[p1]; 942 c2 = second[p2]; 943 if (c1 == ';' && c2 == ';') 944 break; 945 else if (c1 == ';') { 946 p2++; 947 c1 = ' '; 948 } else if (c2 == ';') { 949 p1++; 950 c2 = ' '; 951 } else { 952 p1++; 953 p2++; 954 } 955 956 if (c1 < c2) 957 return -1; 958 else if (c1 > c2) 959 return 1; 960 } 961 } 962 return 0; 963 } 964 965 /* 966 * Insert a node into list with ISO sorting rules 967 * @param cd9660node * The head node of the list 968 * @param cd9660node * The node to be inserted 969 */ 970 static void 971 cd9660_sorted_child_insert(cd9660node *parent, cd9660node *cn_new) 972 { 973 int compare; 974 cd9660node *cn; 975 struct cd9660_children_head *head = &parent->cn_children; 976 977 /* TODO: Optimize? */ 978 cn_new->parent = parent; 979 980 /* 981 * first will either be 0, the . or the .. 982 * if . or .., this means no other entry may be written before first 983 * if 0, the new node may be inserted at the head 984 */ 985 986 TAILQ_FOREACH(cn, head, cn_next_child) { 987 /* 988 * Dont insert a node twice - 989 * that would cause an infinite loop 990 */ 991 if (cn_new == cn) 992 return; 993 994 compare = cd9660_compare_filename(cn_new->isoDirRecord->name, 995 cn->isoDirRecord->name); 996 997 if (compare == 0) 998 compare = cd9660_compare_filename(cn_new->node->name, 999 cn->node->name); 1000 1001 if (compare < 0) 1002 break; 1003 } 1004 if (cn == NULL) 1005 TAILQ_INSERT_TAIL(head, cn_new, cn_next_child); 1006 else 1007 TAILQ_INSERT_BEFORE(cn, cn_new, cn_next_child); 1008 } 1009 1010 /* 1011 * Called After cd9660_sorted_child_insert 1012 * handles file collisions by suffixing each filname with ~n 1013 * where n represents the files respective place in the ordering 1014 */ 1015 static int 1016 cd9660_handle_collisions(cd9660node *colliding, int past) 1017 { 1018 cd9660node *iter, *next, *prev; 1019 int skip; 1020 int delete_chars = 0; 1021 int temp_past = past; 1022 int temp_skip; 1023 int flag = 0; 1024 cd9660node *end_of_range; 1025 1026 for (iter = TAILQ_FIRST(&colliding->cn_children); 1027 iter != NULL && (next = TAILQ_NEXT(iter, cn_next_child)) != NULL;) { 1028 if (strcmp(iter->isoDirRecord->name, 1029 next->isoDirRecord->name) != 0) { 1030 iter = TAILQ_NEXT(iter, cn_next_child); 1031 continue; 1032 } 1033 flag = 1; 1034 temp_skip = skip = cd9660_count_collisions(iter); 1035 end_of_range = iter; 1036 while (temp_skip > 0) { 1037 temp_skip--; 1038 end_of_range = TAILQ_NEXT(end_of_range, cn_next_child); 1039 } 1040 temp_past = past; 1041 while (temp_past > 0) { 1042 if ((next = TAILQ_NEXT(end_of_range, cn_next_child)) != NULL) 1043 end_of_range = next; 1044 else if ((prev = TAILQ_PREV(iter, cd9660_children_head, cn_next_child)) != NULL) 1045 iter = prev; 1046 else 1047 delete_chars++; 1048 temp_past--; 1049 } 1050 skip += past; 1051 iter = cd9660_rename_filename(iter, skip, delete_chars); 1052 } 1053 return flag; 1054 } 1055 1056 1057 static cd9660node * 1058 cd9660_rename_filename(cd9660node *iter, int num, int delete_chars) 1059 { 1060 int i = 0; 1061 int numbts, dot, semi, digit, digits, temp, powers, multiplier, count; 1062 char *naming; 1063 int maxlength; 1064 char *tmp; 1065 1066 if (diskStructure.verbose_level > 0) 1067 printf("Rename_filename called\n"); 1068 1069 /* TODO : A LOT of chanes regarding 8.3 filenames */ 1070 if (diskStructure.isoLevel == 1) 1071 maxlength = 8; 1072 else if (diskStructure.isoLevel == 2) 1073 maxlength = 31; 1074 else 1075 maxlength = ISO_FILENAME_MAXLENGTH_BEFORE_VERSION; 1076 1077 tmp = malloc(ISO_FILENAME_MAXLENGTH_WITH_PADDING); 1078 1079 while (i < num) { 1080 powers = 1; 1081 count = 0; 1082 digits = 1; 1083 multiplier = 1; 1084 while (((int)(i / powers) ) >= 10) { 1085 digits++; 1086 powers = powers * 10; 1087 } 1088 1089 naming = iter->o_name; 1090 1091 /* 1092 while ((*naming != '.') && (*naming != ';')) { 1093 naming++; 1094 count++; 1095 } 1096 */ 1097 1098 dot = -1; 1099 semi = -1; 1100 while (count < maxlength) { 1101 if (*naming == '.') 1102 dot = count; 1103 else if (*naming == ';') { 1104 semi = count; 1105 break; 1106 } 1107 naming++; 1108 count++; 1109 } 1110 1111 if ((count + digits) < maxlength) 1112 numbts = count; 1113 else 1114 numbts = maxlength - (digits); 1115 numbts -= delete_chars; 1116 1117 /* 8.3 rules - keep the extension, add before the dot */ 1118 1119 /* 1120 * This code makes a bunch of assumptions. 1121 * See if you can spot them all :) 1122 */ 1123 1124 /* 1125 if (diskStructure.isoLevel == 1) { 1126 numbts = 8 - digits - delete_chars; 1127 if (dot < 0) { 1128 1129 } else { 1130 if (dot < 8) { 1131 memmove(&tmp[numbts],&tmp[dot],4); 1132 } 1133 } 1134 } 1135 */ 1136 1137 /* (copying just the filename before the '.' */ 1138 memcpy(tmp, (iter->o_name), numbts); 1139 1140 /* adding the appropriate number following the name */ 1141 temp = i; 1142 while (digits > 0) { 1143 digit = (int)(temp / powers); 1144 temp = temp - digit * powers; 1145 sprintf(&tmp[numbts] , "%d", digit); 1146 digits--; 1147 numbts++; 1148 powers = powers / 10; 1149 } 1150 1151 while ((*naming != ';') && (numbts < maxlength)) { 1152 tmp[numbts] = (*naming); 1153 naming++; 1154 numbts++; 1155 } 1156 1157 tmp[numbts] = ';'; 1158 tmp[numbts+1] = '1'; 1159 tmp[numbts+2] = '\0'; 1160 1161 /* 1162 * now tmp has exactly the identifier 1163 * we want so we'll copy it back to record 1164 */ 1165 memcpy((iter->isoDirRecord->name), tmp, numbts + 3); 1166 1167 iter = TAILQ_NEXT(iter, cn_next_child); 1168 i++; 1169 } 1170 1171 free(tmp); 1172 return iter; 1173 } 1174 1175 /* Todo: Figure out why these functions are nec. */ 1176 static void 1177 cd9660_copy_filenames(cd9660node *node) 1178 { 1179 cd9660node *cn; 1180 1181 if (TAILQ_EMPTY(&node->cn_children)) 1182 return; 1183 1184 if (TAILQ_FIRST(&node->cn_children)->isoDirRecord == NULL) { 1185 debug_print_tree(diskStructure.rootNode, 0); 1186 exit(1); 1187 } 1188 1189 TAILQ_FOREACH(cn, &node->cn_children, cn_next_child) { 1190 cd9660_copy_filenames(cn); 1191 memcpy(cn->o_name, cn->isoDirRecord->name, 1192 ISO_FILENAME_MAXLENGTH_WITH_PADDING); 1193 } 1194 } 1195 1196 static void 1197 cd9660_sorting_nodes(cd9660node *node) 1198 { 1199 cd9660node *cn; 1200 1201 TAILQ_FOREACH(cn, &node->cn_children, cn_next_child) 1202 cd9660_sorting_nodes(cn); 1203 cd9660_sort_nodes(node); 1204 } 1205 1206 /* XXX Bubble sort. */ 1207 static void 1208 cd9660_sort_nodes(cd9660node *node) 1209 { 1210 cd9660node *cn, *next; 1211 1212 do { 1213 TAILQ_FOREACH(cn, &node->cn_children, cn_next_child) { 1214 if ((next = TAILQ_NEXT(cn, cn_next_child)) == NULL) 1215 return; 1216 else if (strcmp(next->isoDirRecord->name, 1217 cn->isoDirRecord->name) >= 0) 1218 continue; 1219 TAILQ_REMOVE(&node->cn_children, next, cn_next_child); 1220 TAILQ_INSERT_BEFORE(cn, next, cn_next_child); 1221 break; 1222 } 1223 } while (cn != NULL); 1224 } 1225 1226 static int 1227 cd9660_count_collisions(cd9660node *copy) 1228 { 1229 int count = 0; 1230 cd9660node *iter, *next; 1231 1232 for (iter = copy; 1233 (next = TAILQ_NEXT(iter, cn_next_child)) != NULL; 1234 iter = next) { 1235 if (cd9660_compare_filename(iter->isoDirRecord->name, 1236 next->isoDirRecord->name) == 0) 1237 count++; 1238 else 1239 return count; 1240 } 1241 #if 0 1242 if ((next = TAILQ_NEXT(iter, cn_next_child)) != NULL) { 1243 printf("cd9660_recurse_on_collision: count is %i \n", count); 1244 compare = cd9660_compare_filename(iter->isoDirRecord->name, 1245 next->isoDirRecord->name); 1246 if (compare == 0) { 1247 count++; 1248 return cd9660_recurse_on_collision(next, count); 1249 } else 1250 return count; 1251 } 1252 #endif 1253 return count; 1254 } 1255 1256 static cd9660node * 1257 cd9660_rrip_move_directory(cd9660node *dir) 1258 { 1259 char newname[9]; 1260 cd9660node *tfile; 1261 1262 /* 1263 * This function needs to: 1264 * 1) Create an empty virtual file in place of the old directory 1265 * 2) Point the virtual file to the new directory 1266 * 3) Point the relocated directory to its old parent 1267 * 4) Move the directory specified by dir into rr_moved_dir, 1268 * and rename it to "diskStructure.rock_ridge_move_count" (as a string) 1269 */ 1270 1271 /* First see if the moved directory even exists */ 1272 if (diskStructure.rr_moved_dir == NULL) { 1273 diskStructure.rr_moved_dir = 1274 cd9660_create_directory(ISO_RRIP_DEFAULT_MOVE_DIR_NAME, 1275 diskStructure.rootNode, dir); 1276 if (diskStructure.rr_moved_dir == NULL) 1277 return 0; 1278 } 1279 1280 /* Create a file with the same ORIGINAL name */ 1281 tfile = cd9660_create_file(dir->node->name, dir->parent, dir); 1282 if (tfile == NULL) 1283 return NULL; 1284 1285 diskStructure.rock_ridge_move_count++; 1286 snprintf(newname, sizeof(newname), "%08i", 1287 diskStructure.rock_ridge_move_count); 1288 1289 /* Point to old parent */ 1290 dir->rr_real_parent = dir->parent; 1291 1292 /* Place the placeholder file */ 1293 if (TAILQ_EMPTY(&dir->rr_real_parent->cn_children)) { 1294 TAILQ_INSERT_HEAD(&dir->rr_real_parent->cn_children, tfile, 1295 cn_next_child); 1296 } else { 1297 cd9660_sorted_child_insert(dir->rr_real_parent, tfile); 1298 } 1299 1300 /* Point to new parent */ 1301 dir->parent = diskStructure.rr_moved_dir; 1302 1303 /* Point the file to the moved directory */ 1304 tfile->rr_relocated = dir; 1305 1306 /* Actually move the directory */ 1307 cd9660_sorted_child_insert(diskStructure.rr_moved_dir, dir); 1308 1309 /* TODO: Inherit permissions / ownership (basically the entire inode) */ 1310 1311 /* Set the new name */ 1312 memset(dir->isoDirRecord->name, 0, ISO_FILENAME_MAXLENGTH_WITH_PADDING); 1313 strncpy(dir->isoDirRecord->name, newname, 8); 1314 1315 return dir; 1316 } 1317 1318 static int 1319 cd9660_add_dot_records(cd9660node *root) 1320 { 1321 struct cd9660_children_head *head = &root->cn_children; 1322 cd9660node *cn; 1323 1324 TAILQ_FOREACH(cn, head, cn_next_child) { 1325 if ((cn->type & CD9660_TYPE_DIR) == 0) 1326 continue; 1327 /* Recursion first */ 1328 cd9660_add_dot_records(cn); 1329 } 1330 cd9660_create_special_directory(CD9660_TYPE_DOT, root); 1331 cd9660_create_special_directory(CD9660_TYPE_DOTDOT, root); 1332 return 1; 1333 } 1334 1335 /* 1336 * Convert node to cd9660 structure 1337 * This function is designed to be called recursively on the root node of 1338 * the filesystem 1339 * Lots of recursion going on here, want to make sure it is efficient 1340 * @param struct fsnode * The root node to be converted 1341 * @param struct cd9660* The parent node (should not be NULL) 1342 * @param int Current directory depth 1343 * @param int* Running count of the number of directories that are being created 1344 */ 1345 static void 1346 cd9660_convert_structure(fsnode *root, cd9660node *parent_node, int level, 1347 int *numDirectories, int *error) 1348 { 1349 fsnode *iterator = root; 1350 cd9660node *this_node; 1351 int working_level; 1352 int add; 1353 int flag = 0; 1354 int counter = 0; 1355 1356 /* 1357 * Newer, more efficient method, reduces recursion depth 1358 */ 1359 if (root == NULL) { 1360 warnx("%s: root is null\n", __func__); 1361 return; 1362 } 1363 1364 /* Test for an empty directory - makefs still gives us the . record */ 1365 if ((S_ISDIR(root->type)) && (root->name[0] == '.') 1366 && (root->name[1] == '\0')) { 1367 root = root->next; 1368 if (root == NULL) 1369 return; 1370 } 1371 if ((this_node = cd9660_allocate_cd9660node()) == NULL) { 1372 CD9660_MEM_ALLOC_ERROR(__func__); 1373 } 1374 1375 /* 1376 * To reduce the number of recursive calls, we will iterate over 1377 * the next pointers to the right. 1378 */ 1379 while (iterator != NULL) { 1380 add = 1; 1381 /* 1382 * Increment the directory count if this is a directory 1383 * Ignore "." entries. We will generate them later 1384 */ 1385 if (!S_ISDIR(iterator->type) || 1386 strcmp(iterator->name, ".") != 0) { 1387 1388 /* Translate the node, including its filename */ 1389 this_node->parent = parent_node; 1390 cd9660_translate_node(iterator, this_node); 1391 this_node->level = level; 1392 1393 if (S_ISDIR(iterator->type)) { 1394 (*numDirectories)++; 1395 this_node->type = CD9660_TYPE_DIR; 1396 working_level = level + 1; 1397 1398 /* 1399 * If at level 8, directory would be at 8 1400 * and have children at 9 which is not 1401 * allowed as per ISO spec 1402 */ 1403 if (level == 8) { 1404 if ((!diskStructure.allow_deep_trees) && 1405 (!diskStructure.rock_ridge_enabled)) { 1406 warnx("error: found entry " 1407 "with depth greater " 1408 "than 8."); 1409 (*error) = 1; 1410 return; 1411 } else if (diskStructure. 1412 rock_ridge_enabled) { 1413 working_level = 3; 1414 /* 1415 * Moved directory is actually 1416 * at level 2. 1417 */ 1418 this_node->level = 1419 working_level - 1; 1420 if (cd9660_rrip_move_directory( 1421 this_node) == 0) { 1422 warnx("Failure in " 1423 "cd9660_rrip_" 1424 "move_directory" 1425 ); 1426 (*error) = 1; 1427 return; 1428 } 1429 add = 0; 1430 } 1431 } 1432 1433 /* Do the recursive call on the children */ 1434 if (iterator->child != 0) { 1435 cd9660_convert_structure( 1436 iterator->child, this_node, 1437 working_level, 1438 numDirectories, error); 1439 1440 if ((*error) == 1) { 1441 warnx("%s: Error on recursive " 1442 "call", __func__); 1443 return; 1444 } 1445 } 1446 1447 } else { 1448 /* Only directories should have children */ 1449 assert(iterator->child == NULL); 1450 1451 this_node->type = CD9660_TYPE_FILE; 1452 } 1453 1454 /* 1455 * Finally, do a sorted insert 1456 */ 1457 if (add) { 1458 cd9660_sorted_child_insert( 1459 parent_node, this_node); 1460 } 1461 1462 /*Allocate new temp_node */ 1463 if (iterator->next != 0) { 1464 this_node = cd9660_allocate_cd9660node(); 1465 if (this_node == NULL) 1466 CD9660_MEM_ALLOC_ERROR(__func__); 1467 } 1468 } 1469 iterator = iterator->next; 1470 } 1471 1472 /* cd9660_handle_collisions(first_node); */ 1473 1474 /* TODO: need cleanup */ 1475 cd9660_copy_filenames(parent_node); 1476 1477 do { 1478 flag = cd9660_handle_collisions(parent_node, counter); 1479 counter++; 1480 cd9660_sorting_nodes(parent_node); 1481 } while ((flag == 1) && (counter < 100)); 1482 } 1483 1484 /* 1485 * Clean up the cd9660node tree 1486 * This is designed to be called recursively on the root node 1487 * @param struct cd9660node *root The node to free 1488 * @returns void 1489 */ 1490 static void 1491 cd9660_free_structure(cd9660node *root) 1492 { 1493 cd9660node *cn; 1494 1495 while ((cn = TAILQ_FIRST(&root->cn_children)) != NULL) { 1496 TAILQ_REMOVE(&root->cn_children, cn, cn_next_child); 1497 cd9660_free_structure(cn); 1498 } 1499 free(root); 1500 } 1501 1502 /* 1503 * Be a little more memory conservative: 1504 * instead of having the TAILQ_ENTRY as part of the cd9660node, 1505 * just create a temporary structure 1506 */ 1507 struct ptq_entry 1508 { 1509 TAILQ_ENTRY(ptq_entry) ptq; 1510 cd9660node *node; 1511 } *n; 1512 1513 #define PTQUEUE_NEW(n,s,r,t){\ 1514 n = malloc(sizeof(struct s)); \ 1515 if (n == NULL) \ 1516 return r; \ 1517 n->node = t;\ 1518 } 1519 1520 /* 1521 * Generate the path tables 1522 * The specific implementation of this function is left as an exercise to the 1523 * programmer. It could be done recursively. Make sure you read how the path 1524 * table has to be laid out, it has levels. 1525 * @param struct iso9660_disk *disk The disk image 1526 * @returns int The number of built path tables (between 1 and 4), 0 on failure 1527 */ 1528 static int 1529 cd9660_generate_path_table(void) 1530 { 1531 cd9660node *cn, *dirNode = diskStructure.rootNode; 1532 cd9660node *last = dirNode; 1533 int pathTableSize = 0; /* computed as we go */ 1534 int counter = 1; /* root gets a count of 0 */ 1535 int parentRecNum = 0; /* root's parent is '0' */ 1536 1537 TAILQ_HEAD(cd9660_pt_head, ptq_entry) pt_head; 1538 TAILQ_INIT(&pt_head); 1539 1540 PTQUEUE_NEW(n, ptq_entry, -1, diskStructure.rootNode); 1541 1542 /* Push the root node */ 1543 TAILQ_INSERT_HEAD(&pt_head, n, ptq); 1544 1545 /* Breadth-first traversal of file structure */ 1546 while (pt_head.tqh_first != 0) { 1547 n = pt_head.tqh_first; 1548 dirNode = n->node; 1549 TAILQ_REMOVE(&pt_head, pt_head.tqh_first, ptq); 1550 free(n); 1551 1552 /* Update the size */ 1553 pathTableSize += ISO_PATHTABLE_ENTRY_BASESIZE 1554 + dirNode->isoDirRecord->name_len[0]+ 1555 (dirNode->isoDirRecord->name_len[0] % 2 == 0 ? 0 : 1); 1556 /* includes the padding bit */ 1557 1558 dirNode->ptnumber=counter; 1559 if (dirNode != last) { 1560 last->ptnext = dirNode; 1561 dirNode->ptprev = last; 1562 } 1563 last = dirNode; 1564 1565 parentRecNum = 1; 1566 if (dirNode->parent != 0) 1567 parentRecNum = dirNode->parent->ptnumber; 1568 1569 /* Push children onto queue */ 1570 TAILQ_FOREACH(cn, &dirNode->cn_children, cn_next_child) { 1571 /* 1572 * Dont add the DOT and DOTDOT types to the path 1573 * table. 1574 */ 1575 if ((cn->type != CD9660_TYPE_DOT) 1576 && (cn->type != CD9660_TYPE_DOTDOT)) { 1577 1578 if (S_ISDIR(cn->node->type)) { 1579 PTQUEUE_NEW(n, ptq_entry, -1, cn); 1580 TAILQ_INSERT_TAIL(&pt_head, n, ptq); 1581 } 1582 } 1583 } 1584 counter++; 1585 } 1586 return pathTableSize; 1587 } 1588 1589 void 1590 cd9660_compute_full_filename(cd9660node *node, char *buf, int level) 1591 { 1592 cd9660node *parent; 1593 1594 parent = (node->rr_real_parent == NULL ? 1595 node->parent : node->rr_real_parent); 1596 if (parent != NULL) { 1597 cd9660_compute_full_filename(parent, buf, level + 1); 1598 strcat(buf, node->node->name); 1599 } else { 1600 /* We are at the root */ 1601 strcat(buf, diskStructure.rootFilesystemPath); 1602 if (buf[strlen(buf) - 1] == '/') 1603 buf[strlen(buf) - 1] = '\0'; 1604 } 1605 1606 if (level != 0) 1607 strcat(buf, "/"); 1608 } 1609 1610 /* NEW filename conversion method */ 1611 typedef int(*cd9660_filename_conversion_functor)(const char *, char *, int); 1612 1613 1614 /* 1615 * TODO: These two functions are almost identical. 1616 * Some code cleanup is possible here 1617 * 1618 * XXX bounds checking! 1619 */ 1620 static int 1621 cd9660_level1_convert_filename(const char *oldname, char *newname, int is_file) 1622 { 1623 /* 1624 * ISO 9660 : 10.1 1625 * File Name shall not contain more than 8 d or d1 characters 1626 * File Name Extension shall not contain more than 3 d or d1 characters 1627 * Directory Identifier shall not contain more than 8 d or d1 characters 1628 */ 1629 int namelen = 0; 1630 int extlen = 0; 1631 int found_ext = 0; 1632 1633 while (*oldname != '\0') { 1634 /* Handle period first, as it is special */ 1635 if (*oldname == '.') { 1636 if (found_ext) { 1637 *newname++ = '_'; 1638 extlen ++; 1639 } 1640 else { 1641 *newname++ = '.'; 1642 found_ext = 1; 1643 } 1644 } else { 1645 /* cut RISC OS file type off ISO name */ 1646 if (diskStructure.archimedes_enabled && 1647 *oldname == ',' && strlen(oldname) == 4) 1648 break; 1649 /* Enforce 12.3 / 8 */ 1650 if (((namelen == 8) && !found_ext) || 1651 (found_ext && extlen == 3)) { 1652 break; 1653 } 1654 1655 if (islower((unsigned char)*oldname)) 1656 *newname++ = toupper((unsigned char)*oldname); 1657 else if (isupper((unsigned char)*oldname) 1658 || isdigit((unsigned char)*oldname)) 1659 *newname++ = *oldname; 1660 else 1661 *newname++ = '_'; 1662 1663 if (found_ext) 1664 extlen++; 1665 else 1666 namelen++; 1667 } 1668 oldname ++; 1669 } 1670 if (is_file) { 1671 if (!found_ext && !diskStructure.omit_trailing_period) 1672 *newname++ = '.'; 1673 /* Add version */ 1674 sprintf(newname, ";%i", 1); 1675 } 1676 return namelen + extlen + found_ext; 1677 } 1678 1679 /* XXX bounds checking! */ 1680 static int 1681 cd9660_level2_convert_filename(const char *oldname, char *newname, int is_file) 1682 { 1683 /* 1684 * ISO 9660 : 7.5.1 1685 * File name : 0+ d or d1 characters 1686 * separator 1 (.) 1687 * File name extension : 0+ d or d1 characters 1688 * separator 2 (;) 1689 * File version number (5 characters, 1-32767) 1690 * 1 <= Sum of File name and File name extension <= 30 1691 */ 1692 int namelen = 0; 1693 int extlen = 0; 1694 int found_ext = 0; 1695 1696 while (*oldname != '\0') { 1697 /* Handle period first, as it is special */ 1698 if (*oldname == '.') { 1699 if (found_ext) { 1700 if (diskStructure.allow_multidot) { 1701 *newname++ = '.'; 1702 } else { 1703 *newname++ = '_'; 1704 } 1705 extlen ++; 1706 } 1707 else { 1708 *newname++ = '.'; 1709 found_ext = 1; 1710 } 1711 } else { 1712 /* cut RISC OS file type off ISO name */ 1713 if (diskStructure.archimedes_enabled && 1714 *oldname == ',' && strlen(oldname) == 4) 1715 break; 1716 if ((namelen + extlen) == 30) 1717 break; 1718 1719 if (islower((unsigned char)*oldname)) 1720 *newname++ = toupper((unsigned char)*oldname); 1721 else if (isupper((unsigned char)*oldname) || 1722 isdigit((unsigned char)*oldname)) 1723 *newname++ = *oldname; 1724 else if (diskStructure.allow_multidot && 1725 *oldname == '.') { 1726 *newname++ = '.'; 1727 } else { 1728 *newname++ = '_'; 1729 } 1730 1731 if (found_ext) 1732 extlen++; 1733 else 1734 namelen++; 1735 } 1736 oldname ++; 1737 } 1738 if (is_file) { 1739 if (!found_ext && !diskStructure.omit_trailing_period) 1740 *newname++ = '.'; 1741 /* Add version */ 1742 sprintf(newname, ";%i", 1); 1743 } 1744 return namelen + extlen + found_ext; 1745 } 1746 1747 #if 0 1748 static int 1749 cd9660_joliet_convert_filename(const char *oldname, char *newname, int is_file) 1750 { 1751 /* TODO: implement later, move to cd9660_joliet.c ?? */ 1752 } 1753 #endif 1754 1755 1756 /* 1757 * Convert a file name to ISO compliant file name 1758 * @param char * oldname The original filename 1759 * @param char ** newname The new file name, in the appropriate character 1760 * set and of appropriate length 1761 * @param int 1 if file, 0 if directory 1762 * @returns int The length of the new string 1763 */ 1764 static int 1765 cd9660_convert_filename(const char *oldname, char *newname, int is_file) 1766 { 1767 /* NEW */ 1768 cd9660_filename_conversion_functor conversion_function = 0; 1769 if (diskStructure.isoLevel == 1) 1770 conversion_function = &cd9660_level1_convert_filename; 1771 else if (diskStructure.isoLevel == 2) 1772 conversion_function = &cd9660_level2_convert_filename; 1773 return (*conversion_function)(oldname, newname, is_file); 1774 } 1775 1776 int 1777 cd9660_compute_record_size(cd9660node *node) 1778 { 1779 int size = node->isoDirRecord->length[0]; 1780 1781 if (diskStructure.rock_ridge_enabled) 1782 size += node->susp_entry_size; 1783 size += node->su_tail_size; 1784 size += size & 1; /* Ensure length of record is even. */ 1785 assert(size <= 254); 1786 return size; 1787 } 1788 1789 static void 1790 cd9660_populate_dot_records(cd9660node *node) 1791 { 1792 node->dot_record->fileDataSector = node->fileDataSector; 1793 memcpy(node->dot_record->isoDirRecord,node->isoDirRecord, 34); 1794 node->dot_record->isoDirRecord->name_len[0] = 1; 1795 node->dot_record->isoDirRecord->name[0] = 0; 1796 node->dot_record->isoDirRecord->name[1] = 0; 1797 node->dot_record->isoDirRecord->length[0] = 34; 1798 node->dot_record->fileRecordSize = 1799 cd9660_compute_record_size(node->dot_record); 1800 1801 if (node == diskStructure.rootNode) { 1802 node->dot_dot_record->fileDataSector = node->fileDataSector; 1803 memcpy(node->dot_dot_record->isoDirRecord,node->isoDirRecord, 1804 34); 1805 } else { 1806 node->dot_dot_record->fileDataSector = 1807 node->parent->fileDataSector; 1808 memcpy(node->dot_dot_record->isoDirRecord, 1809 node->parent->isoDirRecord,34); 1810 } 1811 node->dot_dot_record->isoDirRecord->name_len[0] = 1; 1812 node->dot_dot_record->isoDirRecord->name[0] = 1; 1813 node->dot_dot_record->isoDirRecord->name[1] = 0; 1814 node->dot_dot_record->isoDirRecord->length[0] = 34; 1815 node->dot_dot_record->fileRecordSize = 1816 cd9660_compute_record_size(node->dot_dot_record); 1817 } 1818 1819 /* 1820 * @param struct cd9660node *node The node 1821 * @param int The offset (in bytes) - SHOULD align to the beginning of a sector 1822 * @returns int The total size of files and directory entries (should be 1823 * a multiple of sector size) 1824 */ 1825 static int 1826 cd9660_compute_offsets(cd9660node *node, int startOffset) 1827 { 1828 /* 1829 * This function needs to compute the size of directory records and 1830 * runs, file lengths, and set the appropriate variables both in 1831 * cd9660node and isoDirEntry 1832 */ 1833 int used_bytes = 0; 1834 int current_sector_usage = 0; 1835 cd9660node *child; 1836 fsinode *inode; 1837 int r; 1838 1839 assert(node != NULL); 1840 1841 1842 /* 1843 * NOTE : There needs to be some special case detection for 1844 * the "real root" node, since for it, node->node is undefined 1845 */ 1846 1847 node->fileDataSector = -1; 1848 1849 if (node->type & CD9660_TYPE_DIR) { 1850 node->fileRecordSize = cd9660_compute_record_size(node); 1851 /*Set what sector this directory starts in*/ 1852 node->fileDataSector = 1853 CD9660_BLOCKS(diskStructure.sectorSize,startOffset); 1854 1855 cd9660_bothendian_dword(node->fileDataSector, 1856 node->isoDirRecord->extent); 1857 1858 /* 1859 * First loop over children, need to know the size of 1860 * their directory records 1861 */ 1862 node->fileSectorsUsed = 1; 1863 TAILQ_FOREACH(child, &node->cn_children, cn_next_child) { 1864 node->fileDataLength += 1865 cd9660_compute_record_size(child); 1866 if ((cd9660_compute_record_size(child) + 1867 current_sector_usage) >= 1868 diskStructure.sectorSize) { 1869 current_sector_usage = 0; 1870 node->fileSectorsUsed++; 1871 } 1872 1873 current_sector_usage += 1874 cd9660_compute_record_size(child); 1875 } 1876 1877 cd9660_bothendian_dword(node->fileSectorsUsed * 1878 diskStructure.sectorSize,node->isoDirRecord->size); 1879 1880 /* 1881 * This should point to the sector after the directory 1882 * record (or, the first byte in that sector) 1883 */ 1884 used_bytes += node->fileSectorsUsed * diskStructure.sectorSize; 1885 1886 for (child = TAILQ_NEXT(node->dot_dot_record, cn_next_child); 1887 child != NULL; child = TAILQ_NEXT(child, cn_next_child)) { 1888 /* Directories need recursive call */ 1889 if (S_ISDIR(child->node->type)) { 1890 r = cd9660_compute_offsets(child, 1891 used_bytes + startOffset); 1892 1893 if (r != -1) 1894 used_bytes += r; 1895 else 1896 return -1; 1897 } 1898 } 1899 1900 /* Explicitly set the . and .. records */ 1901 cd9660_populate_dot_records(node); 1902 1903 /* Finally, do another iteration to write the file data*/ 1904 for (child = TAILQ_NEXT(node->dot_dot_record, cn_next_child); 1905 child != NULL; 1906 child = TAILQ_NEXT(child, cn_next_child)) { 1907 /* Files need extent set */ 1908 if (S_ISDIR(child->node->type)) 1909 continue; 1910 child->fileRecordSize = 1911 cd9660_compute_record_size(child); 1912 1913 child->fileSectorsUsed = 1914 CD9660_BLOCKS(diskStructure.sectorSize, 1915 child->fileDataLength); 1916 1917 inode = child->node->inode; 1918 if ((inode->flags & FI_ALLOCATED) == 0) { 1919 inode->ino = 1920 CD9660_BLOCKS(diskStructure.sectorSize, 1921 used_bytes + startOffset); 1922 inode->flags |= FI_ALLOCATED; 1923 used_bytes += child->fileSectorsUsed * 1924 diskStructure.sectorSize; 1925 } else { 1926 INODE_WARNX(("%s: already allocated inode %d " 1927 "data sectors at %" PRIu32, __func__, 1928 (int)inode->st.st_ino, inode->ino)); 1929 } 1930 child->fileDataSector = inode->ino; 1931 cd9660_bothendian_dword(child->fileDataSector, 1932 child->isoDirRecord->extent); 1933 } 1934 } 1935 1936 return used_bytes; 1937 } 1938 1939 #if 0 1940 /* Might get rid of this func */ 1941 static int 1942 cd9660_copy_stat_info(cd9660node *from, cd9660node *to, int file) 1943 { 1944 to->node->inode->st.st_dev = 0; 1945 to->node->inode->st.st_ino = 0; 1946 to->node->inode->st.st_size = 0; 1947 to->node->inode->st.st_blksize = from->node->inode->st.st_blksize; 1948 to->node->inode->st.st_atime = from->node->inode->st.st_atime; 1949 to->node->inode->st.st_mtime = from->node->inode->st.st_mtime; 1950 to->node->inode->st.st_ctime = from->node->inode->st.st_ctime; 1951 to->node->inode->st.st_uid = from->node->inode->st.st_uid; 1952 to->node->inode->st.st_gid = from->node->inode->st.st_gid; 1953 to->node->inode->st.st_mode = from->node->inode->st.st_mode; 1954 /* Clear out type */ 1955 to->node->inode->st.st_mode = to->node->inode->st.st_mode & ~(S_IFMT); 1956 if (file) 1957 to->node->inode->st.st_mode |= S_IFREG; 1958 else 1959 to->node->inode->st.st_mode |= S_IFDIR; 1960 return 1; 1961 } 1962 #endif 1963 1964 static cd9660node * 1965 cd9660_create_virtual_entry(const char *name, cd9660node *parent, int file, 1966 int insert) 1967 { 1968 cd9660node *temp; 1969 fsnode * tfsnode; 1970 1971 assert(parent != NULL); 1972 1973 temp = cd9660_allocate_cd9660node(); 1974 if (temp == NULL) 1975 return NULL; 1976 1977 if ((tfsnode = malloc(sizeof(fsnode))) == NULL) { 1978 CD9660_MEM_ALLOC_ERROR("cd9660_create_virtual_entry"); 1979 return NULL; 1980 } 1981 1982 /* Assume for now name is a valid length */ 1983 if ((tfsnode->name = malloc(strlen(name) + 1)) == NULL) { 1984 CD9660_MEM_ALLOC_ERROR("cd9660_create_virtual_entry"); 1985 return NULL; 1986 } 1987 1988 if ((temp->isoDirRecord = 1989 malloc(sizeof(iso_directory_record_cd9660))) == NULL) { 1990 CD9660_MEM_ALLOC_ERROR("cd9660_create_virtual_entry"); 1991 return NULL; 1992 } 1993 1994 strcpy(tfsnode->name, name); 1995 1996 cd9660_convert_filename(tfsnode->name, temp->isoDirRecord->name, file); 1997 1998 temp->node = tfsnode; 1999 temp->parent = parent; 2000 2001 if (insert) { 2002 if (temp->parent != NULL) { 2003 temp->level = temp->parent->level + 1; 2004 if (!TAILQ_EMPTY(&temp->parent->cn_children)) 2005 cd9660_sorted_child_insert(temp->parent, temp); 2006 else 2007 TAILQ_INSERT_HEAD(&temp->parent->cn_children, 2008 temp, cn_next_child); 2009 } 2010 } 2011 2012 if (parent->node != NULL) { 2013 tfsnode->type = parent->node->type; 2014 } 2015 2016 /* Clear out file type bits */ 2017 tfsnode->type &= ~(S_IFMT); 2018 if (file) 2019 tfsnode->type |= S_IFREG; 2020 else 2021 tfsnode->type |= S_IFDIR; 2022 2023 /* Indicate that there is no spec entry (inode) */ 2024 tfsnode->flags &= ~(FSNODE_F_HASSPEC); 2025 #if 0 2026 cd9660_copy_stat_info(parent, temp, file); 2027 #endif 2028 return temp; 2029 } 2030 2031 static cd9660node * 2032 cd9660_create_file(const char * name, cd9660node *parent, cd9660node *me) 2033 { 2034 cd9660node *temp; 2035 2036 temp = cd9660_create_virtual_entry(name,parent,1,1); 2037 if (temp == NULL) 2038 return NULL; 2039 2040 temp->fileDataLength = 0; 2041 2042 temp->type = CD9660_TYPE_FILE | CD9660_TYPE_VIRTUAL; 2043 2044 if ((temp->node->inode = calloc(1, sizeof(fsinode))) == NULL) 2045 return NULL; 2046 *temp->node->inode = *me->node->inode; 2047 2048 if (cd9960_translate_node_common(temp) == 0) 2049 return NULL; 2050 return temp; 2051 } 2052 2053 /* 2054 * Create a new directory which does not exist on disk 2055 * @param const char * name The name to assign to the directory 2056 * @param const char * parent Pointer to the parent directory 2057 * @returns cd9660node * Pointer to the new directory 2058 */ 2059 static cd9660node * 2060 cd9660_create_directory(const char *name, cd9660node *parent, cd9660node *me) 2061 { 2062 cd9660node *temp; 2063 2064 temp = cd9660_create_virtual_entry(name,parent,0,1); 2065 if (temp == NULL) 2066 return NULL; 2067 temp->node->type |= S_IFDIR; 2068 2069 temp->type = CD9660_TYPE_DIR | CD9660_TYPE_VIRTUAL; 2070 2071 if ((temp->node->inode = calloc(1, sizeof(fsinode))) == NULL) 2072 return NULL; 2073 *temp->node->inode = *me->node->inode; 2074 2075 if (cd9960_translate_node_common(temp) == 0) 2076 return NULL; 2077 return temp; 2078 } 2079 2080 static cd9660node * 2081 cd9660_create_special_directory(u_char type, cd9660node *parent) 2082 { 2083 cd9660node *temp, *first; 2084 char na[2]; 2085 2086 assert(parent != NULL); 2087 2088 if (type == CD9660_TYPE_DOT) 2089 na[0] = 0; 2090 else if (type == CD9660_TYPE_DOTDOT) 2091 na[0] = 1; 2092 else 2093 return 0; 2094 2095 na[1] = 0; 2096 if ((temp = cd9660_create_virtual_entry(na, parent, 0, 0)) == NULL) 2097 return NULL; 2098 2099 temp->parent = parent; 2100 temp->type = type; 2101 temp->isoDirRecord->length[0] = 34; 2102 /* Dot record is always first */ 2103 if (type == CD9660_TYPE_DOT) { 2104 parent->dot_record = temp; 2105 TAILQ_INSERT_HEAD(&parent->cn_children, temp, cn_next_child); 2106 /* DotDot should be second */ 2107 } else if (type == CD9660_TYPE_DOTDOT) { 2108 parent->dot_dot_record = temp; 2109 /* 2110 * If the first child is the dot record, insert 2111 * this second. Otherwise, insert it at the head. 2112 */ 2113 if ((first = TAILQ_FIRST(&parent->cn_children)) == NULL || 2114 (first->type & CD9660_TYPE_DOT) == 0) { 2115 TAILQ_INSERT_HEAD(&parent->cn_children, temp, 2116 cn_next_child); 2117 } else { 2118 TAILQ_INSERT_AFTER(&parent->cn_children, first, temp, 2119 cn_next_child); 2120 } 2121 } 2122 2123 return temp; 2124 } 2125 2126 int 2127 cd9660_add_generic_bootimage(const char *bootimage) 2128 { 2129 struct stat stbuf; 2130 2131 assert(bootimage != NULL); 2132 2133 if (*bootimage == '\0') { 2134 warnx("Error: Boot image must be a filename"); 2135 return 0; 2136 } 2137 2138 if ((diskStructure.generic_bootimage = strdup(bootimage)) == NULL) { 2139 warn("%s: strdup", __func__); 2140 return 0; 2141 } 2142 2143 /* Get information about the file */ 2144 if (lstat(diskStructure.generic_bootimage, &stbuf) == -1) 2145 err(EXIT_FAILURE, "%s: lstat(\"%s\")", __func__, 2146 diskStructure.generic_bootimage); 2147 2148 if (stbuf.st_size > 32768) { 2149 warnx("Error: Boot image must be no greater than 32768 bytes"); 2150 return 0; 2151 } 2152 2153 if (diskStructure.verbose_level > 0) { 2154 printf("Generic boot image image has size %lld\n", 2155 (long long)stbuf.st_size); 2156 } 2157 2158 diskStructure.has_generic_bootimage = 1; 2159 2160 return 1; 2161 } 2162