книги хакеры / журнал хакер / 054_Optimized

_

Thread(SocksHandlerTreadProc){

SOCKET s

= (SOCKET)pParam;

SOCKET tunnelSock = 0;

_try {

int iConnectResult = 0,

iReceiveRes = 0, iSendRes = 0, iSocketsSet =

0;

SOCKS4_REQUEST socks4Request, socks4Response;

SOCKADDR_IN remoteAddr = {0};

fd_

set fds_read;

FD_

ZERO(&fds_read);

FD_SET(s, &fds_read);

TIMEVAL tv = {0};

tv.tv_sec = 30;

char cSocksVersion = 0;

int iSelectResult = select(0, &fds_

if

read, NULL, NULL, &tv);

(iSelectResult == SOCKET_ERROR || !iSelectResult) _leave;

u_long ulVal = 0;

выделяются красненьким

Я ими активно

пользуюсь и тебе советую. Если ты юзаешь свои макросы, то очень удоб

ioctlsocket(s, FIONBIO , &ulVal);

программы. Для добавления ключевых слов

if

(cSocksVersion == 4){

PEEK);

â

директории, где

лежит

òâîÿ VS.NET,

iReceiveRes = recv(s, ((char*)&socks4Request),

sizeof(socks4Request),

Common7\IDE, и записать

0);

notepad'ом в него нужные ключевые слова по одному в

ñòðî

if (iReceiveRes == SOCKET_ERROR) _

ке. После этого

перезагрузи VS.NET и

радуйся красивому сорцу.

ulVal = 1;

leave;

Но вернемся к нашим баранам. В

ioctlsocket(s, FIONBIO , &ulVal);

FlushRecvBufferUntil(s,

'\0');

необходимости

созданной нити мы будем обрабатывать входящие

создавать новые коннекты с

серверами, указанными

if

(socks4Request.

в запросах. Вот как это происходит. Сначала

ulVal = 0;

ucCommand == 1){

принимаем один байт, в нем содержится

версия требуемого сокс-

ди сообщений.

протокола. Причем

получим этот байт,

íå

удаляя его из очере-

ioctlsocket(s, FIONBIO,

&ulVal);

Для этого при вызове winsock-

remoteAddr.sin_

функции recv в

четвертом

MSG_PEEK. Теперь,

параметре пе-

INET;

при следующем чтении из сокета, мы

сначала

memcpy(&(remoteAddr.sin_

получим этот же

addr),&(socks4Request.

байт и только потом все остальное. Это необходимо для

отброса

ëèø-

sizeof(remoteAddr.sin_addr));

dwDestIp),

íèõ áàéò ïðè

remoteAddr.sin_port = socks4Request.

ðó

проверке версии. Итак, если версия равна четырем, то заполним

структу-

socks4Request типа SOCKS4_REQUEST из нашего сокета:

tunnelSock

= socket(AF_

wDestPort;

INET, SOCK_

STREAM, 0);

recv(s, (char*)&socks4Request,

sizeof(socks4Request), 0);

iConnectResult = connect(tunnelSock,

(struct

sockaddr*) &remoteAddr,

sizeof(remoteAddr));

Далее

запустим

функцию FlushRecvBufferUntil, с помощью которой

if

(iConnectResult

!= SOCKET_ERROR){

пропускаем мимо

socks4Response.

= 90;

óøåé âñå

лишнее, что хочет сообщить нам клиент. И только когда

socks4Response.

ucCommand

заголовок запроса к

прокси полностью принят (пришел '\0'),

ìû

ucVersion = 0;

iSendRes = send(s, (char*)&socks4Response,

получили из запроса).

попытаемся сконнектиться с dwDestIp (его

sizeof(socks4Response), 0);

} else _leave;

При успешном соединении

} else _leave;

создаем бесконечный цикл, в котором будем совершать

îá-

мен данными в обоих сокетах: полученных

}

else _leave;

select è

accept`ом и connect`ом. С помощью функции

äèì

пары макросов проверим, есть ли данные в

сокетах. Для этого сначала созда-

ulVal = 0;

переменную

ioctlsocket(tunnelSock, FIONBIO,

типа fd_set, потом макросом FD_ZERO обнулим ее, а FD_

&ulVal);

âèì ê

íåé

указатель на сокет, из которого будем

читать данные.

SET`îì äîáà-

ioctlsocket(s, FIONBIO, &ulVal);

öèþ

Теперь, вызвав функ-

select и передав во втором

параметре

íàøó

переменную, мы

можем

воспользо-

char

swapBuffer[4096*16];

ваться макросом FD_

ISSET. С его помощью мы

узнаем состояние

очереди

чтения. И

int iSendResult, iRecvResult;

только когда она не пуста, читаем

tv.tv_sec = 2;

данные из одного сокета и пишем их в другой.

потоке, нам надо

while (true){

избавиться от одного нежелательного эффекта. Дело в

òîì, ÷òî

FD_

запущена, и ждет получения

функция recv

останавли-

ZERO(&fds_read);

FD_SET(s, &fds_read);

ки придут данные

данных. Если в момент

такой останов-

в другой сокет, наша

программа не

FD_

сможет их перенаправить, так как

SET(tunnelSock, &fds_read);

выяснилось, выйти из этого положения довольно

просто. Нашим соке-

iSocketsSet = select(0,

&fds_read, NULL,

NULL, &tv);

там необходимо сообщить

if (iSocketsSet

> 0){

некоторый параметр, заставляющий recv и другие

подобные

в отдельном

потоке.

if (FD_ISSET(s, &fds_read)){

Вот как это выглядит в нашей программе:

u_

iRecvResult = recv(s,

swapBuffer, sizeof(swapBuffer), MSG_PEEK);

if

(iRecvResult ==

long ulVal = 0;

0)_leave;

ioctlsocket(s, FIONBIO,

else

iRecvResult = recv(s, swapBuffer, sizeof(swapBuffer), 0);

&ulVal);

if (iRecvResult != SOCKET_

iSendResult

=

ERROR && iRecvResult > 0 )

send(tunnelSock, swapBuffer, iRecvResult, 0);

Сокеты, не

останавливающие

else _

}

leave;