Function l_connect

src/lsocketlib.cpp:118–167

}

static int l_connect (lua_State *L) {
  const char *host = luaL_checkstring(L, 1);
  auto port = static_cast<uint16_t>(luaL_checkinteger(L, 2));
  const char *const opts[] = { "tcp", "udp", nullptr };
  bool udp = luaL_checkoption(L, 3, "tcp", opts);

  StandaloneSocket& ss = pushsocket(L);

  if (udp) {
    ss.sock = ss.sched.addSocket();
    const bool host_is_ip_addr = ss.sock->peer.ip.fromString(host);
    ss.sock->peer.port = soup::Endianness::toNetwork(soup::native_u16_t(port));
#if SOUP_WINDOWS
    ss.sock->init(ss.sock->peer.ip.isV4() ? AF_INET : AF_INET6, SOCK_DGRAM);  /* init socket right away so we can use udpServerSend as client */
#else
    ss.sock->init(AF_INET6, SOCK_DGRAM);  /* init socket right away so we can use udpServerSend as client */
#endif
    ss.udp = true;
    ss.recvLoopUdp(*ss.sock);
    if (!host_is_ip_addr) {
      auto spTask = ss.sched.add<soup::ResolveIpAddrTask>(host);
      ss.sched.tick();
      lua_assert(!spTask->isWorkDone());
      if (lua_isyieldable(L)) {
        const auto ctx = reinterpret_cast<lua_KContext>(spTask.get());
        spTask.reset();
        return lua_yieldk(L, 0, ctx, connectudpcont);
      }
      do {
        soup::os::sleep(1);
        ss.sched.tick();
      } while (!spTask->isWorkDone());
      return restconnectudp(L, spTask.get());
    }
    return 1;
  }

  if (!lua_isyieldable(L)) {
    ss.sock = ss.sched.addSocket();
    if (l_unlikely(!ss.sock->connect(host, port)))
      return 0;
    ss.recvLoop();
    return 1;
  }

  auto pTask = ss.sched.add<soup::netConnectTask>(host, port).get();
  ss.sched.tick();
  lua_assert(!pTask->isWorkDone());
  return lua_yieldk(L, 0, reinterpret_cast<lua_KContext>(pTask), connectcont);
}

static int l_send (lua_State *L) {
  size_t len;