1
1
openmpi/orte/dss/dss_arith.c

749 строки
19 KiB
C
Исполняемый файл

/*
* Copyright (c) 2004-2005 The Trustees of Indiana University.
* All rights reserved.
* Copyright (c) 2004-2005 The Trustees of the University of Tennessee.
* All rights reserved.
* Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
* University of Stuttgart. All rights reserved.
* Copyright (c) 2004-2005 The Regents of the University of California.
* All rights reserved.
* $COPYRIGHT$
*
* Additional copyrights may follow
*
* $HEADER$
*/
#include "orte_config.h"
#include <sys/types.h>
#if HAVE_NETINET_IN_H
#include <netinet/in.h>
#endif
#include "orte/mca/errmgr/errmgr.h"
#include "orte/dss/dss_internal.h"
#include "opal/util/output.h"
static void orte_dss_arith_int(int *value, int *operand, orte_dss_arith_op_t operation);
static void orte_dss_arith_uint(uint *value, uint *operand, orte_dss_arith_op_t operation);
static void orte_dss_arith_size(size_t *value, size_t *operand, orte_dss_arith_op_t operation);
static void orte_dss_arith_pid(pid_t *value, pid_t *operand, orte_dss_arith_op_t operation);
static void orte_dss_arith_byte(uint8_t *value, uint8_t *operand, orte_dss_arith_op_t operation);
static void orte_dss_arith_int8(int8_t *value, int8_t *operand, orte_dss_arith_op_t operation);
static void orte_dss_arith_int16(int16_t *value, int16_t *operand, orte_dss_arith_op_t operation);
static void orte_dss_arith_uint16(uint16_t *value, uint16_t *operand, orte_dss_arith_op_t operation);
static void orte_dss_arith_int32(int32_t *value, int32_t *operand, orte_dss_arith_op_t operation);
static void orte_dss_arith_uint32(uint32_t *value, uint32_t *operand, orte_dss_arith_op_t operation);
static void orte_dss_arith_int64(int64_t *value, int64_t *operand, orte_dss_arith_op_t operation);
static void orte_dss_arith_uint64(uint64_t *value, uint64_t *operand, orte_dss_arith_op_t operation);
/* some weird ones - but somebody *might* want to do it, I suppose... */
static void orte_dss_arith_data_type(orte_data_type_t *value, orte_data_type_t *operand, orte_dss_arith_op_t operation);
static void orte_dss_arith_daemon_cmd(orte_daemon_cmd_flag_t *value, orte_daemon_cmd_flag_t *operand, orte_dss_arith_op_t operation);
int orte_dss_arith(orte_data_value_t *value, void *operand, orte_dss_arith_op_t operation, orte_data_type_t type)
{
/* check for error */
if (NULL == value || NULL == operand) {
ORTE_ERROR_LOG(ORTE_ERR_BAD_PARAM);
return ORTE_ERR_BAD_PARAM;
}
if (type != value->type) {
ORTE_ERROR_LOG(ORTE_ERR_TYPE_MISMATCH);
return ORTE_ERR_TYPE_MISMATCH;
}
/* Lookup the arith function for this type and call it */
switch(type) {
case ORTE_INT:
orte_dss_arith_int(value->data, operand, operation);
break;
case ORTE_UINT:
orte_dss_arith_uint(value->data, operand, operation);
break;
case ORTE_SIZE:
orte_dss_arith_size(value->data, operand, operation);
break;
case ORTE_PID:
orte_dss_arith_pid(value->data, operand, operation);
break;
case ORTE_BYTE:
case ORTE_UINT8:
orte_dss_arith_byte(value->data, operand, operation);
break;
case ORTE_INT8:
orte_dss_arith_int8(value->data, operand, operation);
break;
case ORTE_INT16:
orte_dss_arith_int16(value->data, operand, operation);
break;
case ORTE_UINT16:
orte_dss_arith_uint16(value->data, operand, operation);
break;
case ORTE_INT32:
orte_dss_arith_int32(value->data, operand, operation);
break;
case ORTE_UINT32:
orte_dss_arith_uint32(value->data, operand, operation);
break;
case ORTE_INT64:
orte_dss_arith_int64(value->data, operand, operation);
break;
case ORTE_UINT64:
orte_dss_arith_uint64(value->data, operand, operation);
break;
default:
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return ORTE_ERR_OPERATION_UNSUPPORTED;
}
return ORTE_SUCCESS;
}
int orte_dss_increment(orte_data_value_t *value)
{
int one;
unsigned int uone;
size_t sone;
pid_t pone;
uint8_t u8one;
int8_t i8one;
uint16_t u16one;
int16_t i16one;
uint32_t u32one;
int32_t i32one;
uint64_t u64one;
int64_t i64one;
orte_daemon_cmd_flag_t daemoncmdone;
orte_data_type_t datatypeone;
/* check for error */
if (NULL == value) {
ORTE_ERROR_LOG(ORTE_ERR_BAD_PARAM);
return ORTE_ERR_BAD_PARAM;
}
/* Lookup the arith function for this type and call it */
switch(value->type) {
case ORTE_INT:
one = 1;
orte_dss_arith_int(value->data, &one, ORTE_DSS_ADD);
break;
case ORTE_UINT:
uone = 1;
orte_dss_arith_uint(value->data, &uone, ORTE_DSS_ADD);
break;
case ORTE_SIZE:
sone = 1;
orte_dss_arith_size(value->data, &sone, ORTE_DSS_ADD);
break;
case ORTE_PID:
pone = 1;
orte_dss_arith_pid(value->data, &pone, ORTE_DSS_ADD);
break;
case ORTE_BYTE:
case ORTE_UINT8:
u8one = 1;
orte_dss_arith_byte(value->data, &u8one, ORTE_DSS_ADD);
break;
case ORTE_INT8:
i8one = 1;
orte_dss_arith_int8(value->data, &i8one, ORTE_DSS_ADD);
break;
case ORTE_INT16:
i16one = 1;
orte_dss_arith_int16(value->data, &i16one, ORTE_DSS_ADD);
break;
case ORTE_UINT16:
u16one = 1;
orte_dss_arith_uint16(value->data, &u16one, ORTE_DSS_ADD);
break;
case ORTE_INT32:
i32one = 1;
orte_dss_arith_int32(value->data, &i32one, ORTE_DSS_ADD);
break;
case ORTE_UINT32:
u32one = 1;
orte_dss_arith_uint32(value->data, &u32one, ORTE_DSS_ADD);
break;
case ORTE_INT64:
i64one = 1;
orte_dss_arith_int64(value->data, &i64one, ORTE_DSS_ADD);
break;
case ORTE_UINT64:
u64one = 1;
orte_dss_arith_uint64(value->data, &u64one, ORTE_DSS_ADD);
break;
case ORTE_DAEMON_CMD:
daemoncmdone = 1;
orte_dss_arith_daemon_cmd(value->data, &daemoncmdone, ORTE_DSS_ADD);
break;
case ORTE_DATA_TYPE:
datatypeone = 1;
orte_dss_arith_data_type(value->data, &datatypeone, ORTE_DSS_ADD);
break;
default:
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return ORTE_ERR_OPERATION_UNSUPPORTED;
}
return ORTE_SUCCESS;
}
int orte_dss_decrement(orte_data_value_t *value)
{
int one;
unsigned int uone;
size_t sone;
pid_t pone;
uint8_t u8one;
int8_t i8one;
uint16_t u16one;
int16_t i16one;
uint32_t u32one;
int32_t i32one;
uint64_t u64one;
int64_t i64one;
orte_daemon_cmd_flag_t daemoncmdone;
orte_data_type_t datatypeone;
/* check for error */
if (NULL == value) {
ORTE_ERROR_LOG(ORTE_ERR_BAD_PARAM);
return ORTE_ERR_BAD_PARAM;
}
/* Lookup the arith function for this type and call it */
switch(value->type) {
case ORTE_INT:
one = 1;
orte_dss_arith_int(value->data, &one, ORTE_DSS_SUB);
break;
case ORTE_UINT:
uone = 1;
orte_dss_arith_uint(value->data, &uone, ORTE_DSS_SUB);
break;
case ORTE_SIZE:
sone = 1;
orte_dss_arith_size(value->data, &sone, ORTE_DSS_SUB);
break;
case ORTE_PID:
pone = 1;
orte_dss_arith_pid(value->data, &pone, ORTE_DSS_SUB);
break;
case ORTE_BYTE:
case ORTE_UINT8:
u8one = 1;
orte_dss_arith_byte(value->data, &u8one, ORTE_DSS_SUB);
break;
case ORTE_INT8:
i8one = 1;
orte_dss_arith_int8(value->data, &i8one, ORTE_DSS_SUB);
break;
case ORTE_INT16:
i16one = 1;
orte_dss_arith_int16(value->data, &i16one, ORTE_DSS_SUB);
break;
case ORTE_UINT16:
u16one = 1;
orte_dss_arith_uint16(value->data, &u16one, ORTE_DSS_SUB);
break;
case ORTE_INT32:
i32one = 1;
orte_dss_arith_int32(value->data, &i32one, ORTE_DSS_SUB);
break;
case ORTE_UINT32:
u32one = 1;
orte_dss_arith_uint32(value->data, &u32one, ORTE_DSS_SUB);
break;
case ORTE_INT64:
i64one = 1;
orte_dss_arith_int64(value->data, &i64one, ORTE_DSS_SUB);
break;
case ORTE_UINT64:
u64one = 1;
orte_dss_arith_uint64(value->data, &u64one, ORTE_DSS_SUB);
break;
case ORTE_DAEMON_CMD:
daemoncmdone = 1;
orte_dss_arith_daemon_cmd(value->data, &daemoncmdone, ORTE_DSS_SUB);
break;
case ORTE_DATA_TYPE:
datatypeone = 1;
orte_dss_arith_data_type(value->data, &datatypeone, ORTE_DSS_SUB);
break;
default:
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return ORTE_ERR_OPERATION_UNSUPPORTED;
}
return ORTE_SUCCESS;
}
/*
* NUMERIC arith FUNCTIONS
*/
static void orte_dss_arith_int(int *value, int *operand, orte_dss_arith_op_t operation)
{
switch(operation) {
case ORTE_DSS_ADD:
(*value) += *operand;
return;
case ORTE_DSS_SUB:
(*value) -= *operand;
return;
case ORTE_DSS_MUL:
(*value) *= *operand;
return;
case ORTE_DSS_DIV:
if (0 == *operand) {
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
(*value) /= *operand;
return;
default:
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
return;
}
static void orte_dss_arith_uint(uint *value, uint *operand, orte_dss_arith_op_t operation)
{
switch(operation) {
case ORTE_DSS_ADD:
(*value) += *operand;
return;
case ORTE_DSS_SUB:
(*value) -= *operand;
return;
case ORTE_DSS_MUL:
(*value) *= *operand;
return;
case ORTE_DSS_DIV:
if (0 == *operand) {
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
(*value) /= *operand;
return;
default:
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
return;
}
static void orte_dss_arith_size(size_t *value, size_t *operand, orte_dss_arith_op_t operation)
{
switch(operation) {
case ORTE_DSS_ADD:
(*value) += *operand;
return;
case ORTE_DSS_SUB:
(*value) -= *operand;
return;
case ORTE_DSS_MUL:
(*value) *= *operand;
return;
case ORTE_DSS_DIV:
if (0 == *operand) {
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
(*value) /= *operand;
return;
default:
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
return;
}
static void orte_dss_arith_pid(pid_t *value, pid_t *operand, orte_dss_arith_op_t operation)
{
switch(operation) {
case ORTE_DSS_ADD:
(*value) += *operand;
return;
case ORTE_DSS_SUB:
(*value) -= *operand;
return;
case ORTE_DSS_MUL:
(*value) *= *operand;
return;
case ORTE_DSS_DIV:
if (0 == *operand) {
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
(*value) /= *operand;
return;
default:
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
return;
}
static void orte_dss_arith_byte(uint8_t *value, uint8_t *operand, orte_dss_arith_op_t operation)
{
switch(operation) {
case ORTE_DSS_ADD:
(*value) += *operand;
return;
case ORTE_DSS_SUB:
(*value) -= *operand;
return;
case ORTE_DSS_MUL:
(*value) *= *operand;
return;
case ORTE_DSS_DIV:
if (0 == *operand) {
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
(*value) /= *operand;
return;
default:
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
return;
}
static void orte_dss_arith_int8(int8_t *value, int8_t *operand, orte_dss_arith_op_t operation)
{
switch(operation) {
case ORTE_DSS_ADD:
(*value) += *operand;
return;
case ORTE_DSS_SUB:
(*value) -= *operand;
return;
case ORTE_DSS_MUL:
(*value) *= *operand;
return;
case ORTE_DSS_DIV:
if (0 == *operand) {
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
(*value) /= *operand;
return;
default:
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
return;
}
static void orte_dss_arith_int16(int16_t *value, int16_t *operand, orte_dss_arith_op_t operation)
{
switch(operation) {
case ORTE_DSS_ADD:
(*value) += *operand;
return;
case ORTE_DSS_SUB:
(*value) -= *operand;
return;
case ORTE_DSS_MUL:
(*value) *= *operand;
return;
case ORTE_DSS_DIV:
if (0 == *operand) {
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
(*value) /= *operand;
return;
default:
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
return;
}
static void orte_dss_arith_uint16(uint16_t *value, uint16_t *operand, orte_dss_arith_op_t operation)
{
switch(operation) {
case ORTE_DSS_ADD:
(*value) += *operand;
return;
case ORTE_DSS_SUB:
(*value) -= *operand;
return;
case ORTE_DSS_MUL:
(*value) *= *operand;
return;
case ORTE_DSS_DIV:
if (0 == *operand) {
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
(*value) /= *operand;
return;
default:
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
return;
}
static void orte_dss_arith_int32(int32_t *value, int32_t *operand, orte_dss_arith_op_t operation)
{
switch(operation) {
case ORTE_DSS_ADD:
(*value) += *operand;
return;
case ORTE_DSS_SUB:
(*value) -= *operand;
return;
case ORTE_DSS_MUL:
(*value) *= *operand;
return;
case ORTE_DSS_DIV:
if (0 == *operand) {
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
(*value) /= *operand;
return;
default:
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
return;
}
static void orte_dss_arith_uint32(uint32_t *value, uint32_t *operand, orte_dss_arith_op_t operation)
{
switch(operation) {
case ORTE_DSS_ADD:
(*value) += *operand;
return;
case ORTE_DSS_SUB:
(*value) -= *operand;
return;
case ORTE_DSS_MUL:
(*value) *= *operand;
return;
case ORTE_DSS_DIV:
if (0 == *operand) {
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
(*value) /= *operand;
return;
default:
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
return;
}
static void orte_dss_arith_int64(int64_t *value, int64_t *operand, orte_dss_arith_op_t operation)
{
switch(operation) {
case ORTE_DSS_ADD:
(*value) += *operand;
return;
case ORTE_DSS_SUB:
(*value) -= *operand;
return;
case ORTE_DSS_MUL:
(*value) *= *operand;
return;
case ORTE_DSS_DIV:
if (0 == *operand) {
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
(*value) /= *operand;
return;
default:
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
return;
}
static void orte_dss_arith_uint64(uint64_t *value, uint64_t *operand, orte_dss_arith_op_t operation)
{
switch(operation) {
case ORTE_DSS_ADD:
(*value) += *operand;
return;
case ORTE_DSS_SUB:
(*value) -= *operand;
return;
case ORTE_DSS_MUL:
(*value) *= *operand;
return;
case ORTE_DSS_DIV:
if (0 == *operand) {
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
(*value) /= *operand;
return;
default:
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
return;
}
static void orte_dss_arith_data_type(orte_data_type_t *value, orte_data_type_t *operand, orte_dss_arith_op_t operation)
{
switch(operation) {
case ORTE_DSS_ADD:
(*value) += *operand;
return;
case ORTE_DSS_SUB:
(*value) -= *operand;
return;
case ORTE_DSS_MUL:
(*value) *= *operand;
return;
case ORTE_DSS_DIV:
if (0 == *operand) {
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
(*value) /= *operand;
return;
default:
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
return;
}
static void orte_dss_arith_daemon_cmd(orte_daemon_cmd_flag_t *value, orte_daemon_cmd_flag_t *operand, orte_dss_arith_op_t operation)
{
switch(operation) {
case ORTE_DSS_ADD:
(*value) += *operand;
return;
case ORTE_DSS_SUB:
(*value) -= *operand;
return;
case ORTE_DSS_MUL:
(*value) *= *operand;
return;
case ORTE_DSS_DIV:
if (0 == *operand) {
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
(*value) /= *operand;
return;
default:
ORTE_ERROR_LOG(ORTE_ERR_OPERATION_UNSUPPORTED);
return;
}
return;
}