From 0606e7010c7789cabacb33050bcb6a170ad7773b Mon Sep 17 00:00:00 2001
From: B3n30 <benediktthomas@gmail.com>
Date: Sun, 22 Jul 2018 12:15:03 +0200
Subject: [PATCH] Services/HTTP: Implemented the AddRequestHeader service
 function.

---
 src/core/hle/service/http_c.cpp | 37 ++++++++++++++++++++++++++++++++-
 src/core/hle/service/http_c.h   | 15 +++++++++++++
 2 files changed, 51 insertions(+), 1 deletion(-)

diff --git a/src/core/hle/service/http_c.cpp b/src/core/hle/service/http_c.cpp
index 12dae5671..3c55b3c84 100644
--- a/src/core/hle/service/http_c.cpp
+++ b/src/core/hle/service/http_c.cpp
@@ -188,6 +188,41 @@ void HTTP_C::CloseContext(Kernel::HLERequestContext& ctx) {
     rb.Push(RESULT_SUCCESS);
 }
 
+void HTTP_C::AddRequestHeader(Kernel::HLERequestContext& ctx) {
+    IPC::RequestParser rp(ctx, 0x11, 3, 4);
+    const u32 context_handle = rp.Pop<u32>();
+    const u32 name_size = rp.Pop<u32>();
+    const u32 value_size = rp.Pop<u32>();
+    const std::vector<u8> name_buffer = rp.PopStaticBuffer();
+    Kernel::MappedBuffer& value_buffer = rp.PopMappedBuffer();
+    const std::string name(name_buffer.begin(), name_buffer.end() - 1);
+    std::string value(value_size - 1, '\0');
+    value_buffer.Read(&value[0], 0, value_size - 1);
+
+    auto itr = contexts.find(context_handle);
+    if (itr == contexts.end()) {
+        IPC::RequestBuilder rb = rp.MakeBuilder(1, 0);
+        rb.Push(ERROR_CONTEXT_ERROR);
+        LOG_ERROR(Service_HTTP, "called, context {} not found", context_handle);
+        return;
+    }
+
+    ASSERT(itr->second.state == RequestState::NotStarted);
+    ASSERT(std::find_if(itr->second.headers.begin(), itr->second.headers.end(),
+                        [&name](const Context::RequestHeader& m) -> bool {
+                            return m.name == name;
+                        }) == itr->second.headers.end());
+
+    itr->second.headers.push_back(Context::RequestHeader{name, value});
+
+    IPC::RequestBuilder rb = rp.MakeBuilder(1, 2);
+    rb.Push(RESULT_SUCCESS);
+    rb.PushMappedBuffer(value_buffer);
+
+    LOG_WARNING(Service_HTTP, "called, name={}, value={}, context_handle={}", name, value,
+                context_handle);
+}
+
 HTTP_C::HTTP_C() : ServiceFramework("http:C", 32) {
     static const FunctionInfo functions[] = {
         {0x00010044, &HTTP_C::Initialize, "Initialize"},
@@ -206,7 +241,7 @@ HTTP_C::HTTP_C() : ServiceFramework("http:C", 32) {
         {0x000E0040, nullptr, "SetProxyDefault"},
         {0x000F00C4, nullptr, "SetBasicAuthorization"},
         {0x00100080, nullptr, "SetSocketBufferSize"},
-        {0x001100C4, nullptr, "AddRequestHeader"},
+        {0x001100C4, &HTTP_C::AddRequestHeader, "AddRequestHeader"},
         {0x001200C4, nullptr, "AddPostDataAscii"},
         {0x001300C4, nullptr, "AddPostDataBinary"},
         {0x00140082, nullptr, "AddPostDataRaw"},
diff --git a/src/core/hle/service/http_c.h b/src/core/hle/service/http_c.h
index cc4c334a9..a548114ea 100644
--- a/src/core/hle/service/http_c.h
+++ b/src/core/hle/service/http_c.h
@@ -54,6 +54,21 @@ private:
      */
     void CloseContext(Kernel::HLERequestContext& ctx);
 
+    /**
+     * HTTP_C::AddRequestHeader service function
+     *  Inputs:
+     * 1 : Context handle
+     * 2 : Header name buffer size, including null-terminator.
+     * 3 : Header value buffer size, including null-terminator.
+     * 4 : (HeaderNameSize<<14) | 0xC02
+     * 5 : Header name data pointer
+     * 6 : (HeaderValueSize<<4) | 10
+     * 7 : Header value data pointer
+     *  Outputs:
+     *      1 : Result of function, 0 on success, otherwise error code
+     */
+    void AddRequestHeader(Kernel::HLERequestContext& ctx);
+
     Kernel::SharedPtr<Kernel::SharedMemory> shared_memory = nullptr;
 
     std::unordered_map<u32, Context> contexts;