/*******************************************************
 Copyright (C) 2018-2021 Georges Da Costa <georges.da-costa@irit.fr>

    This file is part of Mojitos.

    Mojitos is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    Mojitos is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with MojitO/S.  If not, see <https://www.gnu.org/licenses/>.

 *******************************************************/
#include <stdlib.h>
#include <fcntl.h>
#include <stdio.h>
#include <string.h>

#include <glob.h>

#include <stdint.h>

struct network_t {
  uint64_t values[4];
  uint64_t tmp_values[4];
  int sources[4];
};
unsigned int _get_network(uint64_t* results, int* sources);


unsigned int init_infiniband(char* infi_path, void**ptr) {
  if(infi_path==NULL)
    return 0;

  if(strcmp(infi_path,"X")==0) {

    glob_t res;
  
    glob("/sys/class/infiniband/*/ports/*/counters/", 0, NULL, &res);
    if(res.gl_pathc == 0)
      return 0;
    infi_path = res.gl_pathv[0];
  }
  
  char *filenames[] = {"%s/port_rcv_packets",
		       "%s/port_rcv_data",
		       "%s/port_xmit_packets",
		       "%s/port_xmit_data"};

  struct network_t *state = malloc(sizeof(struct network_t));

  char buffer[1024];
  for(int i=0; i<4; i++) {
    sprintf(buffer, filenames[i], infi_path);
    state->sources[i] = open(buffer, O_RDONLY);
  }
  
  *ptr = (void*) state;
  _get_network(state->values, state->sources);

  return 4;
}