summaryrefslogtreecommitdiff
path: root/tools/editor/fileserver
diff options
context:
space:
mode:
Diffstat (limited to 'tools/editor/fileserver')
-rw-r--r--tools/editor/fileserver/SCsub7
-rw-r--r--tools/editor/fileserver/editor_file_server.cpp329
-rw-r--r--tools/editor/fileserver/editor_file_server.h59
3 files changed, 395 insertions, 0 deletions
diff --git a/tools/editor/fileserver/SCsub b/tools/editor/fileserver/SCsub
new file mode 100644
index 0000000000..b525fb3f75
--- /dev/null
+++ b/tools/editor/fileserver/SCsub
@@ -0,0 +1,7 @@
+Import('env')
+Export('env')
+env.add_source_files(env.tool_sources,"*.cpp")
+
+
+
+
diff --git a/tools/editor/fileserver/editor_file_server.cpp b/tools/editor/fileserver/editor_file_server.cpp
new file mode 100644
index 0000000000..730378f953
--- /dev/null
+++ b/tools/editor/fileserver/editor_file_server.cpp
@@ -0,0 +1,329 @@
+#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);
+ 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();
+ thread=Thread::create(_thread_start,this);
+ wait_mutex = Mutex::create();
+ quit=false;
+ active=false;
+ cmd=CMD_NONE;
+
+ EDITOR_DEF("file_server/port",6010);
+ EDITOR_DEF("file_server/password","");
+}
+
+EditorFileServer::~EditorFileServer() {
+
+ quit=true;
+ Thread::wait_to_finish(thread);
+ memdelete(wait_mutex);
+}
diff --git a/tools/editor/fileserver/editor_file_server.h b/tools/editor/fileserver/editor_file_server.h
new file mode 100644
index 0000000000..587b2c4fdb
--- /dev/null
+++ b/tools/editor/fileserver/editor_file_server.h
@@ -0,0 +1,59 @@
+#ifndef EDITOR_FILE_SERVER_H
+#define EDITOR_FILE_SERVER_H
+
+#include "object.h"
+#include "os/thread.h"
+#include "io/tcp_server.h"
+#include "io/packet_peer.h"
+#include "io/file_access_network.h"
+
+class EditorFileServer : public Object {
+
+ OBJ_TYPE(EditorFileServer,Object);
+
+ enum Command {
+ CMD_NONE,
+ CMD_ACTIVATE,
+ CMD_STOP,
+ };
+
+
+ struct ClientData {
+
+ Thread *thread;
+ Ref<StreamPeerTCP> connection;
+ Map<int,FileAccess*> files;
+ EditorFileServer *efs;
+ bool quit;
+
+ };
+
+ Ref<TCP_Server> server;
+ Set<Thread*> to_wait;
+
+ static void _close_client(ClientData *cd);
+ static void _subthread_start(void*s);
+
+ Mutex *wait_mutex;
+ Thread *thread;
+ static void _thread_start(void*);
+ bool quit;
+ Command cmd;
+
+ String password;
+ int port;
+ bool active;
+
+
+public:
+
+ void start();
+ void stop();
+
+ bool is_active() const;
+
+ EditorFileServer();
+ ~EditorFileServer();
+};
+
+#endif // EDITOR_FILE_SERVER_H
generated by cgit v1.2.3 (git 2.25.1) at 2024-11-24 10:51:01 +0000