Системное программирование в среде Windows

Харт Джонсон М.

 /* Символы конца строки (нулевые символы) могут присутствовать, а могут */

 /* и не присутствовать. */

 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;

 /* Считать до символа новой строки, оставляя остаточные данные в статическом буфере. */

 for (k = 0; k < p->static_buf_len && p->static_buf[k] != '\0'; k++) { 

message[k] = p->static_buf[k];

 } /* k – количество переданных символов. */

 if (k < p->static_buf_len) { /* В статическом буфере обнаружен нулевой символ. */

message[k] = '\0';

p->static_buf_len –= (k + 1); /* Скорректировать состояние статического буфера. */

memcpy(p->static_buf, &(p->static_buf[k + 1]), p->static_buf_len);

return FALSE; /* Входные данные сокета не требуются. */

 }

 /* Передан весь статический буфер. Признак конца строки не обнаружен.*/

 nRemainRecv = sizeof(TempBuf) – 1 – p->static_buf_len;

 pBuffer = message + p->static_buf_len;

 p->static_buf_len = 0;

 while (nRemainRecv > 0 && !Disconnect) {

nXfer = recv(sd, TempBuf, nRemainRecv, 0);

if (nXfer <= 0) {

Disconnect = TRUE;

continue;

}

nRemainRecv –= nXfer;

/* Передать в целевое сообщение все символы вплоть до нулевого, если таковой имеется. */

for (k =0; k < nXfer && TempBuf[k] != '\0'; k++) {

*pBuffer = TempBuf[k];

pBuffer++;

}

if (k >= nXfer) { /*Признак конца строки не обнаружен, читать дальше*/

nRemainRecv –= nXfer;

} else { /* Обнаружен признак конца строки. */

*pBuffer = '\0';

nRemainRecv = 0;

memcpy(p->static_buf, &TempBuf[k + 1], nXfer – k – 1);

p->static_buf_len = nXfer – k – 1;

}

 }

 return Disconnect;

}

_declspec(dllexport)