#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 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 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 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); 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); List local_ip; IP::get_singleton()->get_local_addresses(&local_ip); EDITOR_DEF("file_server/port",6010); String lip; String hint; for(List::Element *E=local_ip.front();E;E=E->next()) { String ip = E->get(); if (ip=="127.0.0.1") continue; if (lip!="") lip=ip; if (hint!="") hint+=","; hint+=ip; } EDITOR_DEF("file_server/host",lip); EditorSettings::get_singleton()->add_property_hint(PropertyInfo(Variant::STRING,"file_server/host",PROPERTY_HINT_ENUM,hint)); EDITOR_DEF("file_server/password",""); } EditorFileServer::~EditorFileServer() { quit=true; Thread::wait_to_finish(thread); memdelete(wait_mutex); }