main.c

#include <stdio.h>
#include <unistd.h>
#include <argp.h>

#include "util.h"
#include "board.h"
#include "cpu.h"
#include "fan.h"
#include "signal.h"
#include "config.h"

#define MIN(a,b) (((a)<(b))?(a):(b))
#define MAX(a,b) (((a)>(b))?(a):(b))
#define CLAMP(x, upper, lower) (MIN(upper, MAX(x, lower)))

const char* argp_program_version = "ubnt-fan-speed 0.0.1";
const char* argp_program_bug_address = "leandre.gohy@hexeo.be";
static struct argp_option options[] = {
  { "config", 'c', "path", 0, "Config file path", 0 },
  { 0 }
};

static error_t parse_opt(int key, char* arg,struct argp_state* state) {
  config_t* config = state->input;
  switch (key) {
    case 'c':
      config->path = arg;
      break;
    default:
      return ARGP_ERR_UNKNOWN;
  }
  return 0;
}

static struct argp argp = { options, parse_opt, 0, 0, 0, 0, 0 };

config_t g_config;
ubnt_board_t g_board;

int main(int argc, char** argv) {
  short fan_speed;
  unsigned char max_cpu_id;
  unsigned int temperature;

  argp_parse(&argp, argc, argv, 0, 0, &g_config);
  signal_init();

  read_config(&g_config);
  board_init(&g_board, &g_config);

  printf("ubnt-fan-speed: Detected board id: 0x%04x\n", g_board.id);
  if (g_board.id == 0) {
    fprintf(stderr, "ubnt-fan-speed: Unsupported board\n");
    return 1;
  }

  printf("ubnt-fan-speed: Starting...\n");

  max_cpu_id = 3;
  fan_speed = g_board.minimum_fan_speed;
  g_board.set_online_cpus(max_cpu_id);
  g_board.set_fan_mode(FAN_MODE_MANUAL);
  g_board.set_fan_speed(fan_speed);

  while (1) {
    g_board.set_fan_mode(FAN_MODE_MANUAL);

    temperature = g_board.get_temperature();

    if (temperature >= g_board.fan_critical_threshold && fan_speed < 255) {
      fan_speed = 255;
      g_board.set_fan_speed(fan_speed);
    }
    else if (temperature >= g_board.fan_high_threshold && fan_speed < 255) {
      fan_speed = CLAMP(fan_speed + 5, g_board.minimum_fan_speed, 255);
      g_board.set_fan_speed(fan_speed);
    }
    else if (temperature <= g_board.fan_low_threshold && fan_speed > g_board.minimum_fan_speed) {
      fan_speed = CLAMP(fan_speed - 5, g_board.minimum_fan_speed, 255);
      g_board.set_fan_speed(fan_speed);
    }

    if (temperature <= g_board.cpu_low_threshold && max_cpu_id < 3) {
      max_cpu_id = 3;
      g_board.set_online_cpus(max_cpu_id);
    }
    else if (temperature >= g_board.cpu_medium_threshold && max_cpu_id > 2) {
      max_cpu_id = 2;
      g_board.set_online_cpus(max_cpu_id);
    }
    else if (temperature >= g_board.cpu_high_threshold && max_cpu_id > 1) {
      max_cpu_id = 1;
      g_board.set_online_cpus(max_cpu_id);
    }
    else if (temperature >= g_board.cpu_critical_threshold && max_cpu_id > 0) {
      max_cpu_id = 0;
      g_board.set_online_cpus(max_cpu_id);
    }

    sleep(1);
  }

  return 1;
}