URLSessionTask.c

Go to the documentation of this file.

/*
 * Objectively: Ultra-lightweight object oriented framework for GNU C.
 * Copyright (C) 2014 Jay Dolan <jay@jaydolan.com>
 *
 * This software is provided 'as-is', without any express or implied
 * warranty. In no event will the authors be held liable for any damages
 * arising from the use of this software.
 *
 * Permission is granted to anyone to use this software for any purpose,
 * including commercial applications, and to alter it and redistribute it
 * freely, subject to the following restrictions:
 *
 * 1. The origin of this software must not be misrepresented; you must not
 * claim that you wrote the original software. If you use this software
 * in a product, an acknowledgment in the product documentation would be
 * appreciated but is not required.
 *
 * 2. Altered source versions must be plainly marked as such, and must not be
 * misrepresented as being the original software.
 *
 * 3. This notice may not be removed or altered from any source distribution.
 */
 
#include <assert.h>
#include <string.h>
#include <stdlib.h>
#include <curl/curl.h>
 
#include "URLRequest.h"
#include "URLResponse.h"
#include "URLSession.h"
#include "URLSessionTask.h"
 
#define _Class _URLSessionTask
 
#pragma mark - Object
 
static Object *copy(const Object *self) {
 
  return NULL;
}
 
static void dealloc(Object *self) {
 
  URLSessionTask *this = (URLSessionTask *) self;
 
  if (this->error) {
    free(this->error);
  }
 
  release(this->request);
  release(this->session);
  release(this->response);
 
  super(Object, self, dealloc);
}
 
#pragma mark - URLSessionTask
 
static void cancel(URLSessionTask *self) {
 
  switch (self->state) {
    case URLSESSIONTASK_RESUMING:
    case URLSESSIONTASK_RESUMED:
    case URLSESSIONTASK_SUSPENDING:
    case URLSESSIONTASK_SUSPENDED:
      self->state = URLSESSIONTASK_CANCELING;
      break;
    default:
      break;
  }
}
 
static void execute(URLSessionTask *self) {
 
  $(self, setup);
 
  curl_easy_perform(self->locals.handle);
 
  curl_easy_getinfo(self->locals.handle, CURLINFO_RESPONSE_CODE, (long *) &self->response->httpStatusCode);
 
  self->state = URLSESSIONTASK_COMPLETED;
 
  $(self, teardown);
}
 
static URLSessionTask *initWithRequestInSession(URLSessionTask *self, struct URLRequest *request,
    struct URLSession *session, URLSessionTaskCompletion completion) {
 
  assert(request);
  assert(session);
 
  self = (URLSessionTask *) super(Object, self, init);
  if (self) {
 
    self->error = calloc(1, CURL_ERROR_SIZE);
    assert(self->error);
 
    self->request = retain(request);
    self->session = retain(session);
 
    self->response = $(alloc(URLResponse), init);
 
    self->completion = completion;
 
    self->state = URLSESSIONTASK_SUSPENDED;
  }
 
  return self;
}
 
static void resume(URLSessionTask *self) {
 
  switch (self->state) {
    case URLSESSIONTASK_SUSPENDING:
    case URLSESSIONTASK_SUSPENDED:
      self->state = URLSESSIONTASK_RESUMING;
      break;
    default:
      break;
  }
}
 
static void requestHeaders_enumerator(const Dictionary *dictionary, ident obj, ident key, ident data) {
 
  String *header = $(alloc(String), initWithFormat, "%s: %s", ((String * ) key)->chars, ((String * ) obj)->chars);
 
  struct curl_slist **headers = (struct curl_slist **) data;
  *headers = curl_slist_append(*headers, header->chars);
 
  release(header);
}
 
static size_t responseHeader(char *buffer, size_t size, size_t count, void *data) {
 
  URLSessionTask *this = (URLSessionTask *) data;
 
  char *header = calloc(count + 1, 1);
  memcpy(header, buffer, count);
 
  char *delim = strchr(header, ':');
  if (delim) {
    *delim = 0;
 
    char *field = header;
    char *value = strtrim(delim + 1);
 
    $(this->response, setValueForHTTPHeaderField, value, field);
 
    free(value);
  }
 
  free(header);
  return count;
}
 
static int progress(ident self, curl_off_t bytesExpectedToReceive, curl_off_t bytesReceived,
    curl_off_t bytesExpectedToSend, curl_off_t bytesSent) {
 
  URLSessionTask *this = (URLSessionTask *) self;
 
  curl_easy_getinfo(this->locals.handle, CURLINFO_RESPONSE_CODE, (long *) &this->response->httpStatusCode);
 
  this->bytesExpectedToReceive = bytesExpectedToReceive;
  this->bytesExpectedToSend = bytesExpectedToSend;
 
  if (this->progress) {
    this->progress(this);
  }
 
  return 0;
}
 
static void setup(URLSessionTask *self) {
 
  self->locals.handle = curl_easy_init();
  assert(self->locals.handle);
 
  curl_easy_setopt(self->locals.handle, CURLOPT_ERRORBUFFER, self->error);
  curl_easy_setopt(self->locals.handle, CURLOPT_FOLLOWLOCATION, true);
 
  const URLSessionConfiguration *config = self->session->configuration;
  if (config->connectTimeout) {
    curl_easy_setopt(self->locals.handle, CURLOPT_CONNECTTIMEOUT, config->connectTimeout);
  }
  if (config->timeout) {
    curl_easy_setopt(self->locals.handle, CURLOPT_TIMEOUT, config->timeout);
  }
  if (!config->longPolling) {
    curl_easy_setopt(self->locals.handle, CURLOPT_LOW_SPEED_LIMIT, 1L);
    curl_easy_setopt(self->locals.handle, CURLOPT_LOW_SPEED_TIME, 10L);
  }
 
  curl_easy_setopt(self->locals.handle, CURLOPT_HEADERFUNCTION, responseHeader);
  curl_easy_setopt(self->locals.handle, CURLOPT_HEADERDATA, self);
 
  curl_easy_setopt(self->locals.handle, CURLOPT_XFERINFOFUNCTION, progress);
  curl_easy_setopt(self->locals.handle, CURLOPT_XFERINFODATA, self);
 
  Data *body = self->request->httpBody;
  if (body) {
    curl_easy_setopt(self->locals.handle, CURLOPT_POSTFIELDS, body->bytes);
    curl_easy_setopt(self->locals.handle, CURLOPT_POSTFIELDSIZE, body->length);
  }
 
  struct curl_slist *httpHeaders = NULL;
  const Dictionary *headers = NULL;
 
  headers = self->session->configuration->httpHeaders;
  if (headers) {
    $(headers, enumerateObjectsAndKeys, requestHeaders_enumerator, &httpHeaders);
  }
 
  headers = self->request->httpHeaders;
  if (headers) {
    $(headers, enumerateObjectsAndKeys, requestHeaders_enumerator, &httpHeaders);
  }
 
  curl_easy_setopt(self->locals.handle, CURLOPT_HTTPHEADER, httpHeaders);
 
  self->locals.requestHeaders = httpHeaders;
 
  switch (self->request->httpMethod) {
    case HTTP_POST:
      curl_easy_setopt(self->locals.handle, CURLOPT_POST, true);
      break;
    case HTTP_PUT:
      curl_easy_setopt(self->locals.handle, CURLOPT_PUT, true);
      break;
    case HTTP_DELETE:
      curl_easy_setopt(self->locals.handle, CURLOPT_CUSTOMREQUEST, "DELETE");
      break;
    case HTTP_PATCH:
      curl_easy_setopt(self->locals.handle, CURLOPT_CUSTOMREQUEST, "PATCH");
      break;
    case HTTP_HEAD:
      curl_easy_setopt(self->locals.handle, CURLOPT_NOBODY, true);
      break;
    case HTTP_OPTIONS:
      curl_easy_setopt(self->locals.handle, CURLOPT_CUSTOMREQUEST, "OPTIONS");
      break;
    default:
      break;
  }
 
  curl_easy_setopt(self->locals.handle, CURLOPT_URL, self->request->url->urlString->chars);
}
 
static void suspend(URLSessionTask *self) {
 
  switch (self->state) {
    case URLSESSIONTASK_RESUMING:
    case URLSESSIONTASK_RESUMED:
      self->state = URLSESSIONTASK_SUSPENDING;
      break;
    default:
      break;
  }
}
 
static void teardown(URLSessionTask *self) {
 
  if (self->locals.handle) {
    curl_easy_cleanup(self->locals.handle);
    self->locals.handle = NULL;
  }
 
  if (self->locals.requestHeaders) {
    curl_slist_free_all(self->locals.requestHeaders);
    self->locals.requestHeaders = NULL;
  }
}
 
#pragma mark - Class lifecycle
 
static void initialize(Class *clazz) {
 
  ((ObjectInterface *) clazz->interface)->copy = copy;
  ((ObjectInterface *) clazz->interface)->dealloc = dealloc;
 
  ((URLSessionTaskInterface *) clazz->interface)->cancel = cancel;
  ((URLSessionTaskInterface *) clazz->interface)->execute = execute;
  ((URLSessionTaskInterface *) clazz->interface)->initWithRequestInSession = initWithRequestInSession;
  ((URLSessionTaskInterface *) clazz->interface)->resume = resume;
  ((URLSessionTaskInterface *) clazz->interface)->setup = setup;
  ((URLSessionTaskInterface *) clazz->interface)->suspend = suspend;
  ((URLSessionTaskInterface *) clazz->interface)->teardown = teardown;
}
 
Class *_URLSessionTask(void) {
  static Class *clazz;
  static Once once;
 
  do_once(&once, {
    clazz = _initialize(&(const ClassDef) {
      .name = "URLSessionTask",
      .superclass = _Object(),
      .instanceSize = sizeof(URLSessionTask),
      .interfaceOffset = offsetof(URLSessionTask, interface),
      .interfaceSize = sizeof(URLSessionTaskInterface),
      .initialize = initialize,
    });
  });
 
  return clazz;
}
 
#undef _Class