1 /*
2  * Copyright (c) 2011 Anton Khirnov <anton@khirnov.net>
3  *
4  * This file is part of FFmpeg.
5  *
6  * FFmpeg is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * FFmpeg is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with FFmpeg; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20 
21 /**
22  * @file
23  * libcdio CD grabbing
24  */
25 
26 #include "config.h"
27 
28 #if HAVE_CDIO_PARANOIA_H
29 #include <cdio/cdda.h>
30 #include <cdio/paranoia.h>
31 #elif HAVE_CDIO_PARANOIA_PARANOIA_H
32 #include <cdio/paranoia/cdda.h>
33 #include <cdio/paranoia/paranoia.h>
34 #endif
35 
36 #include "libavutil/log.h"
37 #include "libavutil/opt.h"
38 
39 #include "libavformat/avformat.h"
40 #include "libavformat/demux.h"
41 #include "libavformat/internal.h"
42 
43 typedef struct CDIOContext {
44     const AVClass       *class;
45     cdrom_drive_t       *drive;
46     cdrom_paranoia_t *paranoia;
47     int32_t last_sector;
48 
49     /* private options */
50     int speed;
51     int paranoia_mode;
52 } CDIOContext;
53 
read_header(AVFormatContext *ctx)54 static av_cold int read_header(AVFormatContext *ctx)
55 {
56     CDIOContext *s = ctx->priv_data;
57     AVStream *st;
58     int ret, i;
59     char *err = NULL;
60 
61     if (!(st = avformat_new_stream(ctx, NULL)))
62         return AVERROR(ENOMEM);
63     s->drive = cdio_cddap_identify(ctx->url, CDDA_MESSAGE_LOGIT, &err);
64     if (!s->drive) {
65         av_log(ctx, AV_LOG_ERROR, "Could not open drive %s.\n", ctx->url);
66         return AVERROR(EINVAL);
67     }
68     if (err) {
69         av_log(ctx, AV_LOG_VERBOSE, "%s\n", err);
70         free(err);
71     }
72     if ((ret = cdio_cddap_open(s->drive)) < 0 || !s->drive->opened) {
73         av_log(ctx, AV_LOG_ERROR, "Could not open disk in drive %s.\n", ctx->url);
74         return AVERROR(EINVAL);
75     }
76 
77     cdio_cddap_verbose_set(s->drive, CDDA_MESSAGE_LOGIT, CDDA_MESSAGE_LOGIT);
78     if (s->speed)
79         cdio_cddap_speed_set(s->drive, s->speed);
80 
81     s->paranoia = cdio_paranoia_init(s->drive);
82     if (!s->paranoia) {
83         av_log(ctx, AV_LOG_ERROR, "Could not init paranoia.\n");
84         return AVERROR(EINVAL);
85     }
86     cdio_paranoia_modeset(s->paranoia, s->paranoia_mode);
87 
88     st->codecpar->codec_type      = AVMEDIA_TYPE_AUDIO;
89     if (s->drive->bigendianp)
90         st->codecpar->codec_id    = AV_CODEC_ID_PCM_S16BE;
91     else
92         st->codecpar->codec_id    = AV_CODEC_ID_PCM_S16LE;
93     st->codecpar->sample_rate     = 44100;
94     st->codecpar->ch_layout       = (AVChannelLayout)AV_CHANNEL_LAYOUT_STEREO;
95     if (s->drive->audio_last_sector != CDIO_INVALID_LSN &&
96         s->drive->audio_first_sector != CDIO_INVALID_LSN)
97         st->duration           = s->drive->audio_last_sector - s->drive->audio_first_sector;
98     else if (s->drive->tracks)
99         st->duration = s->drive->disc_toc[s->drive->tracks].dwStartSector;
100     avpriv_set_pts_info(st, 64, CDIO_CD_FRAMESIZE_RAW,
101                         2 * st->codecpar->ch_layout.nb_channels * st->codecpar->sample_rate);
102 
103     for (i = 0; i < s->drive->tracks; i++) {
104         char title[16];
105         snprintf(title, sizeof(title), "track %02d", s->drive->disc_toc[i].bTrack);
106         avpriv_new_chapter(ctx, i, st->time_base, s->drive->disc_toc[i].dwStartSector,
107                        s->drive->disc_toc[i+1].dwStartSector, title);
108     }
109 
110     s->last_sector = cdio_cddap_disc_lastsector(s->drive);
111 
112     return 0;
113 }
114 
read_packet(AVFormatContext *ctx, AVPacket *pkt)115 static int read_packet(AVFormatContext *ctx, AVPacket *pkt)
116 {
117     CDIOContext *s = ctx->priv_data;
118     int ret;
119     uint16_t *buf;
120     char *err = NULL;
121 
122     buf = cdio_paranoia_read(s->paranoia, NULL);
123     if (!buf)
124         return AVERROR_EOF;
125 
126     if (err = cdio_cddap_errors(s->drive)) {
127         av_log(ctx, AV_LOG_ERROR, "%s\n", err);
128         free(err);
129         err = NULL;
130     }
131     if (err = cdio_cddap_messages(s->drive)) {
132         av_log(ctx, AV_LOG_VERBOSE, "%s\n", err);
133         free(err);
134         err = NULL;
135     }
136 
137     if ((ret = av_new_packet(pkt, CDIO_CD_FRAMESIZE_RAW)) < 0)
138         return ret;
139     memcpy(pkt->data, buf, CDIO_CD_FRAMESIZE_RAW);
140     return 0;
141 }
142 
read_close(AVFormatContext *ctx)143 static av_cold int read_close(AVFormatContext *ctx)
144 {
145     CDIOContext *s = ctx->priv_data;
146     cdio_paranoia_free(s->paranoia);
147     cdio_cddap_close(s->drive);
148     return 0;
149 }
150 
read_seek(AVFormatContext *ctx, int stream_index, int64_t timestamp, int flags)151 static int read_seek(AVFormatContext *ctx, int stream_index, int64_t timestamp,
152                      int flags)
153 {
154     CDIOContext *s = ctx->priv_data;
155     AVStream *st = ctx->streams[0];
156 
157     cdio_paranoia_seek(s->paranoia, timestamp, SEEK_SET);
158     avpriv_update_cur_dts(ctx, st, timestamp);
159     return 0;
160 }
161 
162 #define OFFSET(x) offsetof(CDIOContext, x)
163 #define DEC AV_OPT_FLAG_DECODING_PARAM
164 static const AVOption options[] = {
165     { "speed",              "set drive reading speed", OFFSET(speed),         AV_OPT_TYPE_INT,   { .i64 = 0 }, 0,       INT_MAX, DEC },
166     { "paranoia_mode",      "set error recovery mode", OFFSET(paranoia_mode), AV_OPT_TYPE_FLAGS, { .i64 = PARANOIA_MODE_DISABLE }, INT_MIN, INT_MAX, DEC, "paranoia_mode" },
167         { "disable",        "apply no fixups",                      0,    AV_OPT_TYPE_CONST, { .i64 = PARANOIA_MODE_DISABLE },   0, 0, DEC, "paranoia_mode" },
168         { "verify",         "verify data integrity in overlap area", 0,   AV_OPT_TYPE_CONST, { .i64 = PARANOIA_MODE_VERIFY },    0, 0, DEC, "paranoia_mode" },
169         { "overlap",        "perform overlapped reads",             0,    AV_OPT_TYPE_CONST, { .i64 = PARANOIA_MODE_OVERLAP },   0, 0, DEC, "paranoia_mode" },
170         { "neverskip",      "do not skip failed reads",             0,    AV_OPT_TYPE_CONST, { .i64 = PARANOIA_MODE_NEVERSKIP }, 0, 0, DEC, "paranoia_mode" },
171         { "full",           "apply all recovery modes",             0,    AV_OPT_TYPE_CONST, { .i64 = PARANOIA_MODE_FULL },      0, 0, DEC, "paranoia_mode" },
172     { NULL },
173 };
174 
175 static const AVClass libcdio_class = {
176     .class_name = "libcdio indev",
177     .item_name  = av_default_item_name,
178     .option     = options,
179     .version    = LIBAVUTIL_VERSION_INT,
180     .category   = AV_CLASS_CATEGORY_DEVICE_AUDIO_INPUT,
181 };
182 
183 const AVInputFormat ff_libcdio_demuxer = {
184     .name           = "libcdio",
185     .read_header    = read_header,
186     .read_packet    = read_packet,
187     .read_close     = read_close,
188     .read_seek      = read_seek,
189     .priv_data_size = sizeof(CDIOContext),
190     .flags          = AVFMT_NOFILE,
191     .priv_class     = &libcdio_class,
192 };
193