Make URI structures reference-counted.
[people/dverkamp/gpxe.git] / src / core / open.c
1 /*
2  * Copyright (C) 2007 Michael Brown <mbrown@fensystems.co.uk>.
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License as
6  * published by the Free Software Foundation; either version 2 of the
7  * License, or any later version.
8  *
9  * This program is distributed in the hope that it will be useful, but
10  * WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * 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., 675 Mass Ave, Cambridge, MA 02139, USA.
17  */
18
19 #include <stdarg.h>
20 #include <string.h>
21 #include <errno.h>
22 #include <gpxe/xfer.h>
23 #include <gpxe/uri.h>
24 #include <gpxe/socket.h>
25 #include <gpxe/open.h>
26
27 /** @file
28  *
29  * Data transfer interface opening
30  *
31  */
32
33 /** Registered URI openers */
34 static struct uri_opener uri_openers[0]
35         __table_start ( struct uri_opener, uri_openers );
36 static struct uri_opener uri_openers_end[0]
37         __table_end ( struct uri_opener, uri_openers );
38
39 /** Registered socket openers */
40 static struct socket_opener socket_openers[0]
41         __table_start ( struct socket_opener, socket_openers );
42 static struct socket_opener socket_openers_end[0]
43         __table_end ( struct socket_opener, socket_openers );
44
45 /**
46  * Open URI
47  *
48  * @v xfer              Data transfer interface
49  * @v uri_string        URI string (e.g. "http://etherboot.org/kernel")
50  * @ret rc              Return status code
51  */
52 int xfer_open_uri ( struct xfer_interface *xfer, const char *uri_string ) {
53         struct uri *uri;
54         struct uri_opener *opener;
55         int rc = -ENOTSUP;
56
57         DBGC ( xfer, "XFER %p opening URI %s\n", xfer, uri_string );
58
59         uri = parse_uri ( uri_string );
60         if ( ! uri )
61                 return -ENOMEM;
62
63         for ( opener = uri_openers ; opener < uri_openers_end ; opener++ ) {
64                 if ( strcmp ( uri->scheme, opener->scheme ) == 0 ) {
65                         rc = opener->open ( xfer, uri );
66                         goto done;
67                 }
68         }
69
70         DBGC ( xfer, "XFER %p attempted to open unsupported URI scheme "
71                "\"%s\"\n", xfer, uri->scheme );
72  done:
73         uri_put ( uri );
74         return rc;
75 }
76
77 /**
78  * Open socket
79  *
80  * @v xfer              Data transfer interface
81  * @v semantics         Communication semantics (e.g. SOCK_STREAM)
82  * @v peer              Peer socket address
83  * @v local             Local socket address, or NULL
84  * @ret rc              Return status code
85  */
86 int xfer_open_socket ( struct xfer_interface *xfer, int semantics,
87                        struct sockaddr *peer, struct sockaddr *local ) {
88         struct socket_opener *opener;
89
90         DBGC ( xfer, "XFER %p opening (%s,%s) socket\n", xfer,
91                socket_semantics_name ( semantics ),
92                socket_family_name ( peer->sa_family ) );
93
94         for ( opener = socket_openers; opener < socket_openers_end; opener++ ){
95                 if ( ( opener->semantics == semantics ) &&
96                      ( opener->family == peer->sa_family ) ) {
97                         return opener->open ( xfer, peer, local );
98                 }
99         }
100
101         DBGC ( xfer, "XFER %p attempted to open unsupported socket type "
102                "(%s,%s)\n", xfer, socket_semantics_name ( semantics ),
103                socket_family_name ( peer->sa_family ) );
104         return -ENOTSUP;
105 }
106
107 /**
108  * Open location
109  *
110  * @v xfer              Data transfer interface
111  * @v type              Location type
112  * @v args              Remaining arguments depend upon location type
113  * @ret rc              Return status code
114  */
115 int xfer_vopen ( struct xfer_interface *xfer, int type, va_list args ) {
116         switch ( type ) {
117         case LOCATION_URI: {
118                 const char *uri_string = va_arg ( args, const char * );
119
120                 return xfer_open_uri ( xfer, uri_string ); }
121         case LOCATION_SOCKET: {
122                 int semantics = va_arg ( args, int );
123                 struct sockaddr *peer = va_arg ( args, struct sockaddr * );
124                 struct sockaddr *local = va_arg ( args, struct sockaddr * );
125
126                 return xfer_open_socket ( xfer, semantics, peer, local ); }
127         default:
128                 DBGC ( xfer, "XFER %p attempted to open unsupported location "
129                        "type %d\n", xfer, type );
130                 return -ENOTSUP;
131         }
132 }
133
134 /**
135  * Open location
136  *
137  * @v xfer              Data transfer interface
138  * @v type              Location type
139  * @v ...               Remaining arguments depend upon location type
140  * @ret rc              Return status code
141  */
142 int xfer_open ( struct xfer_interface *xfer, int type, ... ) {
143         va_list args;
144         int rc;
145
146         va_start ( args, type );
147         rc = xfer_vopen ( xfer, type, args );
148         va_end ( args );
149         return rc;
150 }