Tooltron

#include "query.h"

#include "cache.h"

#include "event.h"

#include "util.h"

#include "log.h"

#include <stdlib.h>

#include <stdio.h>

#include <string.h>

#include <curl/curl.h>

/* Outputs the response to /add_card_event to debug.html */

//#define DEBUG_EVENT_RESPONSE

/* Size of buffer written to by write_buffer(). We are only reading integers

 * back from the server, so we don't need much of the response and anything

 * else can be ignored */

#define WRITE_BUFFER_SIZE 30

static const char *server;

static char *tooltron_password;

static char buffer[WRITE_BUFFER_SIZE];

static int buffer_idx;

int query_init(const char *server_name) {

  CURLcode error_code;

  int len;

  server = server_name;

  error_code = curl_global_init(CURL_GLOBAL_SSL);

  if (error_code) {

    log_print("ERROR: curl_global_init: %s\n", curl_easy_strerror(error_code));

    return error_code;

  }

  tooltron_password = read_file("tooltron_password");

  if (tooltron_password == NULL)

    return 1;

  return 0;

}

void query_cleanup() {

  curl_global_cleanup();

  if (tooltron_password)

    free(tooltron_password);

}

static size_t write_buffer(void *buf_in, size_t size, size_t nmemb, void *userp) {

  size_t to_read;

  to_read = nmemb;

  if (buffer_idx + to_read*size > WRITE_BUFFER_SIZE)

    to_read = (WRITE_BUFFER_SIZE - buffer_idx)/size;

  if (to_read > 0) {

    memcpy(buffer+buffer_idx, buf_in, to_read*size);

    buffer_idx += to_read*size;

  }

  return to_read;

}

static size_t write_ignore(void *buffer, size_t size, size_t nmemb,

    void *userp) {

  return nmemb;

}

/*

 * do_q_user_perm

 *

 * Makes an HTTP request to the CRM server to see what tools user_id has access

 * to. Returns a bitmask, and returns 0 if there was a problem.

 */

unsigned int do_q_user_perm(unsigned int user_id) {

  CURL* handle;

  CURLcode error_code;

  char url[1024];

  int result = 0;

  long response = 0;

  handle = curl_easy_init();

  if (handle == NULL)

    return 0;

  sprintf(url, "http://%s/crm/roboauth/%08x/", server, user_id);

  error_code = curl_easy_setopt(handle, CURLOPT_URL, url);

  if (error_code) goto error;

  buffer_idx = 0;

  error_code = curl_easy_setopt(handle, CURLOPT_WRITEFUNCTION, write_buffer);

  if (error_code) goto error;

  error_code = curl_easy_setopt(handle, CURLOPT_WRITEDATA, NULL);

  if (error_code) goto error;

  error_code = curl_easy_perform(handle);

  if (error_code) goto error;

  error_code = curl_easy_getinfo(handle, CURLINFO_RESPONSE_CODE, &response);

  if (error_code) goto error;

  if (response >= 400)

    log_print("ERROR: response %ld from %s", response, url);

  else if (response > 200)

    log_print("WARNING: response %ld from %s", response, url);

  result = atoi(buffer);

  cache_update(user_id, result);

  curl_easy_cleanup(handle);

  return result;

error:

  log_print("ERROR: curl: %s", curl_easy_strerror(error_code));

  log_print("ERROR:       when authenticating user %08x", user_id);

  curl_easy_cleanup(handle);

  return 0;

}

void do_refresh(unsigned int key) {

  do_q_user_perm(key);

}

/*

 * query_refresh_cache

 *

 * Queries the CRM server to update every user permission entry in the cache.

 */

void query_refresh_cache() {

  cache_foreach(do_refresh);

}

/*

 * query_user_permission

 *

 * Checks whether user_id has permission for tool_id. First checks the cache,

 * then if that fails, calls do_q_user_perm to make an HTTP request to the CRM

 * server.

 */

int query_user_permission(int tool_id, unsigned int user_id) {

  unsigned int result;

  if (cache_lookup(user_id, &result))

    log_print("Serving permissions for %08x from cache", user_id);

  else {

    log_print("Requesting permissions for %08x from server", user_id);

    result = do_q_user_perm(user_id);

  }

  return (result >> tool_id) & 1;

}

/*

 * query_add_event

 *

 * Makes an HTTPS POST request to add an event to the CRM server, including

 * user, tool, start time, and stop time. Reads the password from password.txt.

 * Returns 0 if successful, or 1 if there was an error and the caller should

 * try the same event again later.

 *

 * Times are represented as strftime's "%F %T", which is like "YYYY-MM-DD

 * HH:MM:SS" with 24-hour time

 */

int query_add_event(struct event_t *event) {

  CURL* handle;

  CURLcode error_code;

  struct curl_httppost *formpost = NULL, *lastptr = NULL;

  char buf[1024];

  struct tm *timeinfo;

  long response = 0;

#ifdef DEBUG_EVENT_RESPONSE

  FILE *fdebug;

  fdebug = fopen("debug.html", "w");

#endif

  handle = curl_easy_init();

  if (handle == NULL)

    return 1;

  curl_formadd(&formpost, &lastptr,

      CURLFORM_COPYNAME, "username",

      CURLFORM_COPYCONTENTS, "tooltron",

      CURLFORM_END);

  curl_formadd(&formpost, &lastptr,

      CURLFORM_COPYNAME, "password",

      CURLFORM_COPYCONTENTS, tooltron_password,

      CURLFORM_END);

  timeinfo = localtime(&event->tstart);

  strftime(buf, sizeof(buf), "%F %T", timeinfo);

  curl_formadd(&formpost, &lastptr,

      CURLFORM_COPYNAME, "tstart",

      CURLFORM_COPYCONTENTS, buf,

      CURLFORM_END);

  timeinfo = localtime(&event->tend);

  strftime(buf, sizeof(buf), "%F %T", timeinfo);

  curl_formadd(&formpost, &lastptr,

      CURLFORM_COPYNAME, "tend",

      CURLFORM_COPYCONTENTS, buf,

      CURLFORM_END);

  sprintf(buf, "%08x", event->user);

  curl_formadd(&formpost, &lastptr,

      CURLFORM_COPYNAME, "user_id",

      CURLFORM_COPYCONTENTS, buf,

      CURLFORM_END);

  sprintf(buf, "%d", event->tool_id);

  curl_formadd(&formpost, &lastptr,

      CURLFORM_COPYNAME, "machine_id",

      CURLFORM_COPYCONTENTS, buf,

      CURLFORM_END);

  curl_formadd(&formpost, &lastptr,

      CURLFORM_COPYNAME, "succ",

      CURLFORM_COPYCONTENTS, event->succ? "1" : "0",

      CURLFORM_END);

  sprintf(buf, "https://%s/crm/add_card_event/", server);

  //sprintf(buf, "http://%s/crm/add_card_event/", server);

  error_code = curl_easy_setopt(handle, CURLOPT_URL, buf);

  if (error_code) goto error;

  /* TODO disabling host and peer verification should theoretically be removed

   * eventually */

  error_code = curl_easy_setopt(handle, CURLOPT_SSL_VERIFYHOST, 0L);

  if (error_code) goto error;

  error_code = curl_easy_setopt(handle, CURLOPT_SSL_VERIFYPEER, 0L);

  if (error_code) goto error;

#ifdef DEBUG_EVENT_RESPONSE

  error_code = curl_easy_setopt(handle, CURLOPT_WRITEDATA, fdebug);

#else

  error_code = curl_easy_setopt(handle, CURLOPT_WRITEFUNCTION, write_ignore);

#endif

  if (error_code) goto error;

  error_code = curl_easy_setopt(handle, CURLOPT_HTTPPOST, formpost);

  if (error_code) goto error;

  error_code = curl_easy_perform(handle);

  if (error_code) goto error;

  error_code = curl_easy_getinfo(handle, CURLINFO_RESPONSE_CODE, &response);

  if (error_code) goto error;

  if (response >= 400)

    log_print("ERROR: response %ld from %s", response, buf);

  else if (response > 200)

    log_print("WARNING: response %ld from %s", response, buf);

  curl_easy_cleanup(handle);

  curl_formfree(formpost);

#ifdef DEBUG_EVENT_RESPONSE

  fclose(fdebug);

#endif

  return response >= 300;

error:

  log_print("ERROR: curl: %s", curl_easy_strerror(error_code));

  curl_easy_cleanup(handle);

  curl_formfree(formpost);

#ifdef DEBUG_EVENT_RESPONSE

  fclose(fdebug);

#endif

  return 1;

}