Разработка приложений в среде Linux. Второе издание

 75:  int sizeRead;

 76:  struct tftpPacket data, response;

 77:  int blockNum = 0;

 78:

 79:  request.opcode = ntohs(request.opcode);

 80:  if (request.opcode != RRQ) die("неверный код операции");

 81:

 82:  fileName = request.u.bytes;

 83:  mode = fileName + strlen(fileName) + 1;

 84:

 85:  /* здесь поддерживается только режим bin */

 86:  if (strcmp(mode, "octet")) {

 87:   fprintf(stderr, "неверный режим %s\n", mode);

 88:   exit(1);

 89:  }

 90:

 91:  /* требуется передача при помощи сокета того же семейства и типа,

 92:     с которым мы начинали */

 93:  if ((s = socket(tftpAddr.ai_family, tftpAddr.ai_socktype,

 94:   tftpAddr.ai_protocol)) < 0)

 95:   die("send socket");

 96:

 97:  /* установить удаленный конец сокета на адрес, который

 98:     инициирует данное соединение */

 99:  if (connect(s, &remote, remoteLen))

100:   die("connect");

101:

102:  if ((fd = open(fileName, O_RDONLY)) < 0) {

103:   sendError(s, FILE_NOT_FOUND);

104:   close(s);

105:   return;

106:  }

107:

108:  data.opcode = htons(DATA);

109:  while ((size = read(fd, data.u.data.bytes, 512)) > 0) {

110:   data.u.data.block = htons(++blockNum);

111:

112:   /* размер составляют 2 байта (код операции), 2 байта (номер блока) и данные*/

113:   size += 4;

114:   if (send(s, &data, size, 0) != size)

115:    die("data send");

116:

117:   sizeRead = recv(s, &response, sizeof(response), 0);

118:   if (sizeRead < 0) die("recv ack");

119:

120:   response.opcode = ntohs(response.opcode);

121:   if (response.opcode != ACK) {

122:    fprintf(stderr, "непредвиденный код операции в отклике\n");

123:    exit(1);

124:   }

125:

126:   response.u.ack.block = ntohs(response.u.ack.block);

127:   if (response.u.ack.block != blockNum) {

128:    fprintf(stderr, "получено подтверждение неверного блока\n");

129:    exit(1);

130:   }

131:

132:   /* если блок, который мы только что отправили, содержит

133:      меньше 512 байт, то задача выполнена */

134:   if (size < 516) break;

135:  }

136:

137:  close(s);

138: }

139:

140: int main(int argc, char ** argv) {

141:  struct addrinfo hints, * addr;

142:  char * portAddress = "tftp";

143:  int s;

144:  int rc;

145:  int bytes, fromLen;

146:  struct sockaddr from;

147:  struct tftpPacket packet;

148:

149:  if (argc > 2) {

150:   fprintf(stderr, "использование: tftpserver [порт]\n");

151:   exit(1);

152:  }

153:

154:  if (argv[1]) portAddress = argv[1];

155:

156:  memset(&hints, 0, sizeof (hints));

157:

158:  hints.ai_socktype = SOCK_DGRAM;

159:  hints.ai_flags = AI_ADDRCONFIG | AI_PASSIVE;

160:  if ((rc = getaddrinfo(NULL, portAddress, &hints, &addr)))