src/tests/uri_test.c

   1 #include <stdint.h>
   2 #include <stddef.h>
   3 #include <stdio.h>
   4 #include <string.h>
   5 #include <errno.h>
   6 #include <gpxe/uri.h>
   7
   8 #define URI_MAX_LEN 1024
   9
  10 struct uri_test {
  11         const char *base_uri_string;
  12         const char *relative_uri_string;
  13         const char *resolved_uri_string;
  14 };
  15
  16 static struct uri_test uri_tests[] = {
  17         { "http://www.fensystems.co.uk", "",
  18           "http://www.fensystems.co.uk/" },
  19         { "http://etherboot.org/wiki/page1", "page2",
  20           "http://etherboot.org/wiki/page2" },
  21         { "http://etherboot.org/wiki/page1", "../page3",
  22           "http://etherboot.org/page3" },
  23         { "tftp://192.168.0.1/", "/tftpboot/vmlinuz",
  24           "tftp://192.168.0.1/tftpboot/vmlinuz" },
  25         { "ftp://the%41nswer%3d:%34ty%32wo@ether%62oot.org:8080/p%41th/foo",
  26           "to?%41=b#%43d",
  27           "ftp://theAnswer%3d:4ty2wo@etherboot.org:8080/path/to?a=b#cd" },
  28 #if 0
  29         "http://www.etherboot.org/wiki",
  30         "mailto:bob@nowhere.com",
  31         "ftp://joe:secret@insecure.org:8081/hidden/path/to?what=is#this",
  32 #endif
  33 };
  34
  35 static int test_parse_unparse ( const char *uri_string ) {
  36         char buf[URI_MAX_LEN];
  37         size_t len;
  38         struct uri *uri = NULL;
  39         int rc;
  40
  41         /* Parse and unparse URI */
  42         uri = parse_uri ( uri_string );
  43         if ( ! uri ) {
  44                 rc = -ENOMEM;
  45                 goto done;
  46         }
  47         len = unparse_uri ( buf, sizeof ( buf ), uri, URI_ALL );
  48
  49         /* Compare result */
  50         if ( strcmp ( buf, uri_string ) != 0 ) {
  51                 printf ( "Unparse of \"%s\" produced \"%s\"\n",
  52                          uri_string, buf );
  53                 rc = -EINVAL;
  54                 goto done;
  55         }
  56
  57         rc = 0;
  58
  59  done:
  60         uri_put ( uri );
  61         if ( rc ) {
  62                 printf ( "URI parse-unparse of \"%s\" failed: %s\n",
  63                          uri_string, strerror ( rc ) );
  64         }
  65         return rc;
  66 }
  67
  68 static int test_resolve ( const char *base_uri_string,
  69                           const char *relative_uri_string,
  70                           const char *resolved_uri_string ) {
  71         struct uri *base_uri = NULL;
  72         struct uri *relative_uri = NULL;
  73         struct uri *resolved_uri = NULL;
  74         char buf[URI_MAX_LEN];
  75         size_t len;
  76         int rc;
  77
  78         /* Parse URIs */
  79         base_uri = parse_uri ( base_uri_string );
  80         if ( ! base_uri ) {
  81                 rc = -ENOMEM;
  82                 goto done;
  83         }
  84         relative_uri = parse_uri ( relative_uri_string );
  85         if ( ! relative_uri ) {
  86                 rc = -ENOMEM;
  87                 goto done;
  88         }
  89
  90         /* Resolve URI */
  91         resolved_uri = resolve_uri ( base_uri, relative_uri );
  92         if ( ! resolved_uri ) {
  93                 rc = -ENOMEM;
  94                 goto done;
  95         }
  96
  97         /* Compare result */
  98         len = unparse_uri ( buf, sizeof ( buf ), resolved_uri, URI_ALL );
  99         if ( strcmp ( buf, resolved_uri_string ) != 0 ) {
 100                 printf ( "Resolution of \"%s\"+\"%s\" produced \"%s\"\n",
 101                          base_uri_string, relative_uri_string, buf );
 102                 rc = -EINVAL;
 103                 goto done;
 104         }
 105
 106         rc = 0;
 107
 108  done:
 109         uri_put ( base_uri );
 110         uri_put ( relative_uri );
 111         uri_put ( resolved_uri );
 112         if ( rc ) {
 113                 printf ( "URI resolution of \"%s\"+\"%s\" failed: %s\n",
 114                          base_uri_string, relative_uri_string,
 115                          strerror ( rc ) );
 116         }
 117         return rc;
 118 }
 119
 120 int uri_test ( void ) {
 121         unsigned int i;
 122         struct uri_test *uri_test;
 123         int rc;
 124         int overall_rc = 0;
 125
 126         for ( i = 0 ; i < ( sizeof ( uri_tests ) /
 127                             sizeof ( uri_tests[0] ) ) ; i++ ) {
 128                 uri_test = &uri_tests[i];
 129                 rc = test_parse_unparse ( uri_test->base_uri_string );
 130                 if ( rc != 0 )
 131                         overall_rc = rc;
 132                 rc = test_parse_unparse ( uri_test->relative_uri_string );
 133                 if ( rc != 0 )
 134                         overall_rc = rc;
 135                 rc = test_parse_unparse ( uri_test->resolved_uri_string );
 136                 if ( rc != 0 )
 137                         overall_rc = rc;
 138                 rc = test_resolve ( uri_test->base_uri_string,
 139                                     uri_test->relative_uri_string,
 140                                     uri_test->resolved_uri_string );
 141                 if ( rc != 0 )
 142                         overall_rc = rc;
 143         }
 144
 145         if ( overall_rc )
 146                 printf ( "URI tests failed: %s\n", strerror ( overall_rc ) );
 147         return overall_rc;
 148 }