QEMU-Nyx-fork/nyx/sharedir.c

#include "sharedir.h"
#include <assert.h>
#include <stdio.h>
#include <dirent.h>
#include <stdbool.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>

//#define SHAREDIR_DEBUG

sharedir_t* sharedir_new(void){
	sharedir_t* self = malloc(sizeof(sharedir_t));
  self->dir = NULL;
  self->lookup = kh_init(SHAREDIR_LOOKUP);
  self->last_file_f = NULL;
  self->last_file_obj_ptr = NULL;
  return self;
}

void sharedir_set_dir(sharedir_t* self, const char* dir){
  assert(!self->dir);
  assert(asprintf(&self->dir, "%s", dir) != -1);
}

static bool file_exits(const char* file){
	struct stat sb;   
	return (stat (file, &sb) == 0);
}

static time_t get_file_mod_time(char *file){
    struct stat attr;
    stat(file, &attr);
    return attr.st_mtime;
}

static size_t get_file_size(const char* file){
  struct stat st;
  stat(file, &st);
  return st.st_size;
}

static char* sharedir_scan(sharedir_t* self, const char* file){

  /*
   * Agent is not under our control, but lets roughly constrain
   * it to anything stored in or linked from sharedir
   */
  chdir(self->dir);
  char* real_path = realpath(file, NULL);

  if (file[0] != '/' && !strstr(file, "/../") &&
      real_path && file_exits(real_path)) {
    return real_path;
  }

  free(real_path);
  return NULL;
}

static sharedir_file_t* sharedir_get_object(sharedir_t* self, const char* file){
  khiter_t k;
  int ret;
  sharedir_file_t* obj = NULL;

  k = kh_get(SHAREDIR_LOOKUP, self->lookup, file); 

  if(k != kh_end(self->lookup)){
    /* file already exists in our hash map */
    obj = kh_value(self->lookup, k);

    /* check if file still exists */
    assert(file_exits(obj->path));

    /* check if mod time matches */
    assert(get_file_mod_time(obj->path) == obj->mod_time);

    /* check if file size matches */
    assert(get_file_size(obj->path) == obj->size);

    return obj;
  }
  else{
    /* nope ! */
    char* realpath = sharedir_scan(self, file);
    struct stat sb;
    if(realpath != NULL){
      if (stat(realpath, &sb) == 0 && S_ISDIR(sb.st_mode)){
        return NULL; // is dir
      }
      obj = malloc(sizeof(sharedir_file_t));
      memset(obj, 0x0, sizeof(sharedir_file_t));
      assert(asprintf(&obj->file, "%s", basename(realpath)) != -1);
      obj->path = realpath;
      obj->size = get_file_size(obj->path);
      obj->bytes_left = (uint64_t) obj->size;
      obj->mod_time = get_file_mod_time(obj->path);

      /* put into hash_list */

      char* new_file = NULL;
      assert(asprintf(&new_file, "%s", file) != -1);
      k = kh_put(SHAREDIR_LOOKUP, self->lookup, new_file, &ret);
      kh_value(self->lookup, k) = obj;

      return obj;
    }

    /* file not found */
    return NULL;
  }
}

static FILE* get_file_ptr(sharedir_t* self, sharedir_file_t* obj){
  if(obj == self->last_file_obj_ptr && self->last_file_f){
    return self->last_file_f;
  }

  if(self->last_file_f){
	  fclose(self->last_file_f);
  }

  FILE* f = fopen(obj->path, "r");
  self->last_file_f = f;
  self->last_file_obj_ptr = obj;
  return f;
}

uint64_t sharedir_request_file(sharedir_t* self, const char* file, uint8_t* page_buffer){
  if(!self->dir){
    fprintf(stderr, "WARNING: New file request received, but no share dir configured! [FILE: %s]\n", file);
    return 0xFFFFFFFFFFFFFFFFUL;
  }

  FILE* f = NULL;

  sharedir_file_t* obj = sharedir_get_object(self, file);
  if(obj != NULL){
#ifdef SHAREDIR_DEBUG
    printf("sharedir_get_object->file: %s\n", obj->file);
    printf("sharedir_get_object->path: %s\n", obj->path);
    printf("sharedir_get_object->size: %ld\n", obj->size);
    printf("sharedir_get_object->bytes_left: %ld\n", obj->bytes_left);
#endif
    if(obj->bytes_left >= 0x1000){
      f = get_file_ptr(self, obj);
      fseek(f, obj->size-obj->bytes_left, SEEK_SET);
      assert(fread(page_buffer, 1, 0x1000, f) == 0x1000);
      obj->bytes_left -= 0x1000;
      return 0x1000;
    }
    else {
      if (obj->bytes_left != 0){
        f = get_file_ptr(self, obj);
        fseek(f, obj->size-obj->bytes_left, SEEK_SET);
        assert(fread(page_buffer, 1, obj->bytes_left, f) == obj->bytes_left);

        uint64_t ret_value = obj->bytes_left;
        obj->bytes_left = 0;

        return ret_value;
      }
      else {
        obj->bytes_left = (uint_fast64_t)obj->size;
        return 0;
      }
    }
  }
  else{
	  fprintf(stderr, "WARNING: No such file in sharedir: %s\n", file);
    return 0xFFFFFFFFFFFFFFFFUL;
  }
}