Vreixo Formoso Lopes
15 years ago
parent
2461231664
commit
4b1bcb6a37
@ -0,0 +1,187 @@ |
||||
/*
|
||||
Little test program for libisoburn. |
||||
It grows an iso filesystem on a disc. |
||||
|
||||
Copyright 2007 Vreixo Formoso Lopes <metalpain2002@yahoo.es> |
||||
*/ |
||||
|
||||
|
||||
#include <stdlib.h> |
||||
#include <stdio.h> |
||||
#include <unistd.h> |
||||
#include <getopt.h> |
||||
#include <err.h> |
||||
|
||||
#include <libisofs/libisofs.h> |
||||
#include <libburn/libburn.h> |
||||
#include "../src/libisoburn.h" |
||||
|
||||
const char * const optstring = "JRh"; |
||||
extern char *optarg; |
||||
extern int optind; |
||||
|
||||
static |
||||
void usage() |
||||
{ |
||||
printf("test [OPTIONS] DRIVE DIRECTORY\n"); |
||||
} |
||||
|
||||
static |
||||
void help() |
||||
{ |
||||
printf( |
||||
"Options:\n" |
||||
" -J Add Joliet support\n" |
||||
" -R Add Rock Ridge support\n" |
||||
" -h Print this message\n" |
||||
); |
||||
} |
||||
|
||||
int main(int argc, char **argv) |
||||
{ |
||||
struct burn_drive_info *drives; |
||||
struct iso_volset *volset; |
||||
struct burn_drive *drive; |
||||
struct iso_tree_node_dir *root; |
||||
struct burn_disc *disc; |
||||
enum burn_disc_status state; |
||||
struct isoburn_read_opts ropts; |
||||
int c; |
||||
struct iso_tree_radd_dir_behavior behav = {0,0,0}; |
||||
int flags=0; |
||||
int ret=0; |
||||
|
||||
while ((c = getopt(argc, argv, optstring)) != -1) { |
||||
switch(c) { |
||||
case 'h': |
||||
usage(); |
||||
help(); |
||||
exit(0); |
||||
break; |
||||
case 'J': |
||||
flags |= ECMA119_JOLIET; |
||||
break; |
||||
case 'R': |
||||
flags |= ECMA119_ROCKRIDGE; |
||||
break; |
||||
case '?': |
||||
usage();
|
||||
exit(1); |
||||
break; |
||||
} |
||||
} |
||||
|
||||
if (argc < optind + 1) { |
||||
printf("Please supply device name\n"); |
||||
usage(); |
||||
return 1; |
||||
} |
||||
if (argc < optind + 2) { |
||||
printf("Please supply directory to add to disc\n"); |
||||
usage(); |
||||
return 1; |
||||
} |
||||
|
||||
|
||||
if (!isoburn_initialize()) { |
||||
err(1, "Can't init libisoburn"); |
||||
} |
||||
|
||||
// TODO change this. maybe we can add wrapp in libisoburn
|
||||
iso_msgs_set_severities("NEVER", "DEBUG", "libisofs : "); |
||||
burn_msgs_set_severities("NEVER", "DEBUG", "libburn : "); |
||||
|
||||
printf("Growing drive %s\n", argv[optind]); |
||||
|
||||
if (isoburn_drive_scan_and_grab(&drives, argv[optind], 0) <= 0) { |
||||
err(1, "Can't open device. Are you sure it is a valid drive?\n"); |
||||
} |
||||
drive = drives[0].drive; |
||||
|
||||
/* check for invalid state */ |
||||
state = isoburn_disc_get_status(drive); |
||||
if (state != BURN_DISC_BLANK && state != BURN_DISC_APPENDABLE) { |
||||
printf("Unsuitable disc status"); |
||||
goto exit_cleanup; |
||||
} |
||||
|
||||
/* fill read opts */ |
||||
ropts.norock = 0; |
||||
ropts.nojoliet = 0; |
||||
ropts.preferjoliet = 0; |
||||
ropts.uid = 0; |
||||
ropts.gid = 0; |
||||
ropts.mode = 0555; |
||||
|
||||
if (isoburn_read_volset(drive, &ropts, &volset) <= 0) { |
||||
printf("Can't read volset\n"); |
||||
goto exit_cleanup; |
||||
} |
||||
printf("Volset read\n"); |
||||
fflush(stdout); |
||||
|
||||
if (!volset) { |
||||
// TODO this piece of code should be inside libisoburn
|
||||
struct iso_volume *volume; |
||||
|
||||
root = iso_tree_new_root();
|
||||
volume = iso_volume_new_with_root("NEW DISC", "", "LIBISOBURN", root); |
||||
volset = iso_volset_new( volume, "VOLSETID" ); |
||||
} |
||||
root = iso_volume_get_root(iso_volset_get_volume(volset, 0)); |
||||
|
||||
/* add a new dir */ |
||||
iso_tree_radd_dir(root, argv[optind+1], &behav); |
||||
|
||||
if (isoburn_prepare_disc(drive, &disc) <= 0) { |
||||
printf("Can't prepare disc"); |
||||
goto volset_cleanup; |
||||
} |
||||
|
||||
/* a. write the new image */ |
||||
printf("Adding new data...\n"); |
||||
{ |
||||
struct burn_write_opts *burn_options; |
||||
struct burn_progress progress; |
||||
char reasons[BURN_REASONS_LEN]; |
||||
|
||||
burn_options = burn_write_opts_new(drive); |
||||
burn_drive_set_speed(drive, 0, 0); |
||||
burn_write_opts_set_underrun_proof(burn_options, 1); |
||||
|
||||
if (burn_write_opts_auto_write_type(burn_options, disc, |
||||
reasons, 0) == BURN_WRITE_NONE) { |
||||
printf("Failed to find a suitable write mode:\n%s\n", reasons); |
||||
ret = 1; |
||||
goto volset_cleanup; |
||||
} |
||||
|
||||
/* ok, write the new track */ |
||||
isoburn_disc_write(burn_options, disc); |
||||
burn_write_opts_free(burn_options); |
||||
|
||||
while (burn_drive_get_status(drive, NULL) == BURN_DRIVE_SPAWNING) |
||||
usleep(1002); |
||||
|
||||
while (burn_drive_get_status(drive, &progress) != BURN_DRIVE_IDLE) { |
||||
printf("Writing: sector %d of %d\n", progress.sector, progress.sectors); |
||||
sleep(1); |
||||
} |
||||
} |
||||
|
||||
/* b. write the new vol desc */ |
||||
printf("Writing the new vol desc...\n"); |
||||
if (isoburn_activate_session(drive) <= 0) { |
||||
printf("Ups, new vol desc write failed\n"); |
||||
} |
||||
|
||||
volset_cleanup:; |
||||
iso_volset_free(volset); |
||||
|
||||
exit_cleanup:; |
||||
isoburn_drive_release(drive, 0); |
||||
isoburn_finish(); |
||||
|
||||
exit(ret); |
||||
} |
||||
|
||||
Loading…
Reference in new issue