mirror of https://github.com/OISF/suricata
Minor cosmetic changes to the cuda code.
Moved a couple of functions to more cuda relevant files; Re-structured some data types.pull/400/head
Anoop Saldanha
13 years ago
committed by
Victor Julien
parent
c9f076def3
commit
602c91ed41
@ -0,0 +1,74 @@
|
|||||||
|
/* Copyright (C) 2007-2010 Open Information Security Foundation
|
||||||
|
*
|
||||||
|
* You can copy, redistribute or modify this Program under the terms of
|
||||||
|
* the GNU General Public License version 2 as published by the Free
|
||||||
|
* Software Foundation.
|
||||||
|
*
|
||||||
|
* 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
|
||||||
|
* version 2 along with this program; if not, write to the Free Software
|
||||||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
|
||||||
|
* 02110-1301, USA.
|
||||||
|
*/
|
||||||
|
|
||||||
|
/**
|
||||||
|
* \file
|
||||||
|
*
|
||||||
|
* \author Anoop Saldanha <anoopsaldanha@gmail.com>
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifdef __SC_CUDA_SUPPORT__
|
||||||
|
|
||||||
|
#include "suricata.h"
|
||||||
|
#include "util-mpm.h"
|
||||||
|
#include "util-cuda-handlers.h"
|
||||||
|
#include "util-cuda-vars.h"
|
||||||
|
#include "detect-engine-mpm.h"
|
||||||
|
#include "util-debug.h"
|
||||||
|
#include "util-mpm-ac.h"
|
||||||
|
|
||||||
|
static DetectEngineCtx *cuda_de_ctx = NULL;
|
||||||
|
|
||||||
|
void CudaVarsSetDeCtx(DetectEngineCtx *de_ctx)
|
||||||
|
{
|
||||||
|
if (cuda_de_ctx != NULL) {
|
||||||
|
SCLogError(SC_ERR_FATAL, "CudaVarsSetDeCtx() called more than once. "
|
||||||
|
"This function should be called only once during the "
|
||||||
|
"lifetime of the engine.");
|
||||||
|
exit(EXIT_FAILURE);
|
||||||
|
}
|
||||||
|
|
||||||
|
cuda_de_ctx = de_ctx;
|
||||||
|
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
int CudaThreadVarsInit(CudaThreadVars *ctv)
|
||||||
|
{
|
||||||
|
if (PatternMatchDefaultMatcher() != MPM_AC_CUDA)
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
MpmCudaConf *conf = CudaHandlerGetCudaProfile("mpm");
|
||||||
|
if (conf == NULL) {
|
||||||
|
SCLogError(SC_ERR_AC_CUDA_ERROR, "Error obtaining cuda mpm profile.");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
ctv->mpm_is_cuda = 1;
|
||||||
|
ctv->cuda_ac_cb = CudaHandlerModuleGetData(MPM_AC_CUDA_MODULE_NAME, MPM_AC_CUDA_MODULE_CUDA_BUFFER_NAME);
|
||||||
|
ctv->data_buffer_size_max_limit = conf->data_buffer_size_max_limit;
|
||||||
|
ctv->data_buffer_size_min_limit = conf->data_buffer_size_min_limit;
|
||||||
|
ctv->mpm_proto_tcp_ctx_ts = MpmFactoryGetMpmCtxForProfile(cuda_de_ctx, cuda_de_ctx->sgh_mpm_context_proto_tcp_packet, 0);
|
||||||
|
ctv->mpm_proto_tcp_ctx_tc = MpmFactoryGetMpmCtxForProfile(cuda_de_ctx, cuda_de_ctx->sgh_mpm_context_proto_tcp_packet, 1);
|
||||||
|
ctv->mpm_proto_udp_ctx_ts = MpmFactoryGetMpmCtxForProfile(cuda_de_ctx, cuda_de_ctx->sgh_mpm_context_proto_udp_packet, 0);
|
||||||
|
ctv->mpm_proto_udp_ctx_tc = MpmFactoryGetMpmCtxForProfile(cuda_de_ctx, cuda_de_ctx->sgh_mpm_context_proto_udp_packet, 1);
|
||||||
|
ctv->mpm_proto_other_ctx = MpmFactoryGetMpmCtxForProfile(cuda_de_ctx, cuda_de_ctx->sgh_mpm_context_proto_other_packet, 0);
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif
|
||||||
@ -0,0 +1,65 @@
|
|||||||
|
/* Copyright (C) 2007-2010 Open Information Security Foundation
|
||||||
|
*
|
||||||
|
* You can copy, redistribute or modify this Program under the terms of
|
||||||
|
* the GNU General Public License version 2 as published by the Free
|
||||||
|
* Software Foundation.
|
||||||
|
*
|
||||||
|
* 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
|
||||||
|
* version 2 along with this program; if not, write to the Free Software
|
||||||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
|
||||||
|
* 02110-1301, USA.
|
||||||
|
*/
|
||||||
|
|
||||||
|
/**
|
||||||
|
* \file
|
||||||
|
*
|
||||||
|
* \author Anoop Saldanha <anoopsaldanha@gmail.com>
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifdef __SC_CUDA_SUPPORT__
|
||||||
|
|
||||||
|
#ifndef __UTIL_CUDA_VARS__H__
|
||||||
|
#define __UTIL_CUDA_VARS__H__
|
||||||
|
|
||||||
|
#include "util-cuda-buffer.h"
|
||||||
|
#include "util-mpm.h"
|
||||||
|
#include "threads.h"
|
||||||
|
|
||||||
|
typedef struct CudaThreadVars_ {
|
||||||
|
/* cb - CudaBuffer */
|
||||||
|
CudaBufferData *cuda_ac_cb;
|
||||||
|
|
||||||
|
MpmCtx *mpm_proto_other_ctx;
|
||||||
|
|
||||||
|
MpmCtx *mpm_proto_tcp_ctx_ts;
|
||||||
|
MpmCtx *mpm_proto_udp_ctx_ts;
|
||||||
|
|
||||||
|
MpmCtx *mpm_proto_tcp_ctx_tc;
|
||||||
|
MpmCtx *mpm_proto_udp_ctx_tc;
|
||||||
|
|
||||||
|
uint16_t data_buffer_size_max_limit;
|
||||||
|
uint16_t data_buffer_size_min_limit;
|
||||||
|
|
||||||
|
uint8_t mpm_is_cuda;
|
||||||
|
} CudaThreadVars;
|
||||||
|
|
||||||
|
typedef struct CudaPacketVars_ {
|
||||||
|
uint8_t cuda_mpm_enabled;
|
||||||
|
uint8_t cuda_done;
|
||||||
|
uint16_t cuda_gpu_matches;
|
||||||
|
SCMutex cuda_mutex;
|
||||||
|
SCCondT cuda_cond;
|
||||||
|
uint32_t cuda_results[(UTIL_MPM_CUDA_DATA_BUFFER_SIZE_MAX_LIMIT_DEFAULT * 2) + 1];
|
||||||
|
} CudaPacketVars;
|
||||||
|
|
||||||
|
void CudaVarsSetDeCtx(struct DetectEngineCtx_ *de_ctx);
|
||||||
|
int CudaThreadVarsInit(CudaThreadVars *ctv);
|
||||||
|
|
||||||
|
#endif /* __UTIL_CUDA_VARS__H__ */
|
||||||
|
|
||||||
|
#endif /* __SC_CUDA_SUPPORT__ */
|
||||||
Loading…
Reference in New Issue