discover: Fix boot with initrd error
[petitboot] / lib / url / 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 static const struct pb_scheme_info schemes[] = {
42         {
43                 .scheme = pb_url_file,
44                 .str = "file",
45                 .str_len = sizeof("file") - 1,
46         },
47         {
48                 .scheme = pb_url_ftp,
49                 .str = "ftp",
50                 .str_len = sizeof("ftp") - 1,
51         },
52         {
53                 .scheme = pb_url_http,
54                 .str = "http",
55                 .str_len = sizeof("http") - 1,
56         },
57         {
58                 .scheme = pb_url_https,
59                 .str = "https",
60                 .str_len = sizeof("https") - 1,
61         },
62         {
63                 .scheme = pb_url_nfs,
64                 .str = "nfs",
65                 .str_len = sizeof("nfs") - 1,
66         },
67         {
68                 .scheme = pb_url_sftp,
69                 .str = "sftp",
70                 .str_len = sizeof("sftp") - 1,
71         },
72         {
73                 .scheme = pb_url_tftp,
74                 .str = "tftp",
75                 .str_len = sizeof("tftp") - 1,
76         },
77 };
78
79 static const struct pb_scheme_info *file_scheme = &schemes[0];
80
81 /**
82  * pb_url_find_scheme - Find the pb_scheme_info for a URL string.
83  */
84
85 static const struct pb_scheme_info *pb_url_find_scheme(const char *url)
86 {
87         static const int sep_len = sizeof("://") - 1;
88         static const char *sep = "://";
89         unsigned int i, url_len;
90
91         url_len = strlen(url);
92
93         for (i = 0; i < sizeof(schemes) / sizeof(schemes[0]); i++) {
94                 const struct pb_scheme_info *scheme = &schemes[i];
95
96                 if (url_len < scheme->str_len + sep_len)
97                         continue;
98
99                 if (strncmp(url + scheme->str_len, sep, sep_len))
100                         continue;
101
102                 if (strncasecmp(url, scheme->str, scheme->str_len))
103                         continue;
104
105                 return scheme;
106         }
107
108         /* Assume this is a non-url local file. */
109
110         return file_scheme;
111 }
112
113 /**
114  * pb_url_parse - Parse a remote file URL.
115  * @ctx: The talloc context to associate with the returned string.
116  *
117  * Returns a talloc'ed struct pb_url instance on success, or NULL on error.
118  */
119
120 struct pb_url *pb_url_parse(void *ctx, const char *url_str)
121 {
122         const struct pb_scheme_info *si;
123         struct pb_url *url;
124         const char *p;
125
126         pb_log("%s: '%s'\n", __func__, url_str);
127
128         if (!url_str || !*url_str) {
129                 assert(0 && "bad url");
130                 return NULL;
131         }
132
133         url = talloc_zero(ctx, struct pb_url);
134
135         if (!url)
136                 return NULL;
137
138         si = pb_url_find_scheme(url_str);
139
140         url->scheme = si->scheme;
141         p = url_str + si->str_len + strlen("://");
142
143         url->full = talloc_strdup(url, url_str);
144
145         if (url->scheme == pb_url_file) {
146                 url->port = NULL;
147                 url->host = NULL;
148                 url->path = talloc_strdup(url, p);
149         } else {
150                 int len;
151                 const char *col;
152                 const char *path;
153
154                 path = strchr(p, '/');
155
156                 if (!path) {
157                         pb_log("%s: parse path failed '%s'\n", __func__ , p);
158                         goto fail;
159                 }
160
161                 col = strchr(p, ':');
162
163                 if (col) {
164                         len = path - col - 1;
165                         url->port = len ? talloc_strndup(url, col + 1, len)
166                                 : NULL;
167                         len = col - p;
168                         url->host = len ? talloc_strndup(url, p, len) : NULL;
169                 } else {
170                         url->port = NULL;
171                         url->host = talloc_strndup(url, p, path - p);
172                 }
173
174                 /* remove multiple leading slashes */
175                 for (; *path && *(path+1) == '/'; path++)
176                         ;
177
178                 url->path = talloc_strdup(url, path);
179         }
180
181         p = strrchr(url->path, '/');
182
183         if (p) {
184                 p++;
185                 url->dir = talloc_strndup(url, url->path, p - url->path);
186                 url->file = talloc_strdup(url, p);
187         } else {
188                 url->dir = NULL;
189                 url->file = talloc_strdup(url, url->path);
190         }
191
192         pb_log(" scheme %d\n", url->scheme);
193         pb_log(" host '%s'\n", url->host);
194         pb_log(" port '%s'\n", url->port);
195         pb_log(" path '%s'\n", url->path);
196         pb_log(" dir '%s'\n", url->dir);
197         pb_log(" file '%s'\n", url->file);
198
199         return url;
200
201 fail:
202         talloc_free(url);
203         return NULL;
204 }
205
206 const char *pb_url_scheme_name(enum pb_url_scheme scheme)
207 {
208         unsigned int i;
209
210         for (i = 0; i < sizeof(schemes) / sizeof(schemes[0]); i++)
211                 if (schemes[i].scheme == scheme)
212                         return schemes[i].str;
213
214         return NULL;
215 }