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