access: dtv: add en50221_capmt
[vlc.git] / modules / access / imem-access.c
1 /*****************************************************************************
2  * imem-access.c: In-memory bit stream input for VLC
3  *****************************************************************************
4  * Copyright (C) 2015 RĂ©mi Denis-Courmont
5  *
6  * This program is free software; you can redistribute it and/or modify it
7  * under the terms of the GNU Lesser General Public License as published by
8  * the Free Software Foundation; either version 2.1 of the License, or
9  * (at your option) any later version.
10  *
11  * This program 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
14  * GNU Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public License
17  * along with this program; if not, write to the Free Software Foundation,
18  * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
19  *****************************************************************************/
20
21 #ifdef HAVE_CONFIG_H
22 # include "config.h"
23 #endif
24 #include <assert.h>
25 #include <stdint.h>
26
27 #include <vlc_common.h>
28 #include <vlc_access.h>
29 #include <vlc_plugin.h>
30
31 struct access_sys_t
32 {
33     void *opaque;
34     ssize_t (*read_cb)(void *, unsigned char *, size_t);
35     int (*seek_cb)(void *, uint64_t);
36     void (*close_cb)(void *);
37     uint64_t size;
38 };
39
40 static ssize_t Read(access_t *access, uint8_t *buf, size_t len)
41 {
42     access_sys_t *sys = access->p_sys;
43
44     ssize_t val = sys->read_cb(sys->opaque, buf, len);
45
46     if (val < 0) {
47         msg_Err(access, "read error");
48         val = 0;
49     }
50
51     if (val <= 0)
52         access->info.b_eof = true;
53
54     return val;
55 }
56
57 static int Seek(access_t *access, uint64_t offset)
58 {
59     access_sys_t *sys = access->p_sys;
60
61     assert(sys->seek_cb != NULL);
62
63     if (sys->seek_cb(sys->opaque, offset) != 0)
64         return VLC_EGENERIC;
65
66    access->info.b_eof = false;
67    return VLC_SUCCESS;
68 }
69
70 static int Control(access_t *access, int query, va_list args)
71 {
72     access_sys_t *sys = access->p_sys;
73
74     switch (query)
75     {
76         case ACCESS_CAN_SEEK:
77             *va_arg(args, bool *) = sys->seek_cb != NULL;
78             break;
79
80         case ACCESS_CAN_FASTSEEK:
81             *va_arg(args, bool *) = false;
82             break;
83
84         case ACCESS_CAN_PAUSE:
85         case ACCESS_CAN_CONTROL_PACE:
86             *va_arg(args, bool *) = sys->seek_cb != NULL;
87             break;
88
89         case ACCESS_GET_SIZE:
90             if (sys->size == UINT64_MAX)
91                 return VLC_EGENERIC;
92             *va_arg(args, uint64_t *) = sys->size;
93             break;
94
95         case ACCESS_GET_PTS_DELAY:
96             *va_arg(args, int64_t *) = DEFAULT_PTS_DELAY;
97             break;
98
99         case ACCESS_SET_PAUSE_STATE:
100             break;
101
102         default:
103             return VLC_EGENERIC;
104     }
105     (void) access;
106     return VLC_SUCCESS;
107 }
108
109 static int open_cb_default(void *opaque, void **datap, size_t *sizep)
110 {
111     *datap = opaque;
112     (void) sizep;
113     return 0;
114 }
115
116 static int Open(vlc_object_t *object)
117 {
118     access_t *access = (access_t *)object;
119
120     access_sys_t *sys = malloc(sizeof (*sys));
121     if (unlikely(sys == NULL))
122         return VLC_ENOMEM;
123
124     int (*open_cb)(void *, void **, uint64_t *);
125     void *opaque;
126
127     opaque = var_InheritAddress(access, "imem-data");
128     open_cb = var_InheritAddress(access, "imem-open");
129     sys->opaque = NULL;
130     sys->read_cb = var_InheritAddress(access, "imem-read");
131     sys->seek_cb = var_InheritAddress(access, "imem-seek");
132     sys->close_cb = var_InheritAddress(access, "imem-close");
133     sys->size = UINT64_MAX;
134
135     if (open_cb == NULL)
136         open_cb = open_cb_default;
137     if (sys->read_cb == NULL)
138         goto error;
139
140     if (open_cb(opaque, &sys->opaque, &sys->size)) {
141         msg_Err(access, "open error");
142         goto error;
143     }
144
145     access->pf_read = Read;
146     access->pf_block = NULL;
147     access->pf_seek = (sys->seek_cb != NULL) ? Seek : NULL;
148     access->pf_control = Control;
149
150     access->p_sys = sys;
151     access_InitFields(access);
152     return VLC_SUCCESS;
153 error:
154     free(sys);
155     return VLC_EGENERIC;
156 }
157
158 static void Close(vlc_object_t *object)
159 {
160     access_t *access = (access_t *)object;
161     access_sys_t *sys = access->p_sys;
162
163     if (sys->close_cb != NULL)
164         sys->close_cb(sys->opaque);
165     free(sys);
166 }
167
168 vlc_module_begin()
169     set_shortname(N_("Nemory stream"))
170     set_description(N_("In-memory stream input"))
171     set_category(CAT_INPUT)
172     set_subcategory(SUBCAT_INPUT_ACCESS)
173
174     add_shortcut("imem")
175     set_capability("access", 0)
176     set_callbacks(Open, Close)
177 vlc_module_end()