extra bdstorage tasks

not needed, but felt like adding anyways

src/client/game/demonware/data_types.hpp

@@ -145,28 +145,28 @@ namespace demonware
 	class bdFileQueryResult final : public bdTaskResult
 	{
 	public:
-		std::uint64_t owner_id;
+		std::uint64_t user_id;
 		std::string platform;
 		std::string filename;
 		std::uint32_t errorcode;
-		std::string data;
+		std::string filedata;
 
 		void serialize(byte_buffer* buffer) override
 		{
-			buffer->write_uint64(this->owner_id);
+			buffer->write_uint64(this->user_id);
 			buffer->write_string(this->platform);
 			buffer->write_string(this->filename);
 			buffer->write_uint32(this->errorcode);
-			buffer->write_blob(this->data);
+			buffer->write_blob(this->filedata);
 		}
 
 		void deserialize(byte_buffer* buffer) override
 		{
-			buffer->read_uint64(&this->owner_id);
+			buffer->read_uint64(&this->user_id);
 			buffer->read_string(&this->platform);
 			buffer->read_string(&this->filename);
 			buffer->read_uint32(&this->errorcode);
-			buffer->read_blob(&this->data);
+			buffer->read_blob(&this->filedata);
 		}
 	};
 
@@ -206,4 +206,35 @@ namespace demonware
 			buffer->read_blob(&this->data);
 		}
 	};
+
+	class bdContextUserStorageFileInfo final : public bdTaskResult
+	{
+	public:
+		uint32_t create_time;
+		uint32_t modifed_time;
+		bool priv;
+		uint64_t owner_id;
+		std::string account_type;
+		std::string filename;
+
+		void serialize(byte_buffer* buffer) override
+		{
+			buffer->write_uint32(this->create_time);
+			buffer->write_uint32(this->modifed_time);
+			buffer->write_bool(this->priv);
+			buffer->write_uint64(this->owner_id);
+			buffer->write_string(this->account_type);
+			buffer->write_string(this->filename);
+		}
+
+		void deserialize(byte_buffer* buffer) override
+		{
+			buffer->read_uint32(&this->create_time);
+			buffer->read_uint32(&this->modifed_time);
+			buffer->read_bool(&this->priv);
+			buffer->read_uint64(&this->owner_id);
+			buffer->read_string(&this->account_type);
+			buffer->read_string(&this->filename);
+		}
+	};
 }

src/client/game/demonware/services/bdStorage.cpp

@@ -14,8 +14,10 @@ namespace demonware
 		this->register_task(20, &bdStorage::list_publisher_files);
 		this->register_task(21, &bdStorage::get_publisher_file);
 		this->register_task(24, &bdStorage::upload_and_validate_files);
+		this->register_task(18, &bdStorage::upload_files);
 		this->register_task(16, &bdStorage::get_user_files);
 		this->register_task(12, &bdStorage::get_user_file);
+		this->register_task(10, &bdStorage::set_user_file);
 
 		this->map_publisher_resource("motd-.*\\.txt", DW_MOTD);
 		// this->map_publisher_resource("ffotd-.*\\.ff", DW_FASTFILE);
@@ -128,6 +130,36 @@ namespace demonware
 		}
 	}
 
+	void bdStorage::set_user_file(service_server* server, byte_buffer* buffer) const
+	{
+		bool priv;
+		uint64_t owner;
+		std::string game, filename, data;
+
+		buffer->read_string(&game);
+		buffer->read_string(&filename);
+		buffer->read_bool(&priv);
+		buffer->read_blob(&data);
+		buffer->read_uint64(&owner);
+
+		const auto path = get_user_file_path(filename);
+		utils::io::write_file(path, data);
+
+		auto* info = new bdFileInfo;
+
+		info->file_id = *reinterpret_cast<const uint64_t*>(utils::cryptography::sha1::compute(filename).data());
+		info->filename = filename;
+		info->create_time = uint32_t(time(nullptr));
+		info->modified_time = info->create_time;
+		info->file_size = uint32_t(data.size());
+		info->owner_id = owner;
+		info->priv = priv;
+
+		auto reply = server->create_reply(this->task_id());
+		reply->add(info);
+		reply->send();
+	}
+
 	std::string bdStorage::get_user_file_path(const std::string& name)
 	{
 		return "players2/user/" + name;
@@ -146,10 +178,6 @@ namespace demonware
 
 		auto reply = server->create_reply(this->task_id());
 
-#ifdef DW_DEBUG
-		printf("[DW]: [bdStorage]: upload_and_validate_files has %d files\n", numfiles);
-#endif
-
 		for (uint32_t i = 0; i < numfiles; i++)
 		{
 			std::string filename, data;
@@ -176,16 +204,12 @@ namespace demonware
 			info->data = data;
 
 #ifdef DW_DEBUG
-			printf("[DW]: [bdStorage]: adding \"%s\" to reply\n", filename.data());
+			printf("[DW]: [bdStorage]: set user file: %s\n", filename.data());
 #endif
 
 			reply->add(info);
 		}
 
-#ifdef DW_DEBUG
-		printf("[DW]: [bdStorage]: sending reply for upload_and_validate_files\n");
-#endif
-
 		reply->send();
 	}
 
@@ -194,10 +218,6 @@ namespace demonware
 		std::string context;
 		buffer->read_string(&context);
 
-#ifdef DW_DEBUG
-		printf("[DW]: [bdStorage]: context is '%s'\n", context.data());
-#endif
-
 		uint32_t count;
 		buffer->read_uint32(&count);
 
@@ -220,51 +240,85 @@ namespace demonware
 		{
 			std::string filename;
 			buffer->read_string(&filename);
-
 			filenames.push_back(std::move(filename));
 		}
 
 		auto reply = server->create_reply(this->task_id());
-		uint32_t available = 0;
-
 		for (size_t i = 0u; i < filenames.size(); i++)
 		{
-			auto& name = filenames.at(i);
-			std::string data;
-
-			if (utils::io::read_file(get_user_file_path(name), &data))
-			{
-				auto entry = new bdFileQueryResult;
-				entry->owner_id = user_ctxs.at(i).first;
+			auto* entry = new bdFileQueryResult;
+			entry->user_id = user_ctxs.at(i).first;
 			entry->platform = user_ctxs.at(i).second;
 			entry->filename = filenames.at(i);
 			entry->errorcode = 0;
-				entry->data = data;
+
+			auto& name = filenames.at(i);
+			std::string filedata;
+			if (utils::io::read_file(get_user_file_path(name), &filedata))
+			{
+				entry->filedata = filedata;
+#ifdef DW_DEBUG
+				printf("[DW]: [bdStorage]: get user file: %s\n", name.data());
+#endif
+			}
+			else
+			{
+				entry->errorcode = game::BD_NO_FILE;
+#ifdef DW_DEBUG
+				printf("[DW]: [bdStorage]: missing user file: %s\n", name.data());
+#endif
+			}
 
 			reply->add(entry);
-
-				++available;
-
-#ifdef DW_DEBUG
-				printf("[DW]: [bdStorage]: user file \"%s\" dispatched\n", name.data());
-#endif
-			}
-			else
-			{
-#ifdef DW_DEBUG
-				printf("[DW]: [bdStorage]: user file \"%s\" not found\n", name.data());
-#endif
-			}
 		}
 
-		if (available == count)
-		{
 		reply->send();
 	}
-		else
+
+	void bdStorage::upload_files(service_server* server, byte_buffer* buffer) const
 	{
-			server->create_reply(this->task_id(), game::BD_NO_FILE)->send();
+		uint64_t owner;
+		uint32_t numfiles;
+		std::string game, platform;
+
+		buffer->read_string(&game);
+		buffer->read_uint64(&owner);
+		buffer->read_string(&platform);
+		buffer->read_uint32(&numfiles);
+
+		auto reply = server->create_reply(this->task_id());
+
+		for (uint32_t i = 0; i < numfiles; i++)
+		{
+			std::string filename, data;
+			uint32_t version;
+			bool priv;
+
+			buffer->read_string(&filename);
+			buffer->read_blob(&data);
+			buffer->read_uint32(&version);
+			buffer->read_bool(&priv);
+
+			const auto path = get_user_file_path(filename);
+			utils::io::write_file(path, data);
+
+			auto* info = new bdContextUserStorageFileInfo;
+
+			info->modifed_time = static_cast<uint32_t>(time(nullptr));
+			info->create_time = info->modifed_time;
+			info->priv = priv;
+			info->owner_id = owner;
+			info->account_type = platform;
+			info->filename = filename;
+
+#ifdef DW_DEBUG
+			printf("[DW]: [bdStorage]: set user file: %s\n", filename.data());
+#endif
+
+			reply->add(info);
 		}
+
+		reply->send();
 	}
 
 	void bdStorage::get_user_file(service_server* server, byte_buffer* buffer) const

src/client/game/demonware/services/bdStorage.hpp

@@ -19,8 +19,10 @@ namespace demonware
 		void list_publisher_files(service_server* server, byte_buffer* buffer);
 		void get_publisher_file(service_server* server, byte_buffer* buffer);
 		void upload_and_validate_files(service_server* server, byte_buffer* buffer) const;
+		void upload_files(service_server* server, byte_buffer* buffer) const;
 		void get_user_files(service_server* server, byte_buffer* buffer) const;
 		void get_user_file(service_server* server, byte_buffer* buffer) const;
+		void set_user_file(service_server* server, byte_buffer* buffer) const;
 
 		static std::string get_user_file_path(const std::string& name);
 	};