qres-reserve.c

Go to the documentation of this file.

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>

#include <aquosa/qres_lib.h>
#include <linux/aquosa/qos_debug.h>

extern char **environ;

static inline void qos_assert(int cond, const char *msg) {
  if (! cond) {
    printf("%s\n", msg);
    exit(-1);
  }
}

int main(int argc, char *argv[])
{
  int Q_min = -1, Q = -1, P = -1;
  qos_bw_t bw_min = -1, bw = -1;
  int destroy = 0;
  pid_t pid = -1;
  tid_t tid = -1;
  qres_params_t params;
  qos_rv rv;
  qres_sid_t sid = -1;
  int get_sid = 0;

  argc--;  argv++;
  while (argc > 0) {
    if (strcmp(*argv, "-q") == 0) {
      qos_assert(argc > 1, "Missing option parameter");
      argc--;  argv++;
      Q_min = atoi(*argv);
    } else if (strcmp(*argv, "-Q") == 0) {
      qos_assert(argc > 1, "Missing option parameter");
      argc--;  argv++;
      Q = atoi(*argv);
    } else if (strcmp(*argv, "-P") == 0) {
      qos_assert(argc > 1, "Missing option parameter");
      argc--;  argv++;
      P = atoi(*argv);
    } else if (strcmp(*argv, "-b") == 0) {
      qos_assert(argc > 1, "Missing option parameter");
      argc--;  argv++;
      bw_min = d2bw(atof(*argv));
    } else if (strcmp(*argv, "-B") == 0) {
      qos_assert(argc > 1, "Missing option parameter");
      argc--;  argv++;
      bw = d2bw(atof(*argv));
    } else if (strcmp(*argv, "-p") == 0) {
      qos_assert(argc > 1, "Missing option parameter");
      argc--;  argv++;
      pid = atoi(*argv);
    } else if (strcmp(*argv, "-t") == 0) {
      qos_assert(argc > 1, "Missing option parameter");
      argc--;  argv++;
      tid = atoi(*argv);
    } else if (strcmp(*argv, "-s") == 0) {
      qos_assert(argc > 1, "Missing option parameter");
      argc--;  argv++;
      sid = atoi(*argv);
    } else if (strcmp(*argv, "-d") == 0) {
      destroy = 1;
    } else if (strcmp(*argv, "-g") == 0) {
      get_sid = 1;
    } else if (strcmp(*argv, "-h") == 0 || strcmp(*argv, "--help") == 0) {
      printf("Usage: %s [ options... ]\n", argv[0]);
      printf("  -q Q_min  : set minimum guaranteed budget per server period\n");
      printf("  -Q Q_max  : set required budget per server period\n");
      printf("  -P P      : set server period\n");
      printf("  -b P      : set minimum required bandwidth\n");
      printf("  -B P      : set maximum bandwidth\n");
      printf("  -p pid    : set process id to affect\n");
      printf("  -t tid    : set thread id to affect or 0 for default\n");
      printf("  -s sid    : set id of existing server to affect\n");
      printf("  -g        : get server info by sid or pid/tid\n");
      exit(0);
    } else {
      printf("Unknown option: %s\n", *argv);
      exit(-1);
    }

    argc--;  argv++;
  }

  qos_chk_ok_exit(qres_init());

  if (tid == -1 && pid != -1)
    tid = pid;
  else if (pid == -1 && tid != -1)
    pid = tid;

  if (sid == -1 && pid != -1 && tid != -1)
    qres_get_sid(pid, tid, &sid);

  if ((destroy == 1) && (sid != -1)) {
    //    qos_chk_ok_exit(qres_detach_from_server(pid));
    qos_chk_ok_exit(qres_destroy_server(sid));
    qos_chk_ok_exit(qres_cleanup());
    exit(0);
  }

  if (Q != -1 && P != -1) {
    params.Q = Q;
    params.P = P;
  } else if (bw != -1 && P != -1) {
    params.Q = bw2Q(bw, P);
    params.P = P;
  } else if (sid != -1) {
    if (pid != -1) {
      rv = qres_attach_thread(sid, pid, tid);
      qos_chk_ok_exit(rv);
    }
  }

  if (Q_min != -1)
    params.Q_min = Q_min;
  else if (bw_min != -1)
    params.Q_min = bw2Q(bw_min, params.P);
  else
    params.Q_min = 0;

  params.flags = 0;

  if (get_sid) {
    qos_chk_ok_exit(qres_get_params(sid, &params));
    printf("sid: %d, Q: " QRES_TIME_FMT
             ", q: " QRES_TIME_FMT
             ", P: " QRES_TIME_FMT
             "\n", sid, params.Q, params.Q_min, params.P);
    exit(0);
  }

  qos_chk_ok_exit(qres_create_server(&params, &sid));
  qos_chk_ok_exit(qres_attach_thread(sid, pid, tid));
  if (rv == QOS_E_INCONSISTENT_STATE) {
    /* Maybe already has a server: try changing params */
    rv = qres_set_params(sid, &params);
    qos_chk_ok_exit(rv);
  } else
    qos_chk_ok_exit(rv);

  qos_chk_ok_exit(qres_cleanup());

  return 0;
}

---