Merge pull request #1 from craflin/curl-wrapper

Add CURL wrapper

Author: craflin

CMakeLists.txt

@@ -7,6 +7,16 @@ project(gchsd VERSION 0.2.0)
 set(CDEPLOY_NO_DEBUG_BUILD True)
 set(CDEPLOY_NO_COMPILER True)
 
+find_package(CURL REQUIRED)
+if(NOT TARGET CURL::libcurl)
+    add_library(CURL::libcurl INTERFACE IMPORTED)
+    set_target_properties(CURL::libcurl
+        PROPERTIES 
+            INTERFACE_INCLUDE_DIRECTORIES "${CURL_INCLUDE_DIRS}"
+            INTERFACE_LINK_LIBRARIES "${CURL_LIBRARIES}"
+    )
+endif()
+
 include(CDeploy)
 
 add_subdirectory(3rdparty)
@@ -19,7 +29,7 @@ set(CPACK_PACKAGE_DESCRIPTION_SUMMARY "A mirroring Git HTTP server daemon")
 set(CPACK_PACKAGING_INSTALL_PREFIX "/")
 #set(CPACK_DEBIAN_FILE_NAME "DEB-DEFAULT")
 set(CPACK_DEBIAN_PACKAGE_HOMEPAGE "https://github.com/craflin/gchsd")
-set(CPACK_DEBIAN_PACKAGE_DEPENDS "libc6 (>= 2.15)")
+set(CPACK_DEBIAN_PACKAGE_DEPENDS "libc6 (>= 2.15),libcurl4")
 set(CPACK_DEBIAN_PACKAGE_SECTION "net")
 
 include(CPack)

README.md

@@ -12,6 +12,7 @@ The project is a C++/CURL clone of jonasmalacofilho's [git-cache-http-server](ht
 ## Build Instructions
 
 * Clone the repository and initialize submodules.
+* Install a dev package of curl. (e.g. `libcurl4-openssl-dev`)
 * Build the project with `cmake`.
 * You can build a `deb` package using the target `package` in CMake.
 

src/CMakeLists.txt

@@ -1,6 +1,8 @@
 
 set(sources
     Address.hpp
+    HttpRequest.cpp
+    HttpRequest.hpp
     Main.cpp
     Settings.cpp
     Settings.hpp
@@ -13,7 +15,7 @@ add_executable(gchsd
 )
 
 target_link_libraries(gchsd PRIVATE
-    libnstd::Socket
+    libnstd::Socket CURL::libcurl
 )
 
 source_group("" FILES ${sources})

src/HttpRequest.cpp

@@ -0,0 +1,85 @@
+
+#include <curl/curl.h>
+
+#include "HttpRequest.hpp"
+
+static class Curl
+{
+public:
+  Curl()
+  {
+    curl_global_init(CURL_GLOBAL_ALL);
+  }
+  ~Curl()
+  {
+    curl_global_cleanup();
+  }
+
+} curl;
+
+HttpRequest::HttpRequest() : curl(0) {}
+
+HttpRequest::~HttpRequest()
+{
+  if(curl)
+    curl_easy_cleanup(curl);
+}
+
+bool HttpRequest::get(const String& url, Buffer& data, bool checkCertificate)
+{
+  if(!curl)
+  {
+    curl = curl_easy_init();
+    if(!curl)
+    {
+      error = "Could not initialize curl.";
+      return false;
+    }
+  }
+  else
+    curl_easy_reset(curl);
+
+  struct WriteResult
+  {
+    static size_t writeResponse(void *ptr, size_t size, size_t nmemb, void *stream)
+    {
+      WriteResult* result = (struct WriteResult *)stream;
+      size_t newBytes = size * nmemb;
+      size_t totalSize = result->buffer->size() + newBytes;
+      if(totalSize > (size_t)result->buffer->capacity())
+        result->buffer->reserve(totalSize * 2);
+      result->buffer->append((const byte*)ptr, newBytes);
+      return newBytes;
+    }
+
+    Buffer* buffer;
+  } writeResult = { &data };
+
+  curl_easy_setopt(curl, CURLOPT_URL, (const char*)url);
+  curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, WriteResult::writeResponse);
+  curl_easy_setopt(curl, CURLOPT_WRITEDATA, &writeResult);
+  curl_easy_setopt(curl, CURLOPT_TIMEOUT, 40);
+  if(!checkCertificate)
+    curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, 0);
+
+  data.clear();
+  data.reserve(1500);
+
+  CURLcode status = curl_easy_perform(curl);
+  if(status != 0)
+  {
+    error.printf("%s.", curl_easy_strerror(status));
+    return false;
+  }
+
+  long code;
+  curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, &code);
+  if(code != 200)
+  {
+    error.printf("Server responded with code %u.", (uint)code);
+    return false;
+  }
+
+  error.clear();
+  return true;
+}

src/HttpRequest.hpp

@@ -0,0 +1,21 @@
+
+
+#pragma once
+
+#include <nstd/String.hpp>
+#include <nstd/Buffer.hpp>
+
+class HttpRequest
+{
+public:
+  HttpRequest();
+  ~HttpRequest();
+
+  const String& getErrorString() {return error;}
+
+  bool get(const String& url, Buffer& data, bool checkCertificate = true);
+
+private:
+  void* curl;
+  String error;
+};