Luna/web_serv
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

improved the post receive file

Browse source
This commit is contained in:
Luna 2025-05-03 04:12:38 +02:00
parent 8c3661889f
commit e7b557f754
1 changed files with 32 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

35
main.cpp
View file

@ -1,5 +1,7 @@
#include "netlib/src/netlib.h" #include "netlib/src/netlib.h"
#ifdef __linux__
#include <sys/sendfile.h> #include <sys/sendfile.h>
#endif
#include <print> #include <print>
#include <fstream> #include <fstream>
@ -18,6 +20,20 @@ int atoi_newline(const char *data)
return ret; return ret;
} }
std::string get_filename(std::string_view data)
{
int index = 1;
std::string ret;
while (true)
{
if (data[index] == '"')
break;
ret.push_back(data[index]);
index++;
}
return ret;
}
int main() int main()
{ {
netlib::server_raw server; netlib::server_raw server;
@ -53,7 +69,12 @@ int main()
size = file.tellg() - size; size = file.tellg() - size;
file.close(); file.close();
int filefd = open("../test.html", O_RDONLY); int filefd = open("../test.html", O_RDONLY);
#ifdef __linux__
sendfile(user, filefd, 0, size); sendfile(user, filefd, 0, size);
#endif
#ifdef __APPLE__
sendfile(filefd, user, 0, (long long*)&size, nullptr, 0);
#endif
} }
server.disconnect_user(user); server.disconnect_user(user);
} }
@ -64,10 +85,18 @@ int main()
dat.remove_prefix(strlen("Content-Length: ")); dat.remove_prefix(strlen("Content-Length: "));
int lenght = atoi_newline(dat.data()); int lenght = atoi_newline(dat.data());
std::println("{} {}", dat, lenght); std::println("{} {}", dat, lenght);
pos = dat.find("image/png"); pos = dat.find("filename=");
dat.remove_prefix(pos); dat.remove_prefix(pos);
dat.remove_prefix(strlen("image/png\r\n") + 2); dat.remove_prefix(strlen("filename="));
std::ofstream a("Test.png", std::ios::binary); std::string_view file_data(data);
std::string filename = get_filename(dat);
std::println("Filename {}", filename);
dat.remove_prefix(dat.find("\r\n\r\n"));
dat.remove_prefix(dat.find("\r\n\r\n"));
dat.remove_prefix(4);
std::println("{}", dat);
std::ofstream a(filename, std::ios::binary);
a.write(dat.data(), lenght); a.write(dat.data(), lenght);
netlib::send_packet(std::make_tuple(std::string("HTTP/1.1 201 CREATED")), user); netlib::send_packet(std::make_tuple(std::string("HTTP/1.1 201 CREATED")), user);
server.disconnect_user(user); server.disconnect_user(user);