00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038
00039 module mango.io.Uri;
00040
00041 private import std.ctype;
00042
00043 private import mango.utils.Text,
00044 mango.utils.HeapSlice;
00045
00046 private import mango.io.Buffer,
00047 mango.io.Exception;
00048
00049 private import mango.io.model.IWriter;
00050
00051
00052
00053
00054
00055 extern (C) char* memchr (char *, char, uint);
00056
00057
00058
00059
00060
00061
00062
00063
00064
00065
00066
00067
00068
00069
00070
00071
00072
00073
00074
00075
00076 class Uri : IWritable
00077 {
00078
00079
00080
00081
00082
00083
00084 static this ()
00085 {
00086 error = new IOException ("Invalid URI specification");
00087
00088
00089
00090
00091 foreach (SchemePort sp; schemePorts)
00092 genericSchemes[sp.name] = sp.port;
00093 genericSchemes.rehash;
00094
00095
00096 for (int i='a'; i <= 'z'; ++i)
00097 map[i] = IncGeneric;
00098
00099 for (int i='A'; i <= 'Z'; ++i)
00100 map[i] = IncGeneric;
00101
00102 for (int i='0'; i<='9'; ++i)
00103 map[i] = IncGeneric;
00104
00105
00106 map[':'] |= ExcScheme;
00107 map['/'] |= ExcScheme | ExcAuthority;
00108 map['?'] |= ExcScheme | ExcAuthority | ExcPath;
00109 map['#'] |= ExcScheme | ExcAuthority | ExcPath | ExcQuery;
00110
00111
00112 map['-'] |= IncUser | IncQuery;
00113 map['_'] |= IncUser | IncQuery;
00114 map['.'] |= IncUser | IncQuery;
00115 map['!'] |= IncUser | IncQuery;
00116 map['~'] |= IncUser | IncQuery;
00117 map['*'] |= IncUser | IncQuery;
00118 map['\''] |= IncUser | IncQuery;
00119 map['('] |= IncUser | IncQuery;
00120 map[')'] |= IncUser | IncQuery;
00121
00122
00123 map['+'] |= IncScheme;
00124 map['-'] |= IncScheme;
00125 map['.'] |= IncScheme;
00126
00127
00128 map[';'] |= IncUser;
00129 map[':'] |= IncUser;
00130 map['&'] |= IncUser;
00131 map['='] |= IncUser;
00132 map['+'] |= IncUser;
00133 map['$'] |= IncUser;
00134 map[','] |= IncUser;
00135
00136
00137 map['/'] |= IncPath;
00138 map[';'] |= IncPath;
00139 map[':'] |= IncPath;
00140 map['@'] |= IncPath;
00141 map['&'] |= IncPath;
00142 map['='] |= IncPath;
00143 map['+'] |= IncPath;
00144 map['$'] |= IncPath;
00145 map[','] |= IncPath;
00146
00147
00148 map[';'] |= IncQuery;
00149 map['/'] |= IncQuery;
00150 map['?'] |= IncQuery;
00151 map[':'] |= IncQuery;
00152 map['@'] |= IncQuery;
00153 map['&'] |= IncQuery;
00154 map['='] |= IncQuery;
00155 map['+'] |= IncQuery;
00156 map['$'] |= IncQuery;
00157 map[','] |= IncQuery;
00158 }
00159
00160
00161
00162
00163
00164
00165
00166 this (char[] uri)
00167 {
00168 this();
00169 parse (uri);
00170 }
00171
00172
00173
00174
00175
00176
00177
00178
00179
00180 final static int getDefaultPort (char[] scheme)
00181 {
00182 int port = genericSchemes [scheme];
00183 if (! port)
00184 port = InvalidPort;
00185 return port;
00186 }
00187
00188
00189
00190
00191
00192
00193
00194
00195 char[] getScheme()
00196 {
00197 return scheme;
00198 }
00199
00200
00201
00202
00203
00204
00205
00206
00207 char[] getHost()
00208 {
00209 return host;
00210 }
00211
00212
00213
00214
00215
00216
00217
00218
00219 int getPort()
00220 {
00221 return port;
00222 }
00223
00224
00225
00226
00227
00228
00229
00230
00231 int getValidPort()
00232 {
00233 if (port == InvalidPort)
00234 return getDefaultPort (scheme);
00235 return port;
00236 }
00237
00238
00239
00240
00241
00242
00243
00244
00245 char[] getUserInfo()
00246 {
00247 return userinfo;
00248 }
00249
00250
00251
00252
00253
00254
00255
00256
00257 char[] getPath()
00258 {
00259 return path;
00260 }
00261
00262
00263
00264
00265
00266
00267
00268
00269 char[] getQuery()
00270 {
00271 return query;
00272 }
00273
00274
00275
00276
00277
00278
00279
00280
00281 char[] getFragment()
00282 {
00283 return fragment;
00284 }
00285
00286
00287
00288
00289
00290
00291
00292 bool isGeneric ()
00293 {
00294 return genericSchemes [scheme] != 0;
00295 }
00296
00297
00298
00299
00300
00301
00302
00303
00304 IBuffer write (IBuffer buf)
00305 {
00306 if (scheme.length)
00307 buf.append (scheme).append(":");
00308
00309
00310 if (userinfo.length || host.length || port != InvalidPort)
00311 {
00312 buf.append ("//");
00313
00314 if (userinfo.length)
00315 encode (buf, userinfo, IncUser).append("@");
if (host.length)
buf.append (host);
if (port != InvalidPort && port != getDefaultPort(scheme))
{
char[5] tmp;
buf.append(":").append(Text.itoa (tmp, port));
00316 }
00317 }
00318
00319 if (path.length)
00320 encode (buf, path, IncPath);
00321
00322 if (query.length)
00323 {
00324 buf.append ("?");
00325 encode (buf, query, IncQuery);
00326 }
00327
00328 if (fragment.length)
00329 {
00330 buf.append ("#");
00331 encode (buf, fragment, IncQuery);
00332 }
00333
00334 return buf;
00335 }
00336
00337
00338
00339
00340
00341
00342
00343
00344 void write (IWriter writer)
00345 {
00346 write (writer.getBuffer);
00347 }
00348
00349
00350
00351
00352
00353
00354
00355 override char[] toString ()
00356 {
00357 return write (new GrowableBuffer(1024)).toString;
00358 }
00359
00360
00361
00362
00363
00364
00365
00366
00367
00368 char[] decode (char[] s)
00369 {
00370 int length = s.length;
00371
00372
00373 if (length && memchr (s, '%', length))
00374 {
00375 char* p;
00376 int j;
00377
00378
00379 p = cast(char*) decoded.expand (length);
00380
00381
00382 for (int i; i < length; ++i, ++j, ++p)
00383 {
00384 char c = s[i];
00385 if (c == '%' && (i+2) < length)
00386 {
00387 c = Text.atoi (s[i+1..i+3], 16);
00388 i += 2;
00389 }
00390 *p = c;
00391 }
00392
00393 return cast(char[]) decoded.slice (j);
00394 }
00395
00396
00397 return s;
00398 }
00399
00400
00401
00402
00403
00404
00405
00406
00407 private static IBuffer encode (IBuffer buf, char[] s, int flags)
00408 {
00409 char[3] hex;
00410 int mark;
00411
00412 hex[0] = '%';
00413 foreach (int i, char c; s)
00414 {
00415 if (! (map[c] & flags))
00416 {
00417 buf.append (s[mark..i]);
00418 mark = i+1;
00419
00420 hex[1] = hexDigits [(c >> 4) & 0x0f];
00421 hex[2] = hexDigits [c & 0x0f];
00422 buf.append (hex);
00423 }
00424 }
00425
00426
00427 if (mark < s.length)
00428 buf.append (s[mark..s.length]);
00429
00430 return buf;
00431 }
00432
00433
00434
00435
00436
00437
00438
00439 private this ()
00440 {
00441 port = InvalidPort;
00442 decoded = new HeapSlice (256);
00443 }
00444
00445
00446
00447
00448
00449
00450
00451
00452
00453
00454
00455
00456
00457
00458
00459
00460
00461
00462
00463
00464
00465 private void parse (char[] uri)
00466 {
00467 char c;
00468 int i,
00469 mark,
00470 len = uri.length;
00471
00472
00473 for (i=0; i < len && !(map[c = uri[i]] & ExcScheme); ++i) {}
00474 if (c == ':')
00475 {
00476 scheme = uri[mark..i];
00477 Text.tolower (scheme);
00478 mark = i + 1;
00479 }
00480
00481
00482 if (mark < len-1 && uri[mark] == '/' && uri[mark+1] == '/')
00483 {
00484 for (mark+=2, i=mark; i < len && !(map[uri[i]] & ExcAuthority); ++i) {}
00485 parseAuthority (uri[mark..i]);
00486 mark = i;
00487 }
00488
00489
00490 for (i=mark; i < len && !(map[uri[i]] & ExcPath); ++i) {}
00491 path = decode (uri[mark..i]);
00492 mark = i;
00493
00494
00495 if (mark < len && uri[mark] == '?')
00496 {
00497 for (++mark, i=mark; i < len && uri[i] != '#'; ++i) {}
00498 query = decode (uri[mark..i]);
00499 mark = i;
00500 }
00501
00502
00503 if (mark < len && uri[mark] == '#')
00504 fragment = decode (uri[mark+1..len]);
00505 }
00506
00507
00508
00509
00510
00511
00512
00513
00514
00515
00516
00517
00518
00519
00520
00521
00522
00523 private void parseAuthority (char[] auth)
00524 {
00525 int mark,
00526 len = auth.length;
00527
00528
00529 foreach (int i, char c; auth)
00530 if (c == '@')
00531 {
00532 userinfo = decode (auth[0..i]);
00533 mark = i + 1;
00534 break;
00535 }
00536
00537
00538 for (int i=mark; i < len; ++i)
00539 if (auth [i] == ':')
00540 {
00541 port = Text.atoi (auth [i+1..len]);
00542 len = i;
00543 break;
00544 }
00545
00546
00547 host = auth [mark..len];
00548 }
00549
00550
00551
00552
00553
00554
00555
00556
00557 const int InvalidPort = -1;
00558
00559 private int port;
00560 private char[] host,
00561 path,
00562 query,
00563 scheme,
00564 userinfo,
00565 fragment;
00566 private HeapSlice decoded;
00567
00568 private static ubyte map[256];
00569
00570
00571 private static short[char[]] genericSchemes;
00572
00573 private static const char[] hexDigits = "0123456789abcdef";
00574
00575 private static IOException error;
00576
00577 private enum {
00578 ExcScheme = 0x01,
00579 ExcAuthority = 0x02,
00580 ExcPath = 0x04,
00581 ExcQuery = 0x08,
00582 IncUser = 0x10,
00583 IncPath = 0x20,
00584 IncQuery = 0x40,
00585 IncScheme = 0x80,
00586 IncGeneric = IncScheme | IncUser | IncPath | IncQuery
00587 };
00588
00589 private struct SchemePort
00590 {
00591 char[] name;
00592 int port;
00593 }
00594
00595 private static const SchemePort[] schemePorts =
00596 [
00597 {"coffee", 80},
00598 {"file", InvalidPort},
00599 {"ftp", 21},
00600 {"gopher", 70},
00601 {"hnews", 80},
00602 {"http", 80},
00603 {"http-ng", 80},
00604 {"https", 443},
00605 {"imap", 143},
00606 {"irc", 194},
00607 {"ldap", 389},
00608 {"news", 119},
00609 {"nfs", 2049},
00610 {"nntp", 119},
00611 {"pop", 110},
00612 {"prospero", 1525},
00613 {"rwhois", 4321},
00614 {"sip", InvalidPort},
00615 {"sips", InvalidPort},
00616 {"sipt", InvalidPort},
00617 {"sipu", InvalidPort},
00618 {"shttp", 80},
00619 {"smtp", 25},
00620 {"snews", 563},
00621 {"telnet", 23},
00622 {"vemmi", 575},
00623 {"videotex", 516},
00624 {"wais", 210},
00625 {"whois", 43},
00626 {"whois++", 43},
00627 ];
00628 }
00629
00630
00631
00632
00633
00634
00635
00636
00637 class MutableUri : Uri
00638 {
00639
00640
00641
00642
00643
00644
00645 this ()
00646 {
00647 super();
00648 }
00649
00650
00651
00652
00653
00654
00655
00656 this (char[] uri)
00657 {
00658 super (uri);
00659 }
00660
00661
00662
00663
00664
00665
00666
00667
00668 this (char[] scheme, char[] host, char[] path, char[] query = null)
00669 {
00670 super();
00671
00672 this.scheme = scheme;
00673 this.query = query;
00674 this.host = host;
00675 this.path = path;
00676 }
00677
00678
00679
00680
00681
00682
00683
00684
00685 static MutableUri clone (Uri uri)
00686 {
00687 with (uri)
00688 {
00689 MutableUri ret = new MutableUri (scheme, host, path, query);
00690 ret.userinfo = userinfo;
00691 ret.fragment = fragment;
00692 ret.port = port;
00693 return ret;
00694 }
00695 }
00696
00697
00698
00699
00700
00701
00702
00703 void reset()
00704 {
00705 decoded.reset();
00706 port = InvalidPort;
00707 host = path = query = scheme = userinfo = fragment = null;
00708 }
00709
00710
00711
00712
00713
00714
00715
00716 MutableUri parse (char[] uri)
00717 {
00718 super.parse (uri);
00719 return this;
00720 }
00721
00722
00723
00724
00725
00726
00727
00728 MutableUri setScheme (char[] scheme)
00729 {
00730 this.scheme = scheme;
00731 return this;
00732 }
00733
00734
00735
00736
00737
00738
00739
00740 MutableUri setHost (char[] host)
00741 {
00742 this.host = host;
00743 return this;
00744 }
00745
00746
00747
00748
00749
00750
00751
00752 MutableUri setPort (int port)
00753 {
00754 this.port = port;
00755 return this;
00756 }
00757
00758
00759
00760
00761
00762
00763
00764 MutableUri setUserInfo(char[] userinfo)
00765 {
00766 this.userinfo = userinfo;
00767 return this;
00768 }
00769
00770
00771
00772
00773
00774
00775
00776 MutableUri setQuery (char[] query)
00777 {
00778 this.query = query;
00779 return this;
00780 }
00781
00782
00783
00784
00785
00786
00787
00788 MutableUri setPath (char[] path)
00789 {
00790 this.path = path;
00791 return this;
00792 }
00793
00794
00795
00796
00797
00798
00799
00800 MutableUri setFragment (char[] fragment)
00801 {
00802 this.fragment = fragment;
00803 return this;
00804 }
00805 }
00806
00807
00808