Add example test program and some changes to make it work
Vreixo Formoso Lopes
16 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