FFmpeg源码：ffurl_seek2、ffurl_seek、avio_size函数分析

一、ffurl_seek2函数

ffurl_seek2函数定义在FFmpeg源码（本文演示用的FFmpeg源码版本为7.0.1）的源文件libavformat/avio.c中：

int64_t ffurl_seek2(void *urlcontext, int64_t pos, int whence)
{URLContext *h = urlcontext;int64_t ret;if (!h->prot->url_seek)return AVERROR(ENOSYS);ret = h->prot->url_seek(h, pos, whence & ~AVSEEK_FORCE);return ret;
}

该函数的作用是：改变形参urlcontext访问的资源的下一个读/写操作将使用的位置。

ffurl_seek2函数内部，h->prot->url_seek是函数指针：

typedef struct URLProtocol {
//...int64_t (*url_seek)( URLContext *h, int64_t pos, int whence);
//...
}

h->prot->url_seek不为空时，会调用h->prot->url_seek指向的回调函数：

ret = h->prot->url_seek(h, pos, whence & ~AVSEEK_FORCE);

而函数指针h->prot->url_seek一般指向libavformat/file.c中的file_seek函数。这时，执行ffurl_seek2函数等价于执行file_seek函数。关于file_seek函数用法可以参考：《FFmpeg源码：file_read、file_write、file_seek函数分析》。

所以当h->prot->url_seek指向file_seek函数时，ffurl_seek2函数的作用是：

ffurl_seek2(h，0,AVSEEK_SIZE)：不改变读写位置，返回文件描述符为h->priv_data->fd的文件的大小；

ffurl_seek2(h,0,SEEK_SET)：将读写位置移到文件描述符为h->priv_data->fd的文件的开头，此时返回0；

ffurl_seek2(h，0,SEEK_END)：将读写位置移到文件尾，此时返回文件描述符为h->priv_data->fd的文件的大小；

ffurl_seek2(h，0,SEEK_CUR)：取得目前文件位置。此时返回当前读写位置距离文件开头多少个字节；

二、ffurl_seek函数

ffurl_seek函数定义在libavformat/url.h中：

/*** Change the position that will be used by the next read/write* operation on the resource accessed by h.** @param pos specifies the new position to set* @param whence specifies how pos should be interpreted, it must be* one of SEEK_SET (seek from the beginning), SEEK_CUR (seek from the* current position), SEEK_END (seek from the end), or AVSEEK_SIZE* (return the filesize of the requested resource, pos is ignored).* @return a negative value corresponding to an AVERROR code in case* of failure, or the resulting file position, measured in bytes from* the beginning of the file. You can use this feature together with* SEEK_CUR to read the current file position.*/
static inline int64_t ffurl_seek(URLContext *h, int64_t pos, int whence)
{return ffurl_seek2(h, pos, whence);
}

可以看到该函数内部仅调用了ffurl_seek2函数，所以ffurl_seek函数和ffurl_seek2函数用法完全等价。不同点仅仅在于ffurl_seek函数用static和inline关键字修饰，内联展开节省了每次调用函数的开销，能提高程序性能，具体可以参考：《C语言static和inline》。

三、avio_size函数

avio_size函数定义在libavformat/aviobuf.c中：

/*** Get the filesize.* @return filesize or AVERROR*/
int64_t avio_size(AVIOContext *s)
{FFIOContext *const ctx = ffiocontext(s);int64_t size;if (!s)return AVERROR(EINVAL);if (ctx->written_output_size)return ctx->written_output_size;if (!s->seek)return AVERROR(ENOSYS);size = s->seek(s->opaque, 0, AVSEEK_SIZE);if (size < 0) {if ((size = s->seek(s->opaque, -1, SEEK_END)) < 0)return size;size++;s->seek(s->opaque, s->pos, SEEK_SET);}return size;
}

该函数的作用是：返回文件描述符为s->opaque->priv_data->fd的文件的大小，单位为字节。该函数内部首先会尝试通过fstat函数获取文件大小，如果获取失败，会继续尝试通过lseek函数获取文件大小。

可以看到avio_size函数内部调用了函数指针s->seek指向的回调函数，而s->seek一般指向ffurl_seek2函数，所以下面语句相当于执行了ffurl_seek2(s->opaque，0,AVSEEK_SIZE)，从而能获取到文件大小：

size = s->seek(s->opaque, 0, AVSEEK_SIZE);

从 《FFmpeg源码：file_read、file_write、file_seek函数分析》中我们可以知道，ffurl_seek2(s->opaque，0,AVSEEK_SIZE)实际相当于执行了：

/* XXX: use llseek */
static int64_t file_seek(URLContext *h, int64_t pos, int whence)
{
//...struct stat st;ret = fstat(c->fd, &st);return ret < 0 ? AVERROR(errno) : (S_ISFIFO(st.st_mode) ? 0 : st.st_size);//...
}

即通过fstat函数得到文件大小。

avio_size函数中执行语句“size = s->seek(s->opaque, 0, AVSEEK_SIZE)”后，如果得到的size值小于0，会继续执行：

    if (size < 0) {if ((size = s->seek(s->opaque, -1, SEEK_END)) < 0)return size;size++;s->seek(s->opaque, s->pos, SEEK_SET);}

意思就是如果通过fstat函数获取文件大小失败，继续尝试通过lseek函数获取文件大小：

/* XXX: use llseek */
static int64_t file_seek(URLContext *h, int64_t pos, int whence)
{
//...ret = lseek(c->fd, pos, whence);return ret < 0 ? AVERROR(errno) : ret;
}