mirror of
https://github.com/RGBCube/serenity
synced 2025-07-25 15:07:45 +00:00
pro: Take the URL to download as a command-line argument
Also, don't print anything other than the download payload to stdout. This gives us a very simple HTTP download utility :^)
This commit is contained in:
parent
eb85103271
commit
600c15aa3a
1 changed files with 18 additions and 10 deletions
|
@ -1,21 +1,29 @@
|
||||||
|
#include <AK/URL.h>
|
||||||
|
#include <LibC/SharedBuffer.h>
|
||||||
#include <LibCore/CEventLoop.h>
|
#include <LibCore/CEventLoop.h>
|
||||||
#include <LibProtocol/Client.h>
|
#include <LibProtocol/Client.h>
|
||||||
#include <LibC/SharedBuffer.h>
|
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
|
|
||||||
int main(int argc, char** argv)
|
int main(int argc, char** argv)
|
||||||
{
|
{
|
||||||
(void)argc;
|
if (argc != 2) {
|
||||||
(void)argv;
|
printf("usage: %s <url>\n", argv[0]);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
String url_string = argv[1];
|
||||||
|
URL url(url_string);
|
||||||
|
if (!url.is_valid()) {
|
||||||
|
fprintf(stderr, "'%s' is not a valid URL\n", url_string.characters());
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
CEventLoop loop;
|
CEventLoop loop;
|
||||||
auto protocol_client = LibProtocol::Client::construct();
|
auto protocol_client = LibProtocol::Client::construct();
|
||||||
protocol_client->handshake();
|
protocol_client->handshake();
|
||||||
|
|
||||||
printf("supports HTTP? %u\n", protocol_client->is_supported_protocol("http"));
|
|
||||||
printf(" supports FTP? %u\n", protocol_client->is_supported_protocol("ftp"));
|
|
||||||
|
|
||||||
protocol_client->on_download_finish = [&](i32 download_id, bool success, u32 total_size, i32 shared_buffer_id) {
|
protocol_client->on_download_finish = [&](i32 download_id, bool success, u32 total_size, i32 shared_buffer_id) {
|
||||||
printf("download %d finished, success=%u, shared_buffer_id=%d\n", download_id, success, shared_buffer_id);
|
dbgprintf("download %d finished, success=%u, shared_buffer_id=%d\n", download_id, success, shared_buffer_id);
|
||||||
if (success) {
|
if (success) {
|
||||||
ASSERT(shared_buffer_id != -1);
|
ASSERT(shared_buffer_id != -1);
|
||||||
auto shared_buffer = SharedBuffer::create_from_shared_buffer_id(shared_buffer_id);
|
auto shared_buffer = SharedBuffer::create_from_shared_buffer_id(shared_buffer_id);
|
||||||
|
@ -26,11 +34,11 @@ int main(int argc, char** argv)
|
||||||
};
|
};
|
||||||
|
|
||||||
protocol_client->on_download_progress = [&](i32 download_id, u32 total_size, u32 downloaded_size) {
|
protocol_client->on_download_progress = [&](i32 download_id, u32 total_size, u32 downloaded_size) {
|
||||||
printf("download %d progress: %u / %u\n", download_id, downloaded_size, total_size);
|
dbgprintf("download %d progress: %u / %u\n", download_id, downloaded_size, total_size);
|
||||||
};
|
};
|
||||||
|
|
||||||
i32 download_id = protocol_client->start_download("http://192.168.1.21/");
|
i32 download_id = protocol_client->start_download(url.to_string());
|
||||||
printf("started download with id %d\n", download_id);
|
dbgprintf("started download with id %d\n", download_id);
|
||||||
|
|
||||||
return loop.exec();
|
return loop.exec();
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue