TheCloud/thor-os
1
0
Fork 0
mirror of https://github.com/wichtounet/thor-os.git synced 2025-08-04 01:36:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Complete wget

Browse Source
This commit is contained in:
Baptiste Wicht 2016-09-20 19:50:07 +02:00
parent 8ff35de098
commit 02692534c8
1 changed files with 66 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

87
programs/wget/src/main.cpp
View File

@ -24,29 +24,74 @@ int wget_http(const std::string& url){
if(parts[0] == "http:"){
auto& domain = parts[1];
auto ip = tlib::dns::resolve(domain);
tlib::ip::address ip;
if (tlib::dns::is_ip(domain)) {
auto ip_parts = std::split(domain, '.');
if(ip){
tlib::socket sock(tlib::socket_domain::AF_INET, tlib::socket_type::STREAM, tlib::socket_protocol::TCP);
sock.connect(*ip, 80);
sock.listen(true);
if (!sock) {
tlib::printf("nc: socket error: %s\n", std::error_message(sock.error()));
return 1;
}
//TODO
sock.listen(false);
if (!sock) {
tlib::printf("nc: socket error: %s\n", std::error_message(sock.error()));
return 1;
}
ip = tlib::ip::make_address(std::atoui(ip_parts[0]), std::atoui(ip_parts[1]), std::atoui(ip_parts[2]), std::atoui(ip_parts[3]));
} else {
tlib::print_line("wget: cannot resolve the domain");
auto ip_result = tlib::dns::resolve(domain);
if(!ip_result){
tlib::print_line("wget: cannot resolve the domain");
return 1;
}
ip = *ip_result;
}
tlib::socket sock(tlib::socket_domain::AF_INET, tlib::socket_type::STREAM, tlib::socket_protocol::TCP);
sock.connect(ip, 80);
sock.listen(true);
if (!sock) {
tlib::printf("nc: socket error: %s\n", std::error_message(sock.error()));
return 1;
}
std::string message;
message += "GET ";
if (parts.size() < 3) {
message += '/';
} else {
for (size_t i = 2; i < parts.size(); ++i) {
message += '/';
message += parts[i];
}
}
message += " HTTP/1.1\r\n";
message += "Host: ";
message += domain;
message += "\r\n";
message += "Accept: text/html text/plain\r\n";
message += "User-Agent: wget (Thor OS)\r\n";
message += "\r\n";
sock.send(message.c_str(), message.size());
char message_buffer[2049];
auto size = sock.receive(message_buffer, 2048, 2000);
if (!sock) {
if (sock.error() == std::ERROR_SOCKET_TIMEOUT) {
tlib::printf("Timeout\n");
return 1;
}
tlib::printf("nc: receive error: %s\n", std::error_message(sock.error()));
return 1;
} else {
message_buffer[size] = '\0';
tlib::print(message_buffer);
}
sock.listen(false);
if (!sock) {
tlib::printf("nc: socket error: %s\n", std::error_message(sock.error()));
return 1;
}
} else {