Харт Джонсон М.
Шрифт:
Программа 12.4. SendReceiveSKST: безопасная многопоточная DLL
/* SendReceiveSKST.с — DLL многопоточного потокового сокета. */
/* В качестве разделителей сообщений используются символы конца */
/* строки ('\0'), так что размер сообщения заранее не известен. */
/* Поступающие данные буферизуются и сохраняются в промежутках между */
/* вызовами функций. */
/* Для этой цели используются локальные области хранения потоков */
/* (Thread Local Storage, TLS), обеспечивающие каждый из потоков */
/* собственным закрытым "статическим хранилищем". */
#define _NOEXCLUSIONS
#include "EvryThng.h"
#include "ClntSrvr.h" /* Определяет записи запроса и ответа. */
typedef struct STATIC_BUF_T {
/* "static_buf" содержит "static_buf_len" байтов остаточных данных. */
/* Символы конца строки (нулевые символы) могут присутствовать, а могут */
/* и не присутствовать. */
char static_buf[MAX_RQRS_LEN] ;
LONG32 static_buf_len;
} STATIC_BUF;
static DWORD TlsIx = 0; /* Индекс TLS – ДЛЯ КАЖДОГО ПРОЦЕССА СВОЙ ИНДЕКС.*/
/* Для однопоточной библиотеки использовались бы следующие определения:
static char static_buf [MAX_RQRS_LEN];
static LONG32 static_buf_len; */
/* Основная функция DLL. */
BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved) {
STATIC_BUF * pBuf;
switch (fdwReason) {
case DLL_PROCESS_ATTACH:
TlsIx = TlsAlloc;
/* Для основного потока подключение отсутствует, поэтому во время подключения процесса необходимо выполнить также операции по подключению потока. */
case DLL_THREAD_ATTACH:
/* Указать, что память не была распределена. */
TlsSetValue(TlsIx, NULL);
return TRUE; /* В действительности это значение игнорируется. */
case DLL_PROCESS_DETACH:
/* Отсоединить также основной поток. */
pBuf = TlsGetValue(TlsIx);
if (pBuf != NULL) {
free(pBuf);
pBuf = NULL;
}
return TRUE;
case DLL_THREAD_DETACH:
pBuf = TlsGetValue(TlsIx);
if (pBuf != NULL) {
free(pBuf);
pBuf = NULL;
}
return TRUE;
}
}
_declspec(dllexport)
BOOL ReceiveCSMessage(REQUEST *pRequest, SOCKET sd) {
/* Возвращаемое значение TRUE указывает на ошибку или отсоединение. */
BOOL Disconnect = FALSE;
LONG32 nRemainRecv = 0, nXfer, k; /* Должны быть целыми со знаком. */
LPSTR pBuffer, message;
CHAR TempBuf[MAX_RQRS_LEN + 1];
STATIC_BUF *p;
p = (STATIC_BUF *)TlsGetValue(TlsIx);
if (p == NULL) { /* Инициализация при первом вызове. */
/* Распределять это хранилище будут только те потоки, которым оно */
/* необходимо. Другие типы потоков могут использовать TLS для иных целей. */
р = malloc(sizeof(STATIC_BUF));
TlsSetValue(TlsIx, p);
if (p == NULL) return TRUE; /* Ошибка. */
p->static_buf_len = 0; /* Инициализировать состояние. */
}
message = pRequest->Record;
/* Считать до символа новой строки, оставляя остаточные данные в статическом буфере. */