Function ql_syscall_recvmsg

qiling/os/posix/syscall/socket.py:698–745

(ql: Qiling, sockfd: int, msg_addr: int, flags: int)

def ql_syscall_recvmsg(ql: Qiling, sockfd: int, msg_addr: int, flags: int):
    sock = get_opened_socket(ql.os, sockfd)

    if isinstance(sock, int):
        return sock

    abits = ql.arch.bits
    endian = ql.arch.endian

    msghdr = make_msghdr(abits, endian)
    msg = msghdr.load_from(ql.mem, msg_addr)

    try:
        # TODO: handle the addr
        data, ancdata, mflags, addr = sock.recvmsg(msg.msg_namelen, msg.msg_controllen, flags)
    except ConnectionError:
        return -1

    iovec = make_iovec(abits, endian)
    iovec_addr = msg.msg_iov
    written = 0

    for _ in range(msg.msg_iovlen):
        with iovec.ref(ql.mem, iovec_addr) as obj:
            size = min(obj.iov_len, len(data) - written)
            ql.mem.write(obj.iov_base, data[written:written + size])

        written += size
        iovec_addr += iovec.sizeof()

    cmsghdr = make_cmsghdr(abits, endian)
    cmsg_addr = msg.msg_control

    for cmsg_level, cmsg_type, cmsg_data in ancdata:
        with cmsghdr.ref(ql.mem, cmsg_addr) as obj:
            obj.cmsg_len = len(cmsg_data)
            obj.cmsg_level = cmsg_level
            obj.cmsg_type = cmsg_type

        cmsg_addr += cmsghdr.sizeof()

        ql.mem.write(cmsg_addr, cmsg_data)
        cmsg_addr += len(cmsg_data)

    msg.msg_flags = mflags
    msg.save_to(ql.mem, msg_addr)

    return len(data)


def ql_syscall_recvfrom(ql: Qiling, sockfd: int, buf: int, length: int, flags: int, addr: int, addrlen: int):