File: urlout.cpp

package info (click to toggle)
ucommon 7.0.1-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 3,068 kB
  • sloc: cpp: 45,917; ansic: 714; sh: 226; makefile: 173
file content (98 lines) | stat: -rw-r--r-- 2,873 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
// Copyright (C) 2013-2014 David Sugar, Tycho Softworks.
// Copyright (C) 2015-2020 Cherokees of Idaho.
//
// This file is part of GNU uCommon C++.
//
// GNU uCommon C++ is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published
// by the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// GNU uCommon C++ is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with GNU uCommon C++.  If not, see <http://www.gnu.org/licenses/>.

#include <ucommon/secure.h>
#include <sys/stat.h>

using namespace ucommon;

static shell::flagopt helpflag('h',"--help",    _TEXT("display this list"));
static shell::flagopt althelp('?', NULL, NULL);
static shell::flagopt reqcert('C', NULL, _TEXT("requires certificate"));
static shell::flagopt verified('V', NULL, _TEXT("requires verification"));
static shell::numericopt port('p', "--port", _TEXT("port to use"), "port", 0);

int main(int argc, char **argv)
{
    shell::bind("urlout");
    shell args(argc, argv);

    if(is(helpflag) || is(althelp) || !args() || args() > 1) {
        printf("%s\n", _TEXT("Usage: urlout [options] url-path"));
        printf("\n%s\n", _TEXT("Options:"));
        shell::help();
        printf("\n%s\n", _TEXT("Report bugs to dyfet@gnu.org"));
        return 0;
    }

    bool url_secure = false;
    const char *url = args[0];
    const char *proto = "80";
    secure::client_t ctx = NULL;
    String svc;

    if(eq(url, "https://", 8)) {
        url_secure = true;
        url += 8;
    }
    else if(eq(url, "http://", 7)) {
        url += 7;
    }
    else
        url_secure = true;

    string_t host = url;
    string_t path = strdup(url);

    if(url_secure && secure::init()) {
        proto = "443";
        ctx = secure::client();
        if(ctx->err() != secure::OK)
            shell::errexit(2, _TEXT("%s: no certificates found"), argv[0]);
    }

    if(is(port))
        svc = str(*port);
    else
        svc = proto;

    const char *find = path.find("/");
    if(find)
        path.rsplit(find);
    else
        path = "/";

    host.split(host.find("/"));

    sstream web(ctx);
    web.open(host, svc);

    if(!web.is_open())
        shell::errexit(1, _TEXT("%s: failed to access"), url);

    if(is(verified) && !web.is_verified())
        shell::errexit(8, _TEXT("%s: unverified host"), url);

    if(is(reqcert) && !web.is_certificate())
        shell::errexit(9, _TEXT("%s: no certificate"), url);

    web << "GET /\r\n\r\n";
    web.flush();

    std::cout << web.rdbuf();
}