00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038
00039
00040
00041
00042
00043
00044
00045
00046
00047
00048
00049
00050
00051
00052
00053
00054
00055 #include "../libburn/libburn.h"
00056
00057
00058 #include <stdio.h>
00059 #include <ctype.h>
00060 #include <sys/types.h>
00061 #include <unistd.h>
00062 #include <string.h>
00063 #include <stdlib.h>
00064 #include <time.h>
00065 #include <errno.h>
00066 #include <sys/stat.h>
00067 #include <fcntl.h>
00068
00069
00070
00071
00072
00073
00074
00075
00076
00077 static struct burn_drive_info *drive_list;
00078
00079
00080
00081 static unsigned int drive_count;
00082
00083
00084
00085 static int drive_is_grabbed = 0;
00086
00087
00088 static int current_profile= -1;
00089 static char current_profile_name[80]= {""};
00090
00091
00092
00093
00094 int libburner_aquire_by_adr(char *drive_adr);
00095 int libburner_aquire_by_driveno(int *drive_no);
00096
00097
00098
00099
00100
00101
00102
00103
00104
00105
00106
00107
00108
00109
00110
00111 int libburner_aquire_drive(char *drive_adr, int *driveno)
00112 {
00113 int ret;
00114
00115 if(drive_adr != NULL && drive_adr[0] != 0)
00116 ret = libburner_aquire_by_adr(drive_adr);
00117 else
00118 ret = libburner_aquire_by_driveno(driveno);
00119 if (ret <= 0 || *driveno <= 0)
00120 return ret;
00121 burn_disc_get_profile(drive_list[0].drive, ¤t_profile,
00122 current_profile_name);
00123 if (current_profile_name[0])
00124 printf("Detected media type: %s\n", current_profile_name);
00125 return 1;
00126 }
00127
00128
00129
00130
00131
00132
00133 int libburner_aquire_by_adr(char *drive_adr)
00134 {
00135 int ret;
00136 char libburn_drive_adr[BURN_DRIVE_ADR_LEN];
00137
00138
00139 if (strncmp(drive_adr, "stdio:/dev/fd/", 14) == 0 ||
00140 strcmp(drive_adr, "stdio:-") == 0) {
00141 fprintf(stderr, "Will not work with pseudo-drive '%s'\n",
00142 drive_adr);
00143 return 0;
00144 }
00145
00146
00147 ret = burn_drive_convert_fs_adr(drive_adr, libburn_drive_adr);
00148 if (ret<=0) {
00149 fprintf(stderr, "Address does not lead to a CD burner: '%s'\n",
00150 drive_adr);
00151 return 0;
00152 }
00153 fprintf(stderr,"Aquiring drive '%s' ...\n", libburn_drive_adr);
00154 ret = burn_drive_scan_and_grab(&drive_list, libburn_drive_adr, 1);
00155 if (ret <= 0) {
00156 fprintf(stderr,"FAILURE with persistent drive address '%s'\n",
00157 libburn_drive_adr);
00158 } else {
00159 fprintf(stderr,"Done\n");
00160 drive_is_grabbed = 1;
00161 }
00162 return ret;
00163 }
00164
00165
00166
00167
00168
00169
00170
00171
00172
00173
00174
00175
00176
00177
00178 int libburner_aquire_by_driveno(int *driveno)
00179 {
00180 char adr[BURN_DRIVE_ADR_LEN];
00181 int ret, i;
00182
00183 printf("Beginning to scan for devices ...\n");
00184 while (!burn_drive_scan(&drive_list, &drive_count))
00185 usleep(100002);
00186 if (drive_count <= 0 && *driveno >= 0) {
00187 printf("FAILED (no drives found)\n");
00188 return 0;
00189 }
00190 printf("Done\n");
00191
00192
00193
00194
00195
00196
00197
00198
00199
00200
00201
00202
00203
00204
00205
00206 printf("\nOverview of accessible drives (%d found) :\n",
00207 drive_count);
00208 printf("-----------------------------------------------------------------------------\n");
00209 for (i = 0; i < drive_count; i++) {
00210 if (burn_drive_get_adr(&(drive_list[i]), adr) <=0)
00211 strcpy(adr, "-get_adr_failed-");
00212 printf("%d --drive '%s' : '%s' '%s'\n",
00213 i,adr,drive_list[i].vendor,drive_list[i].product);
00214 }
00215 printf("-----------------------------------------------------------------------------\n\n");
00216
00217
00218
00219
00220
00221
00222
00223
00224
00225
00226
00227
00228
00229
00230
00231
00232
00233
00234
00235
00236
00237
00238
00239
00240
00241
00242 if (*driveno < 0) {
00243 printf("Pseudo-drive \"-\" given : bus scanning done.\n");
00244 return 2;
00245 }
00246 if (drive_count <= *driveno) {
00247 fprintf(stderr,
00248 "Found only %d drives. Number %d not available.\n",
00249 drive_count, *driveno);
00250 return 0;
00251 }
00252
00253
00254 for (i = 0; i < drive_count; i++) {
00255 if (i == *driveno)
00256 continue;
00257 ret = burn_drive_info_forget(&(drive_list[i]),0);
00258 if (ret != 1)
00259 fprintf(stderr, "Cannot drop drive %d. Please report \"ret=%d\" to libburn-hackers@pykix.org\n",
00260 i, ret);
00261 else
00262 printf("Dropped unwanted drive %d\n",i);
00263 }
00264
00265 ret= burn_drive_grab(drive_list[*driveno].drive, 1);
00266 if (ret != 1)
00267 return 0;
00268 drive_is_grabbed = 1;
00269 return 1;
00270 }
00271
00272
00273
00274
00275
00276
00277
00278
00279
00280
00281 int libburner_blank_disc(struct burn_drive *drive, int blank_fast)
00282 {
00283 enum burn_disc_status disc_state;
00284 struct burn_progress p;
00285 double percent = 1.0;
00286
00287 disc_state = burn_disc_get_status(drive);
00288 printf(
00289 "Drive media status: %d (see libburn/libburn.h BURN_DISC_*)\n",
00290 disc_state);
00291 if (current_profile == 0x13) {
00292 ;
00293 } else if (disc_state == BURN_DISC_BLANK) {
00294 fprintf(stderr,
00295 "IDLE: Blank media detected. Will leave it untouched\n");
00296 return 2;
00297 } else if (disc_state == BURN_DISC_FULL ||
00298 disc_state == BURN_DISC_APPENDABLE) {
00299 ;
00300 } else if (disc_state == BURN_DISC_EMPTY) {
00301 fprintf(stderr,"FATAL: No media detected in drive\n");
00302 return 0;
00303 } else {
00304 fprintf(stderr,
00305 "FATAL: Unsuitable drive and media state\n");
00306 return 0;
00307 }
00308 if(!burn_disc_erasable(drive)) {
00309 fprintf(stderr,
00310 "FATAL : Media is not of erasable type\n");
00311 return 0;
00312 }
00313 printf(
00314 "Beginning to %s-blank media.\n", (blank_fast?"fast":"full"));
00315 burn_disc_erase(drive, blank_fast);
00316 sleep(1);
00317 while (burn_drive_get_status(drive, &p) != BURN_DRIVE_IDLE) {
00318 if(p.sectors>0 && p.sector>=0)
00319 percent = 1.0 + ((double) p.sector+1.0)
00320 / ((double) p.sectors) * 98.0;
00321 printf("Blanking ( %.1f%% done )\n", percent);
00322 sleep(1);
00323 }
00324 printf("Done\n");
00325 return 1;
00326 }
00327
00328
00329
00330
00331
00332
00333
00334
00335
00336
00337
00338
00339
00340
00341 int libburner_format(struct burn_drive *drive)
00342 {
00343 struct burn_progress p;
00344 double percent = 1.0;
00345 int ret, status, num_formats, format_flag= 0;
00346 off_t size = 0;
00347 unsigned dummy;
00348 enum burn_disc_status disc_state;
00349
00350 if (current_profile == 0x13) {
00351 fprintf(stderr, "IDLE: DVD-RW media is already formatted\n");
00352 return 2;
00353 } else if (current_profile == 0x41 || current_profile == 0x43) {
00354 disc_state = burn_disc_get_status(drive);
00355 if (disc_state != BURN_DISC_BLANK && current_profile == 0x41) {
00356 fprintf(stderr,
00357 "FATAL: BD-R is not blank. Cannot format.\n");
00358 return 0;
00359 }
00360 ret = burn_disc_get_formats(drive, &status, &size, &dummy,
00361 &num_formats);
00362 if (ret > 0 && status != BURN_FORMAT_IS_UNFORMATTED) {
00363 fprintf(stderr,
00364 "IDLE: BD media is already formatted\n");
00365 return 2;
00366 }
00367 size = 0;
00368 format_flag = 3<<1;
00369 } else if (current_profile == 0x14) {
00370 size = 128 * 1024 * 1024;
00371 format_flag = 1;
00372 } else {
00373 fprintf(stderr, "FATAL: Can only format DVD-RW or BD\n");
00374 return 0;
00375 }
00376
00377 printf("Beginning to format media.\n");
00378 burn_disc_format(drive, size, format_flag);
00379
00380 sleep(1);
00381 while (burn_drive_get_status(drive, &p) != BURN_DRIVE_IDLE) {
00382 if(p.sectors>0 && p.sector>=0)
00383 percent = 1.0 + ((double) p.sector+1.0)
00384 / ((double) p.sectors) * 98.0;
00385 printf("Formatting ( %.1f%% done )\n", percent);
00386 sleep(1);
00387 }
00388 burn_disc_get_profile(drive_list[0].drive, ¤t_profile,
00389 current_profile_name);
00390 if (current_profile == 0x14 || current_profile == 0x13)
00391 printf("Media type now: %4.4xh \"%s\"\n",
00392 current_profile, current_profile_name);
00393 if (current_profile == 0x14) {
00394 fprintf(stderr,
00395 "FATAL: Failed to change media profile to desired value\n");
00396 return 0;
00397 }
00398 return 1;
00399 }
00400
00401
00402
00403
00404
00405
00406
00407
00408
00409
00410
00411
00412
00413
00414
00415
00416
00417 int libburner_payload(struct burn_drive *drive,
00418 char source_adr[][4096], int source_adr_count,
00419 int multi, int simulate_burn, int all_tracks_type)
00420 {
00421 struct burn_source *data_src, *fifo_src[99];
00422 struct burn_disc *target_disc;
00423 struct burn_session *session;
00424 struct burn_write_opts *burn_options;
00425 enum burn_disc_status disc_state;
00426 struct burn_track *track, *tracklist[99];
00427 struct burn_progress progress;
00428 time_t start_time;
00429 int last_sector = 0, padding = 0, trackno, unpredicted_size = 0, fd;
00430 int fifo_chunksize = 2352, fifo_chunks = 1783;
00431 off_t fixed_size;
00432 char *adr, reasons[BURN_REASONS_LEN];
00433 struct stat stbuf;
00434
00435 if (all_tracks_type != BURN_AUDIO) {
00436 all_tracks_type = BURN_MODE1;
00437
00438 padding = 300*1024;
00439 fifo_chunksize = 2048;
00440 fifo_chunks = 2048;
00441 }
00442
00443 target_disc = burn_disc_create();
00444 session = burn_session_create();
00445 burn_disc_add_session(target_disc, session, BURN_POS_END);
00446
00447 for (trackno = 0 ; trackno < source_adr_count; trackno++) {
00448 tracklist[trackno] = track = burn_track_create();
00449 burn_track_define_data(track, 0, padding, 1, all_tracks_type);
00450
00451
00452 adr = source_adr[trackno];
00453 fixed_size = 0;
00454 if (adr[0] == '-' && adr[1] == 0) {
00455 fd = 0;
00456 } else {
00457 fd = open(adr, O_RDONLY);
00458 if (fd>=0)
00459 if (fstat(fd,&stbuf)!=-1)
00460 if((stbuf.st_mode&S_IFMT)==S_IFREG)
00461 fixed_size = stbuf.st_size;
00462 }
00463 if (fixed_size==0)
00464 unpredicted_size = 1;
00465
00466
00467 data_src = NULL;
00468 if (fd>=0)
00469 data_src = burn_fd_source_new(fd, -1, fixed_size);
00470 if (data_src == NULL) {
00471 fprintf(stderr,
00472 "FATAL: Could not open data source '%s'.\n",adr);
00473 if(errno!=0)
00474 fprintf(stderr,"(Most recent system error: %s )\n",
00475 strerror(errno));
00476 return 0;
00477 }
00478
00479 fifo_src[trackno] = burn_fifo_source_new(data_src,
00480 fifo_chunksize, fifo_chunks, 0);
00481 if (fifo_src[trackno] == NULL) {
00482 fprintf(stderr,
00483 "FATAL: Could not create fifo object of 4 MB\n");
00484 return 0;
00485 }
00486
00487
00488 if (burn_track_set_source(track, fifo_src[trackno])
00489 != BURN_SOURCE_OK) {
00490 fprintf(stderr,
00491 "FATAL: Cannot attach source object to track object\n");
00492 return 0;
00493 }
00494
00495 burn_session_add_track(session, track, BURN_POS_END);
00496 printf("Track %d : source is '%s'\n", trackno+1, adr);
00497
00498
00499 burn_source_free(data_src);
00500
00501 }
00502
00503
00504 disc_state = burn_disc_get_status(drive);
00505 if (disc_state != BURN_DISC_BLANK &&
00506 disc_state != BURN_DISC_APPENDABLE) {
00507 if (disc_state == BURN_DISC_FULL) {
00508 fprintf(stderr, "FATAL: Closed media with data detected. Need blank or appendable media.\n");
00509 if (burn_disc_erasable(drive))
00510 fprintf(stderr, "HINT: Try --blank_fast\n\n");
00511 } else if (disc_state == BURN_DISC_EMPTY)
00512 fprintf(stderr,"FATAL: No media detected in drive\n");
00513 else
00514 fprintf(stderr,
00515 "FATAL: Cannot recognize state of drive and media\n");
00516 return 0;
00517 }
00518
00519 burn_options = burn_write_opts_new(drive);
00520 burn_write_opts_set_perform_opc(burn_options, 0);
00521 burn_write_opts_set_multi(burn_options, !!multi);
00522 if(simulate_burn)
00523 printf("\n*** Will TRY to SIMULATE burning ***\n\n");
00524 burn_write_opts_set_simulate(burn_options, simulate_burn);
00525 burn_drive_set_speed(drive, 0, 0);
00526 burn_write_opts_set_underrun_proof(burn_options, 1);
00527 if (burn_write_opts_auto_write_type(burn_options, target_disc,
00528 reasons, 0) == BURN_WRITE_NONE) {
00529 fprintf(stderr, "FATAL: Failed to find a suitable write mode with this media.\n");
00530 fprintf(stderr, "Reasons given:\n%s\n", reasons);
00531 return 0;
00532 }
00533
00534 printf("Burning starts. With e.g. 4x media expect up to a minute of zero progress.\n");
00535 start_time = time(0);
00536 burn_disc_write(burn_options, target_disc);
00537
00538 burn_write_opts_free(burn_options);
00539 while (burn_drive_get_status(drive, NULL) == BURN_DRIVE_SPAWNING)
00540 usleep(100002);
00541 while (burn_drive_get_status(drive, &progress) != BURN_DRIVE_IDLE) {
00542 if (progress.sectors <= 0 ||
00543 (progress.sector >= progress.sectors - 1 &&
00544 !unpredicted_size) ||
00545 (unpredicted_size && progress.sector == last_sector))
00546 printf(
00547 "Thank you for being patient since %d seconds.",
00548 (int) (time(0) - start_time));
00549 else if(unpredicted_size)
00550 printf("Track %d : sector %d", progress.track+1,
00551 progress.sector);
00552 else
00553 printf("Track %d : sector %d of %d",progress.track+1,
00554 progress.sector, progress.sectors);
00555 last_sector = progress.sector;
00556 if (progress.track >= 0 && progress.track < source_adr_count) {
00557 int size, free_bytes, ret;
00558 char *status_text;
00559
00560 ret = burn_fifo_inquire_status(
00561 fifo_src[progress.track], &size, &free_bytes,
00562 &status_text);
00563 if (ret >= 0 )
00564 printf(" [fifo %s, %2d%% fill]", status_text,
00565 (int) (100.0 - 100.0 *
00566 ((double) free_bytes) /
00567 (double) size));
00568 }
00569 printf("\n");
00570 sleep(1);
00571 }
00572 printf("\n");
00573
00574 for (trackno = 0 ; trackno < source_adr_count; trackno++) {
00575 burn_source_free(fifo_src[trackno]);
00576 burn_track_free(tracklist[trackno]);
00577 }
00578 burn_session_free(session);
00579 burn_disc_free(target_disc);
00580 if (multi && current_profile != 0x1a && current_profile != 0x13 &&
00581 current_profile != 0x12 && current_profile != 0x43)
00582
00583 printf("NOTE: Media left appendable.\n");
00584 if (simulate_burn)
00585 printf("\n*** Did TRY to SIMULATE burning ***\n\n");
00586 return 1;
00587 }
00588
00589
00590
00591 static char drive_adr[BURN_DRIVE_ADR_LEN] = {""};
00592 static int driveno = 0;
00593 static int do_blank = 0;
00594 static char source_adr[99][4096];
00595 static int source_adr_count = 0;
00596 static int do_multi = 0;
00597 static int simulate_burn = 0;
00598 static int all_tracks_type = BURN_MODE1;
00599
00600
00601
00602
00603 int libburner_setup(int argc, char **argv)
00604 {
00605 int i, insuffient_parameters = 0, print_help = 0;
00606
00607 for (i = 1; i < argc; ++i) {
00608 if (!strcmp(argv[i], "--audio")) {
00609 all_tracks_type = BURN_AUDIO;
00610
00611 } else if (!strcmp(argv[i], "--blank_fast")) {
00612 do_blank = 1;
00613
00614 } else if (!strcmp(argv[i], "--blank_full")) {
00615 do_blank = 2;
00616
00617 } else if (!strcmp(argv[i], "--burn_for_real")) {
00618 simulate_burn = 0;
00619
00620 } else if (!strcmp(argv[i], "--drive")) {
00621 ++i;
00622 if (i >= argc) {
00623 fprintf(stderr,"--drive requires an argument\n");
00624 return 1;
00625 } else if (strcmp(argv[i], "-") == 0) {
00626 drive_adr[0] = 0;
00627 driveno = -1;
00628 } else if (isdigit(argv[i][0])) {
00629 drive_adr[0] = 0;
00630 driveno = atoi(argv[i]);
00631 } else {
00632 if(strlen(argv[i]) >= BURN_DRIVE_ADR_LEN) {
00633 fprintf(stderr,"--drive address too long (max. %d)\n",
00634 BURN_DRIVE_ADR_LEN-1);
00635 return 2;
00636 }
00637 strcpy(drive_adr, argv[i]);
00638 }
00639 } else if ((!strcmp(argv[i], "--format_overwrite")) ||
00640 (!strcmp(argv[i], "--format"))) {
00641 do_blank = 101;
00642
00643 } else if (!strcmp(argv[i], "--multi")) {
00644 do_multi = 1;
00645
00646 } else if (!strcmp(argv[i], "--stdin_size")) {
00647 i++;
00648
00649 } else if (!strcmp(argv[i], "--try_to_simulate")) {
00650 simulate_burn = 1;
00651
00652 } else if (!strcmp(argv[i], "--help")) {
00653 print_help = 1;
00654
00655 } else if (!strncmp(argv[i], "--",2)) {
00656 fprintf(stderr, "Unidentified option: %s\n", argv[i]);
00657 return 7;
00658 } else {
00659 if(strlen(argv[i]) >= 4096) {
00660 fprintf(stderr, "Source address too long (max. %d)\n", 4096-1);
00661 return 5;
00662 }
00663 if(source_adr_count >= 99) {
00664 fprintf(stderr, "Too many tracks (max. 99)\n");
00665 return 6;
00666 }
00667 strcpy(source_adr[source_adr_count], argv[i]);
00668 source_adr_count++;
00669 }
00670 }
00671 insuffient_parameters = 1;
00672 if (driveno < 0)
00673 insuffient_parameters = 0;
00674 if (source_adr_count > 0)
00675 insuffient_parameters = 0;
00676 if (do_blank)
00677 insuffient_parameters = 0;
00678 if (print_help || insuffient_parameters ) {
00679 printf("Usage: %s\n", argv[0]);
00680 printf(" [--drive <address>|<driveno>|\"-\"] [--audio]\n");
00681 printf(" [--blank_fast|--blank_full|--format] [--try_to_simulate]\n");
00682 printf(" [--multi] [<one or more imagefiles>|\"-\"]\n");
00683 printf("Examples\n");
00684 printf("A bus scan (needs rw-permissions to see a drive):\n");
00685 printf(" %s --drive -\n",argv[0]);
00686 printf("Burn a file to drive chosen by number, leave appendable:\n");
00687 printf(" %s --drive 0 --multi my_image_file\n", argv[0]);
00688 printf("Burn a file to drive chosen by persistent address, close:\n");
00689 printf(" %s --drive /dev/hdc my_image_file\n", argv[0]);
00690 printf("Blank a used CD-RW (is combinable with burning in one run):\n");
00691 printf(" %s --drive /dev/hdc --blank_fast\n",argv[0]);
00692 printf("Blank a used DVD-RW (is combinable with burning in one run):\n");
00693 printf(" %s --drive /dev/hdc --blank_full\n",argv[0]);
00694 printf("Format a DVD-RW, BD-RE or BD-R:\n");
00695 printf(" %s --drive /dev/hdc --format\n", argv[0]);
00696 printf("Burn two audio tracks (to CD only):\n");
00697 printf(" lame --decode -t /path/to/track1.mp3 track1.cd\n");
00698 printf(" test/dewav /path/to/track2.wav -o track2.cd\n");
00699 printf(" %s --drive /dev/hdc --audio track1.cd track2.cd\n", argv[0]);
00700 printf("Burn a compressed afio archive on-the-fly:\n");
00701 printf(" ( cd my_directory ; find . -print | afio -oZ - ) | \\\n");
00702 printf(" %s --drive /dev/hdc -\n", argv[0]);
00703 printf("To be read from *not mounted* media via: afio -tvZ /dev/hdc\n");
00704 if (insuffient_parameters)
00705 return 6;
00706 }
00707 return 0;
00708 }
00709
00710
00711 int main(int argc, char **argv)
00712 {
00713 int ret;
00714
00715 ret = libburner_setup(argc, argv);
00716 if (ret)
00717 exit(ret);
00718
00719 printf("Initializing libburnia-project.org ...\n");
00720 if (burn_initialize())
00721 printf("Done\n");
00722 else {
00723 printf("FAILED\n");
00724 fprintf(stderr,"\nFATAL: Failed to initialize.\n");
00725 exit(33);
00726 }
00727
00728
00729 burn_msgs_set_severities("NEVER", "SORRY", "libburner : ");
00730
00731
00732
00733 burn_set_signal_handling("libburner : ", NULL, 0);
00734
00735
00736 ret = libburner_aquire_drive(drive_adr, &driveno);
00737 if (ret<=0) {
00738 fprintf(stderr,"\nFATAL: Failed to aquire drive.\n");
00739 { ret = 34; goto finish_libburn; }
00740 }
00741 if (ret == 2)
00742 { ret = 0; goto release_drive; }
00743 if (do_blank) {
00744 if (do_blank > 100)
00745 ret = libburner_format(drive_list[driveno].drive);
00746 else
00747 ret = libburner_blank_disc(drive_list[driveno].drive,
00748 do_blank == 1);
00749 if (ret<=0)
00750 { ret = 36; goto release_drive; }
00751 }
00752 if (source_adr_count > 0) {
00753 ret = libburner_payload(drive_list[driveno].drive,
00754 source_adr, source_adr_count,
00755 do_multi, simulate_burn, all_tracks_type);
00756 if (ret<=0)
00757 { ret = 38; goto release_drive; }
00758 }
00759 ret = 0;
00760 release_drive:;
00761 if (drive_is_grabbed)
00762 burn_drive_release(drive_list[driveno].drive, 0);
00763
00764 finish_libburn:;
00765
00766
00767
00768
00769 burn_finish();
00770 exit(ret);
00771 }
00772
00773
00774
00775
00776
00777
00778
00779
00780
00781
00782
00783
00784
00785
00786
00787
00788