Add URL parsing routines
[petitboot] / ui / common / url.c
1 /*
2  *  Copyright (C) 2009 Sony Computer Entertainment Inc.
3  *  Copyright 2009 Sony Corp.
4  *
5  *  This program is free software; you can redistribute it and/or modify
6  *  it under the terms of the GNU General Public License as published by
7  *  the Free Software Foundation; version 2 of the License.
8  *
9  *  This program is distributed in the hope that it will be useful,
10  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
11  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  *  GNU General Public License for more details.
13  *
14  *  You should have received a copy of the GNU General Public License
15  *  along with this program; if not, write to the Free Software
16  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
17  */
18
19 #if defined(HAVE_CONFIG_H)
20 #include "config.h"
21 #endif
22
23 #define _GNU_SOURCE
24 #include <assert.h>
25 #include <string.h>
26
27 #include "log/log.h"
28 #include "talloc/talloc.h"
29 #include "url.h"
30
31 /**
32  * pb_scheme_info - Helper for parsing URLs.
33  */
34
35 struct pb_scheme_info {
36         enum pb_url_scheme scheme;
37         const char *str;
38         unsigned int str_len;
39 };
40
41 /**
42  * pb_url_find_scheme - Find the pb_scheme_info for a URL string.
43  */
44
45 static const struct pb_scheme_info *pb_url_find_scheme(const char *url_str)
46 {
47         static const struct pb_scheme_info a[] = {
48         {
49                 .scheme = pb_url_file,
50                 .str = "file://",
51                 .str_len = sizeof("file://") - 1,
52         },
53         {
54                 .scheme = pb_url_ftp,
55                 .str = "ftp://",
56                 .str_len = sizeof("ftp://") - 1,
57         },
58         {
59                 .scheme = pb_url_http,
60                 .str = "http://",
61                 .str_len = sizeof("http://") - 1,
62         },
63         {
64                 .scheme = pb_url_https,
65                 .str = "https://",
66                 .str_len = sizeof("https://") - 1,
67         },
68         {
69                 .scheme = pb_url_nfs,
70                 .str = "nfs://",
71                 .str_len = sizeof("nfs://") - 1,
72         },
73         {
74                 .scheme = pb_url_sftp,
75                 .str = "sftp://",
76                 .str_len = sizeof("sftp://") - 1,
77         },
78         {
79                 .scheme = pb_url_tftp,
80                 .str = "tftp://",
81                 .str_len = sizeof("tftp://") - 1,
82         },
83         };
84         static const struct pb_scheme_info file_scheme = {
85                 .str = "",
86                 .scheme = pb_url_file,
87         };
88         unsigned int i;
89
90         for (i = 0; i < sizeof(a) / sizeof(a[0]); i++)
91                 if (!strncasecmp(url_str, a[i].str, a[i].str_len))
92                         return &a[i];
93
94         /* Assume this is a non-url local file. */
95
96         return &file_scheme;
97 }
98
99 /**
100  * pb_url_parse - Parse a remote file URL.
101  * @ctx: The talloc context to associate with the returned string.
102  *
103  * Returns a talloc'ed struct pb_url instance on success, or NULL on error.
104  */
105
106 struct pb_url *pb_url_parse(void *ctx, const char *url_str)
107 {
108         const struct pb_scheme_info *si;
109         struct pb_url *url;
110         const char *p;
111
112         pb_log("%s: '%s'\n", __func__, url_str);
113
114         if (!url_str || !*url_str) {
115                 assert(0 && "bad url");
116                 return NULL;
117         }
118
119         url = talloc_zero(ctx, struct pb_url);
120
121         if (!url)
122                 return NULL;
123
124         si = pb_url_find_scheme(url_str);
125
126         url->scheme = si->scheme;
127         p = url_str + si->str_len;
128
129         url->full = talloc_strdup(url, url_str);
130
131         if (url->scheme == pb_url_file) {
132                 url->port = NULL;
133                 url->host = NULL;
134                 url->path = talloc_strdup(url, p);
135         } else {
136                 int len;
137                 const char *col;
138                 const char *path;
139
140                 path = strchr(p, '/');
141
142                 if (!path) {
143                         pb_log("%s: parse path failed '%s'\n", p);
144                         goto fail;
145                 }
146
147                 col = strchr(p, ':');
148
149                 if (col) {
150                         len = path - col - 1;
151                         url->port = len ? talloc_strndup(url, col + 1, len)
152                                 : NULL;
153                         len = col - p;
154                         url->host = len ? talloc_strndup(url, p, len) : NULL;
155                 } else {
156                         url->port = NULL;
157                         url->host = talloc_strndup(url, p, path - p);
158                 }
159                 url->path = talloc_strdup(url, path);
160         }
161
162         p = strrchr(url->path, '/');
163
164         if (p) {
165                 p++;
166                 url->dir = talloc_strndup(url, url->path, p - url->path);
167                 url->file = talloc_strdup(url, p);
168         } else {
169                 url->dir = NULL;
170                 url->file = talloc_strdup(url, url->path);
171         }
172
173         pb_log(" scheme %d\n", url->scheme);
174         pb_log(" host '%s'\n", url->host);
175         pb_log(" port '%s'\n", url->port);
176         pb_log(" path '%s'\n", url->path);
177         pb_log(" dir '%s'\n", url->dir);
178         pb_log(" file '%s'\n", url->file);
179
180         return url;
181
182 fail:
183         talloc_free(url);
184         return NULL;
185 }