diff mbox series

[FFmpeg-devel,2/3] avformat: add ff_match_url_ext()

Message ID 20230515000547.1703-2-michael@niedermayer.cc
State New
Headers show
Series [FFmpeg-devel,1/3] avformat/hls: reduce default max reload to 3 | expand

Checks

Context Check Description
andriy/make_x86 success Make finished
andriy/make_fate_x86 success Make fate finished

Commit Message

Michael Niedermayer May 15, 2023, 12:05 a.m. UTC
Match url against a list of extensions similar to av_match_ext()

Signed-off-by: Michael Niedermayer <michael@niedermayer.cc>
---
 libavformat/format.c   | 25 +++++++++++++++++++++++++
 libavformat/internal.h |  9 +++++++++
 2 files changed, 34 insertions(+)

Comments

Leo Izen May 15, 2023, 3:10 a.m. UTC | #1
On 5/14/23 20:05, Michael Niedermayer wrote:
> Match url against a list of extensions similar to av_match_ext()
> 
> Signed-off-by: Michael Niedermayer <michael@niedermayer.cc>
> ---
>   libavformat/format.c   | 25 +++++++++++++++++++++++++
>   libavformat/internal.h |  9 +++++++++
>   2 files changed, 34 insertions(+)
> 
> diff --git a/libavformat/format.c b/libavformat/format.c
> index 52b814e67d..04fdbacec0 100644
> --- a/libavformat/format.c
> +++ b/libavformat/format.c
> @@ -50,6 +50,31 @@ int av_match_ext(const char *filename, const char *extensions)
>       return 0;
>   }
>   
> +int ff_match_url_ext(const char *url, const char *extensions)
> +{
> +    const char *ext;
> +    URLComponents uc;
> +    int ret;
> +    char scratchpad[128];
> +
> +    if (!url)
> +        return 0;
> +
> +    ret = ff_url_decompose(&uc, url, NULL);
> +    if (ret < 0 || !URL_COMPONENT_HAVE(uc, scheme))
> +        return ret;
> +    for (ext = uc.query; *ext != '.' && ext > uc.path; ext--)
> +        ; > +
> +    if (*ext != '.')
> +        return 0;
> +    if (uc.query - ext > sizeof(scratchpad))
> +        return AVERROR(ENOMEM); //not enough memory in our scratchpad
> +    av_strlcpy(scratchpad, ext + 1, FFMIN(sizeof(scratchpad), uc.query - ext));

This FFMIN is superfluous as sizeof(scratchpad) will always be greater 
than or equal to uc.query - ext after the check above it.

> +
> +    return av_match_name(scratchpad, extensions);
> +}
> +
>   const AVOutputFormat *av_guess_format(const char *short_name, const char *filename,
>                                         const char *mime_type)
>   {
> diff --git a/libavformat/internal.h b/libavformat/internal.h
> index f575064e8f..43466fbe6d 100644
> --- a/libavformat/internal.h
> +++ b/libavformat/internal.h
> @@ -705,6 +705,15 @@ int ff_unlock_avformat(void);
>    */
>   void ff_format_set_url(AVFormatContext *s, char *url);
>   
> +/**
> + * Return a positive value if the given url has one of the given
> + * extensions, 0 otherwise.
> + *
> + * @param filename   file name to check against the given extensions

Copy/paste error, this should be @param url.

> + * @param extensions a comma-separated list of filename extensions
> + */
> +int ff_match_url_ext(const char *url, const char *extensions);
> +
>   struct FFOutputFormat;
>   void avpriv_register_devices(const struct FFOutputFormat * const o[], const AVInputFormat * const i[]);
>
Michael Niedermayer May 15, 2023, 7:12 p.m. UTC | #2
On Sun, May 14, 2023 at 11:10:00PM -0400, Leo Izen wrote:
> On 5/14/23 20:05, Michael Niedermayer wrote:
> > Match url against a list of extensions similar to av_match_ext()
> > 
> > Signed-off-by: Michael Niedermayer <michael@niedermayer.cc>
> > ---
> >   libavformat/format.c   | 25 +++++++++++++++++++++++++
> >   libavformat/internal.h |  9 +++++++++
> >   2 files changed, 34 insertions(+)
> > 
> > diff --git a/libavformat/format.c b/libavformat/format.c
> > index 52b814e67d..04fdbacec0 100644
> > --- a/libavformat/format.c
> > +++ b/libavformat/format.c
> > @@ -50,6 +50,31 @@ int av_match_ext(const char *filename, const char *extensions)
> >       return 0;
> >   }
> > +int ff_match_url_ext(const char *url, const char *extensions)
> > +{
> > +    const char *ext;
> > +    URLComponents uc;
> > +    int ret;
> > +    char scratchpad[128];
> > +
> > +    if (!url)
> > +        return 0;
> > +
> > +    ret = ff_url_decompose(&uc, url, NULL);
> > +    if (ret < 0 || !URL_COMPONENT_HAVE(uc, scheme))
> > +        return ret;
> > +    for (ext = uc.query; *ext != '.' && ext > uc.path; ext--)
> > +        ; > +
> > +    if (*ext != '.')
> > +        return 0;
> > +    if (uc.query - ext > sizeof(scratchpad))
> > +        return AVERROR(ENOMEM); //not enough memory in our scratchpad
> > +    av_strlcpy(scratchpad, ext + 1, FFMIN(sizeof(scratchpad), uc.query - ext));
> 
> This FFMIN is superfluous as sizeof(scratchpad) will always be greater than
> or equal to uc.query - ext after the check above it.

yes, ill remove it, if noone else did it before me


> 
> > +
> > +    return av_match_name(scratchpad, extensions);
> > +}
> > +
> >   const AVOutputFormat *av_guess_format(const char *short_name, const char *filename,
> >                                         const char *mime_type)
> >   {
> > diff --git a/libavformat/internal.h b/libavformat/internal.h
> > index f575064e8f..43466fbe6d 100644
> > --- a/libavformat/internal.h
> > +++ b/libavformat/internal.h
> > @@ -705,6 +705,15 @@ int ff_unlock_avformat(void);
> >    */
> >   void ff_format_set_url(AVFormatContext *s, char *url);
> > +/**
> > + * Return a positive value if the given url has one of the given
> > + * extensions, 0 otherwise.
> > + *
> > + * @param filename   file name to check against the given extensions
> 
> Copy/paste error, this should be @param url.

too late, ive already pushed the patch with this fixed

thx

[...]
diff mbox series

Patch

diff --git a/libavformat/format.c b/libavformat/format.c
index 52b814e67d..04fdbacec0 100644
--- a/libavformat/format.c
+++ b/libavformat/format.c
@@ -50,6 +50,31 @@  int av_match_ext(const char *filename, const char *extensions)
     return 0;
 }
 
+int ff_match_url_ext(const char *url, const char *extensions)
+{
+    const char *ext;
+    URLComponents uc;
+    int ret;
+    char scratchpad[128];
+
+    if (!url)
+        return 0;
+
+    ret = ff_url_decompose(&uc, url, NULL);
+    if (ret < 0 || !URL_COMPONENT_HAVE(uc, scheme))
+        return ret;
+    for (ext = uc.query; *ext != '.' && ext > uc.path; ext--)
+        ;
+
+    if (*ext != '.')
+        return 0;
+    if (uc.query - ext > sizeof(scratchpad))
+        return AVERROR(ENOMEM); //not enough memory in our scratchpad
+    av_strlcpy(scratchpad, ext + 1, FFMIN(sizeof(scratchpad), uc.query - ext));
+
+    return av_match_name(scratchpad, extensions);
+}
+
 const AVOutputFormat *av_guess_format(const char *short_name, const char *filename,
                                       const char *mime_type)
 {
diff --git a/libavformat/internal.h b/libavformat/internal.h
index f575064e8f..43466fbe6d 100644
--- a/libavformat/internal.h
+++ b/libavformat/internal.h
@@ -705,6 +705,15 @@  int ff_unlock_avformat(void);
  */
 void ff_format_set_url(AVFormatContext *s, char *url);
 
+/**
+ * Return a positive value if the given url has one of the given
+ * extensions, 0 otherwise.
+ *
+ * @param filename   file name to check against the given extensions
+ * @param extensions a comma-separated list of filename extensions
+ */
+int ff_match_url_ext(const char *url, const char *extensions);
+
 struct FFOutputFormat;
 void avpriv_register_devices(const struct FFOutputFormat * const o[], const AVInputFormat * const i[]);