sockets - Cómo lanzar sockadd a sockaddr_in en swift

swift3 casting

Question

var sa_tab:[sockaddr?] = [sockaddr](repeating: sockaddr(), count: Int(RTAX_MAX))
let addr:sockaddr = sa_tab[Int(RTAX_DST)]!
let addr_in:sockaddr_in = unsafeBitCast(addr.self, to: UnsafeMutablePointer<sockaddr_in>.self).pointee

m que se cuelga en la tercera línea, no se puede inseguroBitCast entre tipos de diferentes tamaños

Aquí está el método completo.

func ROUNDUP(a:Int) -> Int{
    if (a) > 0 {
        return (1 + (((a) - 1) | (MemoryLayout<CLong>.size - 1)))
    }
    else{

    return MemoryLayout<CLong>.size
    }
}

class func defaultGatewayAddress() -> Int{

        var addressIntValue:UInt32 = 0
        var mib:[Int32] = [CTL_NET, PF_ROUTE, 0, AF_INET, NET_RT_FLAGS, RTF_GATEWAY]
        let l = UnsafeMutablePointer<Int>.allocate(capacity: MemoryLayout<Int>.size)
        var buf: UnsafeMutablePointer<CChar>?
        var p: UnsafeMutablePointer<CChar>?
        var rt: UnsafeMutablePointer<rt_msghdr>?
        var sa:UnsafeMutablePointer<sockaddr>?
        var sa_tab:[sockaddr?] = [sockaddr](repeating: sockaddr(), count: Int(RTAX_MAX))
        var _:Int
        var r:Int = -1

        if(sysctl(&mib, u_int(mib.count), nil, l, nil, 0) < 0) {
            return -1;
        }
        print(Int8.max)
        print(Int8.min)
        if(l.pointee > 0) {

            buf = UnsafeMutablePointer<CChar>.allocate(capacity: l.pointee)

            if(sysctl(&mib, u_int(mib.count), buf, l, nil, 0) < 0) {
                return -1;
            }


            p = buf
            let maxBuf = buf!.advanced(by: l.pointee)

            while (p! < maxBuf) {

                rt = p!.withMemoryRebound(to: rt_msghdr.self, capacity: 1, {$0})
                sa = rt!.advanced(by: 1).withMemoryRebound(to: sockaddr.self, capacity: 1, {$0})

                for i in 0..<RTAX_MAX {
                    if (rt!.pointee.rtm_addrs & (1 << i)) == 1{
                        print("insert /(sa!.pointee) at /(i)")
                        sa_tab.insert(sa!.pointee, at: Int(i))
                        sa = sa!.advanced(by: ROUNDUP(a: Int(sa!.pointee.sa_len)))
                    }
                    else{
                        sa_tab.insert(nil, at: Int(i))
                    }
                    if ((rt!.pointee.rtm_addrs & (RTA_DST|RTA_GATEWAY)) ==  (RTA_DST|RTA_GATEWAY)) &&
                        (sa_tab[Int(RTAX_DST)]?.sa_family == sa_family_t(AF_INET)) &&
                        (sa_tab[Int(RTAX_GATEWAY)]?.sa_family == sa_family_t(AF_INET)){

                        var addr:sockaddr = sa_tab[Int(RTAX_DST)]!

                        let addr_in:sockaddr_in = withUnsafePointer(to: &addr) {
                            $0.withMemoryRebound(to: sockaddr_in.self, capacity: 1) {
                                $0.pointee
                            }
                        }

                        if addr_in.sin_addr.s_addr == 0 {
                            var buffer = [CChar](repeating: CChar(0), count: Int(IFNAMSIZ) + 1)
                            let result = if_indextoname(UInt32((rt?.pointee.rtm_index)!), &buffer)
                            var char = "en0".cString(using: .utf8)
                            #if arch(i386) || arch(x86_64)
                                // This is a Simulator not an idevice
                                char = "en1".cString(using: .utf8)
                            #endif

                            if(strcmp(result, char) == 0){
                                r = 0
                                var gatewayAddr:sockaddr = sa_tab[Int(RTAX_GATEWAY)]!

                                let gatewayAddr_in:sockaddr_in = withUnsafePointer(to: &gatewayAddr) {
                                    $0.withMemoryRebound(to: sockaddr_in.self, capacity: 1) {
                                        $0.pointee
                                    }
                                }
                                addressIntValue = gatewayAddr_in.sin_addr.s_addr
                            }
                        }


                    }
                }

                p = p?.advanced(by: Int((rt?.pointee.rtm_msglen)!))

            }

            buf!.deallocate(capacity: l.pointee)

            let gatewayLongAddress:UInt32 = addressIntValue

            print("gatewayLongAddress: /(gatewayLongAddress)")

            let gatewayIPAddress = String(format:"%d.%d.%d.%d", (gatewayLongAddress & 0xFF),
                                          ((gatewayLongAddress >> 8) & 0xFF),
                                          ((gatewayLongAddress >> 16) & 0xFF),
                                          ((gatewayLongAddress >> 24) & 0xFF))

            print("gatewayIPAddress: /(gatewayIPAddress)")

        }

        return Int(addressIntValue)
    }

Answer 1

Gracias a la respuesta de Martin para este problema, junto con las otras áreas fijadas para obtener la dirección de la puerta de enlace como a continuación:

class func defaultGatewayAddress() -> Int{

        var addressIntValue:UInt32 = 0
        var mib:[Int32] = [CTL_NET, PF_ROUTE, 0, AF_INET, NET_RT_FLAGS, RTF_GATEWAY]
        let l = UnsafeMutablePointer<Int>.allocate(capacity: MemoryLayout<Int>.size)
        var buf: UnsafeMutableRawPointer?
        var p: UnsafeMutableRawPointer?
        var rt: UnsafeMutablePointer<rt_msghdr>?


        var sa_tab:[sockaddr?] = [sockaddr](repeating: sockaddr(), count: Int(RTAX_MAX))

        let lengthSysctl = Darwin.sysctl(UnsafeMutablePointer<Int32>(mutating: &mib), UInt32(mib.count), nil, l, nil, 0)

        if(lengthSysctl < 0) {
            return -1;
        }
        if(l.pointee > 0) {

            buf = malloc(l.pointee)

            let bufferSysctl = Darwin.sysctl(UnsafeMutablePointer<Int32>(mutating: &mib), UInt32(mib.count), buf, l, nil, 0)

            if(bufferSysctl < 0) {
                return -1;
            }

            p = buf
            let maxBuf = buf!.advanced(by: l.pointee)



            while (p! < maxBuf) {

                rt = p!.assumingMemoryBound(to: rt_msghdr.self)

                let destBuf = p!.advanced(by: MemoryLayout<rt_msghdr>.size)
                let gatewayBuf = p!.advanced(by: MemoryLayout<rt_msghdr>.size+MemoryLayout<sockaddr>.size*1)
                let netmaskBuf = p!.advanced(by: MemoryLayout<rt_msghdr>.size+MemoryLayout<sockaddr>.size*2)
                let interfaceNameBuf = p!.advanced(by: MemoryLayout<rt_msghdr>.size+MemoryLayout<sockaddr>.size*4)
                let interfaceAddrBuf = p!.advanced(by: MemoryLayout<rt_msghdr>.size+MemoryLayout<sockaddr>.size*5)


                let dest_sa = destBuf.assumingMemoryBound(to: sockaddr.self)
                let gateway_sa = gatewayBuf.assumingMemoryBound(to: sockaddr.self)
                let netmask_sa = netmaskBuf.assumingMemoryBound(to: sockaddr.self)
                let interfaceName_sa = interfaceNameBuf.assumingMemoryBound(to: sockaddr.self)
                let interfaceAddr_sa = interfaceAddrBuf.assumingMemoryBound(to: sockaddr.self)

                let sockAddrs:[Int32:UnsafeMutablePointer<sockaddr>] = [RTAX_DST:dest_sa, RTAX_GATEWAY:gateway_sa, RTAX_NETMASK:netmask_sa, RTAX_IFP:interfaceName_sa, RTAX_IFA:interfaceAddr_sa]
                for (index, pointer) in sockAddrs {


                    let bin = (rt!.pointee.rtm_addrs & (1 << index))
                    if bin > 0{
                        sa_tab.insert(pointer.pointee, at: Int(index))
                    }
                    else{
                        sa_tab.insert(nil, at: Int(index))
                    }
                }


                print("RTA_DST -> /((sa_tab[Int(RTAX_DST)]?.sa_family == sa_family_t(AF_INET)))")
                print("RTA_GATEWAY -> /((sa_tab[Int(RTAX_GATEWAY)]?.sa_family == sa_family_t(AF_INET)))")
                print("RTAX_DST -> /(RTAX_DST)")
                print("RTAX_GATEWAY -> /(RTAX_GATEWAY)")

                if ((rt!.pointee.rtm_addrs & (RTA_DST|RTA_GATEWAY)) ==  (RTA_DST|RTA_GATEWAY)) &&
                    (sa_tab[Int(RTAX_DST)]?.sa_family == sa_family_t(AF_INET)) &&
                    (sa_tab[Int(RTAX_GATEWAY)]?.sa_family == sa_family_t(AF_INET)){

                    var addr:sockaddr = sa_tab[Int(RTAX_DST)]!

                    let addr_in:sockaddr_in = withUnsafePointer(to: &addr) {
                        $0.withMemoryRebound(to: sockaddr_in.self, capacity: 1) {
                            $0.pointee
                        }
                    }

                    if addr_in.sin_addr.s_addr == 0 {
                        var buffer = [CChar](repeating: CChar(0), count: Int(IFNAMSIZ) + 1)
                        let result = if_indextoname(UInt32((rt?.pointee.rtm_index)!), &buffer)
                        var char = "en0".cString(using: .utf8)
                        #if arch(i386) || arch(x86_64)
                            // This is a Simulator not an idevice
                            char = "en0".cString(using: .utf8)
                        #endif

                        if(strcmp(result, char) == 0){
                            var gatewayAddr:sockaddr = sa_tab[Int(RTAX_GATEWAY)]!

                            let gatewayAddr_in:sockaddr_in = withUnsafePointer(to: &gatewayAddr) {
                                $0.withMemoryRebound(to: sockaddr_in.self, capacity: 1) {
                                    $0.pointee
                                }
                            }
                            addressIntValue = gatewayAddr_in.sin_addr.s_addr
                        }
                    }



                }

                p = p?.advanced(by: Int((rt?.pointee.rtm_msglen)!))

            }

            //            buf!.deallocate

            let gatewayLongAddress:UInt32 = addressIntValue

            print("gatewayLongAddress: /(gatewayLongAddress)")

            let gatewayIPAddress = String(format:"%d.%d.%d.%d", (gatewayLongAddress & 0xFF),
                                          ((gatewayLongAddress >> 8) & 0xFF),
                                          ((gatewayLongAddress >> 16) & 0xFF),
                                          ((gatewayLongAddress >> 24) & 0xFF))

            print("gatewayIPAddress: /(gatewayIPAddress)")

        }

        return Int(addressIntValue)
    }

Answer 2

withMemoryRebound() tomar la dirección de addr (que requiere que addr sea ​​una variable ), use withMemoryRebound() para volver a vincularla temporalmente a un puntero sockaddr_in , que luego se puede desreferenciar:

var addr: sockaddr = ...
let addr_in = withUnsafePointer(to: &addr) {
    $0.withMemoryRebound(to: sockaddr_in.self, capacity: 1) {
        $0.pointee
    }
}

Hay algunos problemas en su método defaultGatewayAddress() :

• sa = sa!.advanced(by: ROUNDUP(a: Int(sa!.pointee.sa_len))) avanza sa por sa_len multiplicado por el tamaño de sockaddr , que no es lo que pretendes.
• La prueba if ((rt!.pointee.rtm_addrs & (RTA_DST|RTA_GATEWAY)) == (RTA_DST|RTA_GATEWAY)) debe realizar después del ciclo for i in 0..<RTAX_MAX .
• Con sa_tab.insert(sa!.pointee, at: Int(i)) inserta nuevos elementos en la matriz en lugar de reemplazarlos.

Tenga en cuenta también que

rt = p!.withMemoryRebound(to: rt_msghdr.self, capacity: 1, {$0})

podría funcionar aquí, pero generalmente no es seguro: el puntero de $0 solo es válido para la ejecución del cierre y no debe pasarse al exterior.

Aquí hay una versión de trabajo de tu código. Es esencialmente una traducción de ¿Cómo puedo determinar la puerta de enlace predeterminada en iPhone? Swift (que a su vez parece estar construido en https://github.com/getlantern/libnatpmp/blob/master/getgateway.c ).

También lo he simplificado un poco y modificado para evitar todos los desempaquetes forzados.

func defaultGatewayAddress() -> in_addr? {

    var defaultGateway: in_addr?

    var mib:[Int32] = [CTL_NET, PF_ROUTE, 0, AF_INET, NET_RT_FLAGS, RTF_GATEWAY]
    var len = 0
    if sysctl(&mib, u_int(mib.count), nil, &len, nil, 0) < 0 {
        return nil
    }
    let buffer = UnsafeMutablePointer<CChar>.allocate(capacity: len)
    defer {
        buffer.deallocate(capacity: len)
    }
    if sysctl(&mib, u_int(mib.count), buffer, &len, nil, 0) < 0 {
        return nil
    }

    var sa_tab = [UnsafePointer<sockaddr>?](repeating: nil, count: Int(RTAX_MAX))

    var ptr = buffer
    while ptr < buffer + len {
        ptr.withMemoryRebound(to: rt_msghdr.self, capacity: 1) { rt in
            var sa = UnsafeMutableRawPointer(rt + 1).assumingMemoryBound(to: sockaddr.self)
            for i in 0..<RTAX_MAX {
                if rt.pointee.rtm_addrs & (1 << i) != 0 {
                    sa_tab[Int(i)] = UnsafePointer(sa)
                    sa = (UnsafeMutableRawPointer(sa) + Int(sa.pointee.sa_len)).assumingMemoryBound(to: sockaddr.self)
                } else {
                     sa_tab[Int(i)] =  nil
                }
            }

            if let dst = sa_tab[Int(RTAX_DST)],
                dst.pointee.sa_family == sa_family_t(AF_INET),
                let gateway = sa_tab[Int(RTAX_GATEWAY)],
                gateway.pointee.sa_family == sa_family_t(AF_INET)
            {
                dst.withMemoryRebound(to: sockaddr_in.self, capacity: 1) { addr in
                    if addr.pointee.sin_addr.s_addr == 0 {

                        var name = [CChar](repeating: CChar(0), count: Int(IFNAMSIZ) + 1)
                        if_indextoname(UInt32((rt.pointee.rtm_index)), &name)
                        if String(cString: name) == "en0" {
                            defaultGateway = gateway.withMemoryRebound(to: sockaddr_in.self, capacity: 1) {
                                $0.pointee.sin_addr
                            }
                        }
                    }
                }

            }
            ptr += Int(rt.pointee.rtm_msglen)
        }
    }

    return defaultGateway
}