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 #if 0
  26         "http://www.etherboot.org/wiki",
  27         "mailto:bob@nowhere.com",
  28         "ftp://joe:secret@insecure.org:8081/hidden/path/to?what=is#this",
  29 #endif
  30 };
  31
  32 static int test_parse_unparse ( const char *uri_string ) {
  33         char buf[URI_MAX_LEN];
  34         size_t len;
  35         struct uri *uri = NULL;
  36         int rc;
  37
  38         /* Parse and unparse URI */
  39         uri = parse_uri ( uri_string );
  40         if ( ! uri ) {
  41                 rc = -ENOMEM;
  42                 goto done;
  43         }
  44         len = unparse_uri ( buf, sizeof ( buf ), uri );
  45
  46         /* Compare result */
  47         if ( strcmp ( buf, uri_string ) != 0 ) {
  48                 printf ( "Unparse of \"%s\" produced \"%s\"\n",
  49                          uri_string, buf );
  50                 rc = -EINVAL;
  51                 goto done;
  52         }
  53
  54         rc = 0;
  55
  56  done:
  57         uri_put ( uri );
  58         if ( rc ) {
  59                 printf ( "URI parse-unparse of \"%s\" failed: %s\n",
  60                          uri_string, strerror ( rc ) );
  61         }
  62         return rc;
  63 }
  64
  65 static int test_resolve ( const char *base_uri_string,
  66                           const char *relative_uri_string,
  67                           const char *resolved_uri_string ) {
  68         struct uri *base_uri = NULL;
  69         struct uri *relative_uri = NULL;
  70         struct uri *resolved_uri = NULL;
  71         char buf[URI_MAX_LEN];
  72         size_t len;
  73         int rc;
  74
  75         /* Parse URIs */
  76         base_uri = parse_uri ( base_uri_string );
  77         if ( ! base_uri ) {
  78                 rc = -ENOMEM;
  79                 goto done;
  80         }
  81         relative_uri = parse_uri ( relative_uri_string );
  82         if ( ! relative_uri ) {
  83                 rc = -ENOMEM;
  84                 goto done;
  85         }
  86
  87         /* Resolve URI */
  88         resolved_uri = resolve_uri ( base_uri, relative_uri );
  89         if ( ! resolved_uri ) {
  90                 rc = -ENOMEM;
  91                 goto done;
  92         }
  93
  94         /* Compare result */
  95         len = unparse_uri ( buf, sizeof ( buf ), resolved_uri );
  96         if ( strcmp ( buf, resolved_uri_string ) != 0 ) {
  97                 printf ( "Resolution of \"%s\"+\"%s\" produced \"%s\"\n",
  98                          base_uri_string, relative_uri_string, buf );
  99                 rc = -EINVAL;
 100                 goto done;
 101         }
 102
 103         rc = 0;
 104
 105  done:
 106         uri_put ( base_uri );
 107         uri_put ( relative_uri );
 108         uri_put ( resolved_uri );
 109         if ( rc ) {
 110                 printf ( "URI resolution of \"%s\"+\"%s\" failed: %s\n",
 111                          base_uri_string, relative_uri_string,
 112                          strerror ( rc ) );
 113         }
 114         return rc;
 115 }
 116
 117 int uri_test ( void ) {
 118         unsigned int i;
 119         struct uri_test *uri_test;
 120         int rc;
 121         int overall_rc = 0;
 122
 123         for ( i = 0 ; i < ( sizeof ( uri_tests ) /
 124                             sizeof ( uri_tests[0] ) ) ; i++ ) {
 125                 uri_test = &uri_tests[i];
 126                 rc = test_parse_unparse ( uri_test->base_uri_string );
 127                 if ( rc != 0 )
 128                         overall_rc = rc;
 129                 rc = test_parse_unparse ( uri_test->relative_uri_string );
 130                 if ( rc != 0 )
 131                         overall_rc = rc;
 132                 rc = test_parse_unparse ( uri_test->resolved_uri_string );
 133                 if ( rc != 0 )
 134                         overall_rc = rc;
 135                 rc = test_resolve ( uri_test->base_uri_string,
 136                                     uri_test->relative_uri_string,
 137                                     uri_test->resolved_uri_string );
 138                 if ( rc != 0 )
 139                         overall_rc = rc;
 140         }
 141
 142         if ( overall_rc )
 143                 printf ( "URI tests failed: %s\n", strerror ( overall_rc ) );
 144         return overall_rc;
 145 }