示例#1
0
int Socket::accept(InetAddr &addr)
{
    struct sockaddr_in saddr;

    int connfd = acceptConn(fd_ , saddr);
    addr.setPureAddr(saddr);
    return connfd;
}
示例#2
0
文件: azcam3.cpp 项目: RTS2/rts2
int AzCam3DataConn::receive (rts2core::Block *block)
{
	recvs++;
	if (isConnState (CONN_DELETE))
		return -1;
	if (sock >= 0 && block->isForRead (sock))
	{
		int rec;
		if (isConnState (CONN_CONNECTING))
		{
			return acceptConn ();
		}

		if (headerSize < 256)
		{
			rec = recv (sock, header + headerSize, 256 - headerSize, 0);
			if (rec < 0)
				return -1;
			headerSize += rec;
			if (headerSize == 256)
			{
				std::cerr << "header " << header << std::endl;
				sscanf (header, "%ld", &dataSize);
				headerSize = 256+1;
			}
			return 0;
		}

		static char rbuf[2048*4];

		rec = recv (sock, rbuf, 2048*4, 0);
		
		if (rec > 0)
		{
			if (outFile == 0)
			{
				outFile = open ("/tmp/m.fits", O_CREAT | O_TRUNC | O_WRONLY, 00666);
				if (outFile < 0)
					return -1;
			}
			write (outFile, rbuf, rec);
			fsync (outFile);
			
			dataSize -= rec;
			if (dataSize <= 0)
			{
				close (outFile);
				dataSize = 0;
				headerSize = 0;
				wasRecvd = true;
				return rec;
			}
		}
		return rec;
	}
	return 0;
}
示例#3
0
文件: sockTestSrv.C 项目: jimix/k42
void
serverLoop()
{
  fd_set fds, orig_fds;
  FDType fd_echolisten, fd_cmdlisten;
  int max_fd;
  int rc;
  PortType port;

  port = 0;
  setupListenSock(&fd_echolisten, &port);
  rc = listen(fd_echolisten, 8);
  if (rc < 0) {
    fprintf(stderr, "Error: server listen failed (%d)\n", errno);
    exit(1);
  }
  printf("Echo Port = %d\n", port);
  EchoConnection::globalInit();

  port = 0;
  setupListenSock(&fd_cmdlisten, &port);
  rc = listen(fd_cmdlisten, 8);
  if (rc < 0) {
    fprintf(stderr, "Error: server listen failed (%d)\n", errno);
    exit(1);
  }
  printf("Cmd Port = %d\n", port);
  CmdConnection::globalInit();

  FD_ZERO(&orig_fds);
  FD_SET(fd_echolisten, &orig_fds);
  FD_SET(fd_cmdlisten, &orig_fds);

  max_fd = fd_cmdlisten;

  Connections::init(max_fd);

  while (1) {

    fds = orig_fds;

    rc = select(max_fd+1, &fds, NULL, NULL, NULL);

    if (rc < 0) {
      fprintf(stderr, "Error: server select failed (%d)\n", errno);
      exit(1);
    }

    if (rc == 0) {
      fprintf(stderr, "What Select timed out\n");
      exit(1);
    } else {
        if (FD_ISSET(fd_echolisten, &fds)) {
            // got a new echo connection
//            printf("got an echo connection on %d\n", fd_echolisten);
            acceptConn(fd_echolisten, &max_fd, &orig_fds, new EchoConnection);
            rc--;
            FD_CLR(fd_echolisten, &fds);
        }
        if (FD_ISSET(fd_cmdlisten, &fds)) {
            // got a new cmd connection
            printf("got an command connection on %d\n", fd_cmdlisten);
            acceptConn(fd_cmdlisten, &max_fd, &orig_fds,
                       new CmdConnection);
            rc--;
            FD_CLR(fd_cmdlisten, &fds);
        }
      FDType i = 0;
      while (rc > 0 && i <= max_fd) {
          if (FD_ISSET(i, &fds)) {
//              printf("Found activity on fd=%d\n", i);
              if (Connections::processConnectionForFD(i)==0) {
                  FD_CLR(i, &orig_fds);
                  close(i);
              }
          }
          i++;
      }

    }
  }
}