asgardius/virtualx-engine
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions
virtualx-engine/tools/editor/fileserver/editor_file_server.cpp
Rémi Verschelde fb277dbde6 i18n: Don't make print_line calls translatable
2016-05-04 15:28:37 +02:00

332 lines
7.6 KiB
C++
Raw Blame History

#include "editor_file_server.h"
#include "io/marshalls.h"
#include "io/marshalls.h"
#include "../editor_settings.h"
//#define DEBUG_PRINT(m_p) print_line(m_p)
#define DEBUG_TIME(m_what) printf("MS: %s - %lli\n",m_what,OS::get_singleton()->get_ticks_usec());
//#define DEBUG_TIME(m_what)
void EditorFileServer::_close_client(ClientData *cd) {
cd->connection->disconnect();
cd->efs->wait_mutex->lock();
cd->efs->to_wait.insert(cd->thread);
cd->efs->wait_mutex->unlock();
while(cd->files.size()) {
memdelete(cd->files.front()->get());
cd->files.erase(cd->files.front());
}
memdelete(cd);
}
void EditorFileServer::_subthread_start(void*s) {
ClientData *cd = (ClientData*)s;
cd->connection->set_nodelay(true);
uint8_t buf4[8];
Error err = cd->connection->get_data(buf4,4);
if (err!=OK) {
_close_client(cd);
ERR_FAIL_COND(err!=OK);
}
int passlen=decode_uint32(buf4);
if (passlen>512) {
_close_client(cd);
ERR_FAIL_COND(passlen>512);
} else if (passlen>0) {
Vector<char> passutf8;
passutf8.resize(passlen+1);
err = cd->connection->get_data((uint8_t*)passutf8.ptr(),passlen);
if (err!=OK) {
_close_client(cd);
ERR_FAIL_COND(err!=OK);
}
passutf8[passlen]=0;
String s;
s.parse_utf8(passutf8.ptr());
if (s!=cd->efs->password) {
encode_uint32(ERR_INVALID_DATA,buf4);
cd->connection->put_data(buf4,4);
OS::get_singleton()->delay_usec(1000000);
_close_client(cd);
ERR_PRINT("CLIENT PASSWORD MISMATCH");
ERR_FAIL();
}
} else {
if (cd->efs->password!="") {
encode_uint32(ERR_INVALID_DATA,buf4);
cd->connection->put_data(buf4,4);
OS::get_singleton()->delay_usec(1000000);
_close_client(cd);
ERR_PRINT("CLIENT PASSWORD MISMATCH (should be empty!)");
ERR_FAIL();
}
}
encode_uint32(OK,buf4);
cd->connection->put_data(buf4,4);
while(!cd->quit) {
//wait for ID
err = cd->connection->get_data(buf4,4);
//#define DEBUG_PRINT(m_p) print_line(m_p)
DEBUG_TIME("get_data")
if (err!=OK) {
_close_client(cd);
ERR_FAIL_COND(err!=OK);
}
int id=decode_uint32(buf4);
//wait for command
err = cd->connection->get_data(buf4,4);
if (err!=OK) {
_close_client(cd);
ERR_FAIL_COND(err!=OK);
}
int cmd=decode_uint32(buf4);
switch(cmd) {
case FileAccessNetwork::COMMAND_FILE_EXISTS:
case FileAccessNetwork::COMMAND_GET_MODTIME:
case FileAccessNetwork::COMMAND_OPEN_FILE: {
DEBUG_TIME("open_file")
err = cd->connection->get_data(buf4,4);
if (err!=OK) {
_close_client(cd);
ERR_FAIL_COND(err!=OK);
}
int namelen=decode_uint32(buf4);
Vector<char> fileutf8;
fileutf8.resize(namelen+1);
err = cd->connection->get_data((uint8_t*)fileutf8.ptr(),namelen);
if (err!=OK) {
_close_client(cd);
ERR_FAIL_COND(err!=OK);
}
fileutf8[namelen]=0;
String s;
s.parse_utf8(fileutf8.ptr());
if (cmd==FileAccessNetwork::COMMAND_FILE_EXISTS) {
print_line("FILE EXISTS: "+s);
}
if (cmd==FileAccessNetwork::COMMAND_GET_MODTIME) {
print_line("MOD TIME: "+s);
}
if (cmd==FileAccessNetwork::COMMAND_OPEN_FILE) {
print_line("OPEN: "+s);
}
if ( !s.begins_with("res://")) {
_close_client(cd);
ERR_FAIL_COND(!s.begins_with("res://"));
}
ERR_CONTINUE(cd->files.has(id));
if (cmd==FileAccessNetwork::COMMAND_FILE_EXISTS) {
encode_uint32(id,buf4);
cd->connection->put_data(buf4,4);
encode_uint32(FileAccessNetwork::RESPONSE_FILE_EXISTS,buf4);
cd->connection->put_data(buf4,4);
encode_uint32(FileAccess::exists(s),buf4);
cd->connection->put_data(buf4,4);
DEBUG_TIME("open_file_end")
break;
}
if (cmd==FileAccessNetwork::COMMAND_GET_MODTIME) {
encode_uint32(id,buf4);
cd->connection->put_data(buf4,4);
encode_uint32(FileAccessNetwork::RESPONSE_GET_MODTIME,buf4);
cd->connection->put_data(buf4,4);
encode_uint64(FileAccess::get_modified_time(s),buf4);
cd->connection->put_data(buf4,8);
DEBUG_TIME("open_file_end")
break;
}
FileAccess *fa = FileAccess::open(s,FileAccess::READ);
if (!fa) {
//not found, continue
encode_uint32(id,buf4);
cd->connection->put_data(buf4,4);
encode_uint32(FileAccessNetwork::RESPONSE_OPEN,buf4);
cd->connection->put_data(buf4,4);
encode_uint32(ERR_FILE_NOT_FOUND,buf4);
cd->connection->put_data(buf4,4);
DEBUG_TIME("open_file_end")
break;
}
encode_uint32(id,buf4);
cd->connection->put_data(buf4,4);
encode_uint32(FileAccessNetwork::RESPONSE_OPEN,buf4);
cd->connection->put_data(buf4,4);
encode_uint32(OK,buf4);
cd->connection->put_data(buf4,4);
encode_uint64(fa->get_len(),buf4);
cd->connection->put_data(buf4,8);
cd->files[id]=fa;
DEBUG_TIME("open_file_end")
} break;
case FileAccessNetwork::COMMAND_READ_BLOCK: {
err = cd->connection->get_data(buf4,8);
if (err!=OK) {
_close_client(cd);
ERR_FAIL_COND(err!=OK);
}
ERR_CONTINUE(!cd->files.has(id));
uint64_t offset = decode_uint64(buf4);
err = cd->connection->get_data(buf4,4);
if (err!=OK) {
_close_client(cd);
ERR_FAIL_COND(err!=OK);
}
int blocklen=decode_uint32(buf4);
ERR_CONTINUE(blocklen > (16*1024*1024));
cd->files[id]->seek(offset);
Vector<uint8_t> buf;
buf.resize(blocklen);
int read = cd->files[id]->get_buffer(buf.ptr(),blocklen);
ERR_CONTINUE(read<0);
print_line("GET BLOCK - offset: "+itos(offset)+", blocklen: "+itos(blocklen));
//not found, continue
encode_uint32(id,buf4);
cd->connection->put_data(buf4,4);
encode_uint32(FileAccessNetwork::RESPONSE_DATA,buf4);
cd->connection->put_data(buf4,4);
encode_uint64(offset,buf4);
cd->connection->put_data(buf4,8);
encode_uint32(read,buf4);
cd->connection->put_data(buf4,4);
cd->connection->put_data(buf.ptr(),read);
} break;
case FileAccessNetwork::COMMAND_CLOSE: {
print_line("CLOSED");
ERR_CONTINUE(!cd->files.has(id));
memdelete(cd->files[id]);
cd->files.erase(id);
} break;
}
}
_close_client(cd);
}
void EditorFileServer::_thread_start(void*s) {
EditorFileServer *self=(EditorFileServer*)s;
while(!self->quit) {
if (self->cmd==CMD_ACTIVATE) {
self->server->listen(self->port);
self->active=true;
self->cmd=CMD_NONE;
} else if (self->cmd==CMD_STOP) {
self->server->stop();
self->active=false;
self->cmd=CMD_NONE;
}
if (self->active) {
if (self->server->is_connection_available()) {
ClientData *cd = memnew( ClientData);
cd->connection=self->server->take_connection();
cd->efs=self;
cd->quit=false;
cd->thread=Thread::create(_subthread_start,cd);
}
}
self->wait_mutex->lock();
while (self->to_wait.size()) {
Thread *w = self->to_wait.front()->get();
self->to_wait.erase(w);
self->wait_mutex->unlock();
Thread::wait_to_finish(w);
memdelete(w);
self->wait_mutex->lock();
}
self->wait_mutex->unlock();
OS::get_singleton()->delay_usec(100000);
}
}
void EditorFileServer::start() {
stop();
port=EDITOR_DEF("file_server/port",6010);
password=EDITOR_DEF("file_server/password","");
cmd=CMD_ACTIVATE;
}
bool EditorFileServer::is_active() const {
return active;
}
void EditorFileServer::stop(){
cmd=CMD_STOP;
}
EditorFileServer::EditorFileServer() {
server = TCP_Server::create_ref();
wait_mutex = Mutex::create();
quit=false;
active=false;
cmd=CMD_NONE;
thread=Thread::create(_thread_start,this);
EDITOR_DEF("file_server/port",6010);
EDITOR_DEF("file_server/password","");
}
EditorFileServer::~EditorFileServer() {
quit=true;
Thread::wait_to_finish(thread);
memdelete(thread);
memdelete(wait_mutex);
}
Reference in a new issue View git blame Copy permalink