git.sesse.net Git - ffmpeg/blob - libavformat/libsmbclient.c

   1 /*
   2  * Copyright (c) 2014 Lukasz Marek <lukasz.m.luki@gmail.com>
   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 #include <libsmbclient.h>
  22 #include "libavutil/avstring.h"
  23 #include "libavutil/opt.h"
  24 #include "avformat.h"
  25 #include "internal.h"
  26 #include "url.h"
  27
  28 typedef struct {
  29     const AVClass *class;
  30     SMBCCTX *ctx;
  31     int fd;
  32     int64_t filesize;
  33     int trunc;
  34     int timeout;
  35     char *workgroup;
  36 } LIBSMBContext;
  37
  38 static void libsmbc_get_auth_data(SMBCCTX *c, const char *server, const char *share,
  39                                   char *workgroup, int workgroup_len,
  40                                   char *username, int username_len,
  41                                   char *password, int password_len)
  42 {
  43     /* Do nothing yet. Credentials are passed via url.
  44      * Callback must exists, there might be a segmentation fault otherwise. */
  45 }
  46
  47 static av_cold int libsmbc_connect(URLContext *h)
  48 {
  49     LIBSMBContext *libsmbc = h->priv_data;
  50
  51     libsmbc->ctx = smbc_new_context();
  52     if (!libsmbc->ctx) {
  53         av_log(h, AV_LOG_ERROR, "Cannot create context: %s.\n", strerror(errno));
  54         return AVERROR(errno);
  55     }
  56     if (!smbc_init_context(libsmbc->ctx)) {
  57         av_log(h, AV_LOG_ERROR, "Cannot initialize context: %s.\n", strerror(errno));
  58         return AVERROR(errno);
  59     }
  60     smbc_set_context(libsmbc->ctx);
  61
  62     smbc_setOptionUserData(libsmbc->ctx, h);
  63     smbc_setFunctionAuthDataWithContext(libsmbc->ctx, libsmbc_get_auth_data);
  64
  65     if (libsmbc->timeout != -1)
  66         smbc_setTimeout(libsmbc->ctx, libsmbc->timeout);
  67     if (libsmbc->workgroup)
  68         smbc_setWorkgroup(libsmbc->ctx, libsmbc->workgroup);
  69
  70     if (smbc_init(NULL, 0) < 0) {
  71         av_log(h, AV_LOG_ERROR, "Initialization failed: %s\n", strerror(errno));
  72         return AVERROR(errno);
  73     }
  74     return 0;
  75 }
  76
  77 static av_cold int libsmbc_close(URLContext *h)
  78 {
  79     LIBSMBContext *libsmbc = h->priv_data;
  80     if (libsmbc->fd >= 0) {
  81         smbc_close(libsmbc->fd);
  82         libsmbc->fd = -1;
  83     }
  84     if (libsmbc->ctx) {
  85         smbc_free_context(libsmbc->ctx, 1);
  86         libsmbc->ctx = NULL;
  87     }
  88     return 0;
  89 }
  90
  91 static av_cold int libsmbc_open(URLContext *h, const char *url, int flags)
  92 {
  93     LIBSMBContext *libsmbc = h->priv_data;
  94     int access, ret;
  95     struct stat st;
  96
  97     libsmbc->fd = -1;
  98     libsmbc->filesize = -1;
  99
 100     if ((ret = libsmbc_connect(h)) < 0)
 101         goto fail;
 102
 103     if ((flags & AVIO_FLAG_WRITE) && (flags & AVIO_FLAG_READ)) {
 104         access = O_CREAT | O_RDWR;
 105         if (libsmbc->trunc)
 106             access |= O_TRUNC;
 107     } else if (flags & AVIO_FLAG_WRITE) {
 108         access = O_CREAT | O_WRONLY;
 109         if (libsmbc->trunc)
 110             access |= O_TRUNC;
 111     } else
 112         access = O_RDONLY;
 113
 114     /* 0666 = -rw-rw-rw- = read+write for everyone, minus umask */
 115     if ((libsmbc->fd = smbc_open(url, access, 0666)) < 0) {
 116         ret = AVERROR(errno);
 117         av_log(h, AV_LOG_ERROR, "File open failed: %s\n", strerror(errno));
 118         goto fail;
 119     }
 120
 121     if (smbc_fstat(libsmbc->fd, &st) < 0)
 122         av_log(h, AV_LOG_WARNING, "Cannot stat file: %s\n", strerror(errno));
 123     else
 124         libsmbc->filesize = st.st_size;
 125
 126     return 0;
 127   fail:
 128     libsmbc_close(h);
 129     return ret;
 130 }
 131
 132 static int64_t libsmbc_seek(URLContext *h, int64_t pos, int whence)
 133 {
 134     LIBSMBContext *libsmbc = h->priv_data;
 135     int64_t newpos;
 136
 137     if (whence == AVSEEK_SIZE) {
 138         if (libsmbc->filesize == -1) {
 139             av_log(h, AV_LOG_ERROR, "Error during seeking: filesize is unknown.\n");
 140             return AVERROR(EIO);
 141         } else
 142             return libsmbc->filesize;
 143     }
 144
 145     if ((newpos = smbc_lseek(libsmbc->fd, pos, whence)) < 0) {
 146         int err = errno;
 147         av_log(h, AV_LOG_ERROR, "Error during seeking: %s\n", strerror(err));
 148         return AVERROR(err);
 149     }
 150
 151     return newpos;
 152 }
 153
 154 static int libsmbc_read(URLContext *h, unsigned char *buf, int size)
 155 {
 156     LIBSMBContext *libsmbc = h->priv_data;
 157     int bytes_read;
 158
 159     if ((bytes_read = smbc_read(libsmbc->fd, buf, size)) < 0) {
 160         av_log(h, AV_LOG_ERROR, "Read error: %s\n", strerror(errno));
 161         return AVERROR(errno);
 162     }
 163
 164     return bytes_read;
 165 }
 166
 167 static int libsmbc_write(URLContext *h, const unsigned char *buf, int size)
 168 {
 169     LIBSMBContext *libsmbc = h->priv_data;
 170     int bytes_written;
 171
 172     if ((bytes_written = smbc_write(libsmbc->fd, buf, size)) < 0) {
 173         av_log(h, AV_LOG_ERROR, "Write error: %s\n", strerror(errno));
 174         return AVERROR(errno);
 175     }
 176
 177     return bytes_written;
 178 }
 179
 180 #define OFFSET(x) offsetof(LIBSMBContext, x)
 181 #define D AV_OPT_FLAG_DECODING_PARAM
 182 #define E AV_OPT_FLAG_ENCODING_PARAM
 183 static const AVOption options[] = {
 184     {"timeout",   "set timeout in ms of socket I/O operations",    OFFSET(timeout), AV_OPT_TYPE_INT, {.i64 = -1}, -1, INT_MAX, D|E },
 185     {"truncate",  "truncate existing files on write",              OFFSET(trunc),   AV_OPT_TYPE_INT, { .i64 = 1 }, 0, 1, E },
 186     {"workgroup", "set the workgroup used for making connections", OFFSET(workgroup), AV_OPT_TYPE_STRING, { 0 }, 0, 0, D|E },
 187     {NULL}
 188 };
 189
 190 static const AVClass libsmbclient_context_class = {
 191     .class_name     = "libsmbc",
 192     .item_name      = av_default_item_name,
 193     .option         = options,
 194     .version        = LIBAVUTIL_VERSION_INT,
 195 };
 196
 197 URLProtocol ff_libsmbclient_protocol = {
 198     .name                = "smb",
 199     .url_open            = libsmbc_open,
 200     .url_read            = libsmbc_read,
 201     .url_write           = libsmbc_write,
 202     .url_seek            = libsmbc_seek,
 203     .url_close           = libsmbc_close,
 204     .priv_data_size      = sizeof(LIBSMBContext),
 205     .priv_data_class     = &libsmbclient_context_class,
 206     .flags               = URL_PROTOCOL_FLAG_NETWORK,
 207 };