/******************************************************************************/
/* */
/* Copyright (c) International Business Machines Corp., 2007 */
/* */
/* This program 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 2 of the License, or */
/* (at your option) any later version. */
/* */
/* This program 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 this program; if not, write to the Free Software */
/* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */
/* */
/******************************************************************************/
/******************************************************************************/
/* */
/* File: libcontrollers.c */
/* */
/* Description: This file contains the definitions for the functions/apis */
/* for the controllers library. This library is used by the */
/* controllers testcases. */
/* */
/* Author: Sudhir Kumar skumar@linux.vnet.ibm.com */
/* */
/* History: */
/* Created- 15/02/2008 -Sudhir Kumar <skumar@linux.vnet.ibm.com> */
/* */
/******************************************************************************/
#include "libcontrollers.h"
/*
* Function: scan_shares_file()
* This function scans all the shares files under the mountpoint
* of the controller and returns the total added shares of all
* the groups (currently excludes default group) ??
*/
int scan_shares_files(unsigned int *shares_pointer)
{
struct stat statbuffer;
DIR *dp;
char *path_pointer;
/*
* Check if we can get stat of the file
*/
if (lstat(fullpath, &statbuffer) < 0) {
error_function("Can not read stat for file ", fullpath);
return -1;
}
if (S_ISDIR(statbuffer.st_mode) == 0) { /* not a directory */
/*
* We run all user tasks in the created groups and not default groups. So
* exclude the shares of default group. FLAG to ensure dir_pointer is non NULL
*/
if ((FLAG == 1)
&& (strcmp(fullpath, "/dev/cpuctl/cpu.shares") != 0)
&& (strcmp(dir_pointer->d_name, "cpu.shares") == 0)) {
*shares_pointer += read_shares_file(fullpath);
}
return 0;
}
/*
* Now it's a directory. let the path_pointer point to the end
* of fullpath to append new files names
*/
path_pointer = fullpath + strlen(fullpath);
*path_pointer++ = '/';
*path_pointer = 0;
if ((dp = opendir(fullpath)) == NULL) { /* Error in opening directory */
error_function("Can't open ", fullpath);
return -1;
}
/*
* search all groups recursively and get total shares
*/
while ((dir_pointer = readdir(dp)) != NULL) { /* Error in reading directory */
if ((strcmp(dir_pointer->d_name, ".") == 0)
|| (strcmp(dir_pointer->d_name, "..") == 0))
continue; /* ignore current and parent directory */
FLAG = 1;
strcpy(path_pointer, dir_pointer->d_name); /* append name to fullpath */
if ((retval = scan_shares_files(shares_pointer)) != 0)
break;
}
/*
* This directory is searched fully. let us go back to parent directory again
*/
path_pointer[-1] = 0;
if (closedir(dp) < 0) {
error_function("Could not close dir ", fullpath);
return -1;
}
return 0;
}
/*
* Function: read_file ()
* This function is written keeping in mind the support
* to read other files also if required in future.
* Each file under a group contains some diff parameter/s
*/
int read_file(char *filepath, int action, unsigned int *value)
{
int num_line = 0;
FILE *fp;
int tmp;
switch (action) {
case GET_SHARES:
tmp = read_shares_file(filepath);
if (tmp == -1)
return -1;
*value = (unsigned int)tmp;
break;
case GET_TASKS:
fp = fopen(filepath, "r");
if (fp == NULL) {
error_function("Could not open file", filepath);
return -1;
}
while (fgets(target, LINE_MAX, fp) != NULL)
num_line++;
*value = (unsigned int)num_line;
if (fclose(fp)) {
error_function("Could not close file", filepath);
return -1;
}
break;
default:
error_function("Wrong action type passed to fun read_file for ",
filepath);
return -1;
}
return 0;
}
/*
* Function: error_function()
* Prints error message and returns -1
*/
static inline void error_function(char *msg1, char *msg2)
{
fprintf(stdout, "ERROR: %s ", msg1);
fprintf(stdout, "%s\n", msg2);
}
/* Function: read_shares_file()
* Reads shares value from a given shares file and writes them to
* the given pointer location. Returns 0 if success
*/
int read_shares_file(char *filepath)
{
FILE *fp;
unsigned int shares;
fp = fopen(filepath, "r");
if (fp == NULL) {
error_function("Could not open file", filepath);
return -1;
}
fscanf(fp, "%u", &shares);
if (fclose(fp)) {
error_function("Could not close file", filepath);
return -1;
}
return shares;
}
/* Function: write_to_file()
* writes value to shares file or pid to tasks file
*/
int write_to_file(char *file, const char *mode, unsigned int value)
{
FILE *fp;
fp = fopen(file, mode);
if (fp == NULL) {
error_function("in opening file for writing:", file);
return -1;
}
fprintf(fp, "%u\n", value);
fclose(fp);
return 0;
}
/* Function: signal_handler_alarm()
* signal handler for the new action
*/
void signal_handler_alarm(int signal)
{
timer_expired = 1;
}