Уильям Стивенс - UNIX - разработка сетевых приложений

13 if (errno != EINPROGRESS)

14 return (-1);

15 /* Пока соединение устанавливается, мы можем заняться чем-то другим */

16 if (n == 0)

17 goto done; /* функция connect завершилась немедленно */

18 FD_ZERO(&rset);

19 FDSET(sockfd, &rset);

20 wset = rset;

21 tval.tv_sec = nsec;

22 tval.tv_usec = 0;

23 if ((n = Select(sockfd + 1, &rset, &wset, NULL,

24 nsec ? &tval : NULL)) == 0) {

25 close(sockfd); /* тайм-аут */

26 errno = ETIMEDOUT;

27 return (-1);

28 }

29 if (FD_ISSET(sockfd, &rset) || FD_ISSET(sockfd, &wset)) {

30 len = sizeof(error);

31 if (getsockopt(sockfd, SOL_SOCKET, SO_ERROR, &error, &len) < 0)

32 return (-1); /*в Solaris ошибка, ожидающая обработки */

33 } else

34 err_quit("select error: sockfd not set");

35 done:

36 Fcntl(sockfd, F_SETFL, flags); /* восстанавливаем флаги, задающие статус файла */

37 if (error) {

38 close(sockfd); /* на всякий случай */

39 errno = error;

40 return (-1);

41 }

42 return (0);

43 }

9-10 Мы вызываем функцию fcntl, которая делает сокет неблокируемым.

11-14 Мы вызываем неблокируемую функцию connect. Ошибка, которую мы ожидаем ( EINPROGRESS), указывает на то, что установление соединения началось, но еще не завершилось [128, с. 466]. Любая другая ошибка возвращается вызывающему процессу.

15 На этом этапе мы можем делать все, что захотим, ожидая завершения установления соединения.

16-17 Если неблокируемая функция connectвозвратила нуль, установление соединения завершилось. Как мы сказали, это может произойти, когда сервер находится на том же узле, что и клиент.

18-24 Мы вызываем функцию selectи ждем, когда сокет будет готов либо для чтения, либо для записи. Мы обнуляем rset, включаем бит, соответствующий sockfdв этом наборе дескрипторов и затем копируем rsetв wset. Это присваивание, возможно, является структурным присваиванием, поскольку обычно наборы дескрипторов представляются как структуры. Далее мы инициализируем структуру timevalи затем вызываем функцию select. Если вызывающий процесс задает четвертый аргумент нулевым (что соответствует использованию тайм-аута по умолчанию), следует задать в качестве последнего аргумента функции selectпустой указатель, а не структуру timevalс нулевым значением (означающим, что мы не ждем вообще).

25-28 Если функция selectвозвращает нуль, это означает, что время таймера истекло, и мы возвращаем вызывающему процессу ошибку ETIMEDOUT. Мы также закрываем сокет, чтобы трехэтапное рукопожатие не продолжалось.

29-34 Если дескриптор готов для чтения или для записи, мы вызываем функцию getsockopt, чтобы получить ошибку сокета ( SO_ERROR), ожидающую обработки. Если соединение завершилось успешно, это значение будет нулевым. Если при установлении соединения произошла ошибка, это значение является значением переменной errno, соответствующей ошибке соединения (например, ECONNREFUSED, ETIMEDOUTи т.д.). Мы также сталкиваемся с нашей первой проблемой переносимости. Если происходит ошибка, Беркли-реализации функции getsockoptвозвращают нуль, а ошибка, ожидающая обработки, возвращается в нашей переменной error. Но в системе Solaris сама функция getsockoptвозвращает -1, а переменная errnoпри этом принимает значение, соответствующее ошибке, ожидающей обработки. В нашем коде обрабатываются оба сценария.

36-42 Мы восстанавливаем флаги, задающие статус файла, и возвращаемся. Если наша переменная errno имеет ненулевое значение в результате выполнения функции getsockopt, это значение хранится в переменной errno, и функция возвращает -1.

Как мы сказали ранее, проблемы переносимости для функции connectсвязаны с различными реализациями сокетов и отключения блокировки. Во-первых, возможно, что установление соединения завершится и придут данные для собеседника до того, как будет вызвана функция select. В этом случае сокет будет готов для чтения и для записи при успешном выполнении функции, как и при неудачном установленном соединении. В нашем коде, показанном в листинге 16.7, этот сценарий обрабатывается при помощи вызова функции getsockoptи проверки на наличие ошибки, ожидающей обработки, для сокета.