in progress

Author: usk94

arp.rb

@@ -1,26 +1,75 @@
 require "./ethernet"
 require "./utils"
+require "socket"
+
+ARPHRD_ETHER = 1
+ARPOP_REQUEST = 1
+ETH_ALEN = 6
+ETH_P_ARP = [0x0806].pack("S>").unpack1("S")
+ETH_P_IP = 0x0800
+ETH_TYPE_NUMBER_ARP = 0x0806
+PACKET_BROADCAST = 1
+PACKET_HOST = 0
+TIMEOUT_TIME = 3
+AF_PACKET = 0x11
+ETH_P_ALL = [0x0003].pack('S>').unpack('S').first
+SIOCGIFINDEX    = 0x8933   # bits/ioctls.h
+SIOCGIFHWADDR   = 0x8927   # linux/sockios.h
 
 class Arp
+  # TODO: attr_readerいらないかも
+  include Utils
   attr_reader :hardware_type, :protocol_type, :hardware_size, :protocol_size, :opcode, :sender_mac_addr,
               :sender_ip_addr, :target_mac_addr, :target_ip_addr
-  def initialize(sender_mac_addr, sender_ip_addr, target_ip_addr)
-    @hardware_type = 0x0001
-    @protocol_type = 0x0800
-    @hardware_size = 0x06
-    @protocol_size = 0x04
-    @opcode = 0x0001
-    @sender_mac_addr = sender_mac_addr
-    @sender_ip_addr = sender_ip_addr
-    @target_mac_addr = 0x000000000000
-    @target_ip_addr = target_ip_addr
+  def initialize(target_ip_addr)
+    @hardware_type = [0x0001].pack("S>")
+    @protocol_type = [0x0800].pack("S>")
+    @hardware_size = [0x06].pack("C")
+    @protocol_size = [0x04].pack("C")
+    @opcode = [0x0001].pack("S>")
+    @sender_mac_addr = get_packed_local_mac_addr
+    @sender_ip_addr = get_packed_local_ip_addr
+    @target_mac_addr = [0].pack("C") * 6
+    @target_ip_addr = target_ip_addr.split(".").map(&:to_i).pack("C*")
+  end
+
+  def sum_data_without_address
+    @hardware_type + @protocol_type + @hardware_size + @protocol_size + @opcode
+  end
+
+  def sum_address
+    sum_data_without_address + @sender_mac_addr + @sender_ip_addr + @target_mac_addr + @target_ip_addr
+  end
+
+  def bind_if(socket, interface)
+    # Get the system's internal interface index value
+    ifreq = [ interface, '' ].pack('a16a16')
+    socket.ioctl(SIOCGIFINDEX, ifreq)
+    index_str = ifreq[16, 4]
+
+    # Build our sockaddr_ll struct so we can bind to this interface. The struct
+    # is defined in linux/if_packet.h and requires the interface index.
+    eth_p_all_hbo = [ ETH_P_ALL ].pack('S').unpack('S>').first
+    sll = [ Socket::AF_PACKET, eth_p_all_hbo, index_str ].pack('SS>a16')
+    socket.bind(sll)
   end
 
   def send
-    utils = Utils.new
-    Ethernet.new("ARP", utils.get_local_mac_and_ip_addr.mac_addr, "ff:ff:ff:ff:ff:ff")
+    type = [0x0806].pack("S>")
+    src_mac_addr = get_packed_local_mac_addr
+    dst_mac_addr = [255].pack("C") * 6
+    header = Ethernet.new(type, src_mac_addr, dst_mac_addr).header
+    data = sum_data_without_address + sum_address
+    request = header + data
+    s = Socket.new(AF_PACKET, Socket::SOCK_RAW, ETH_P_ALL)
+    # ARPはUDP通信でもTCP通信でもない？
+    # client_socket_addr = Socket.pack_sockaddr_in(8863, Socket.gethostname)
+    # s.bind(client_socket_addr)
+    # bind_if(s)
+    bind_if(s, "en0")
+    res = s.send(request, 0)
+    p res
   end
 end
 
-a = Arp.new(1, 1, 1)
-p a
+Arp.new("10.14.69.254").send

ping.rb

@@ -0,0 +1,45 @@
+require "socket"
+
+class Ping
+  ICMP_ECHOREPLY = 0 # Echo reply
+  ICMP_ECHO      = 8 # Echo request
+  ICMP_SUBCODE   = 0
+  AF_PACKET = 0x11
+  TYPE_ICMP_ECHO_REQUEST = 0x08
+
+  # def initialize
+  #   @seq = 0
+  #   @bind_port = 0
+  #   @bind_host = nil
+  #   @data_size = 56
+  #   @data = ''
+
+  #   0.upto(@data_size){ |n| @data << (n % 256).chr }
+
+  #   @ping_id = (Thread.current.object_id ^ Process.pid) & 0xffff
+  # end
+
+  # def to_trans_data()
+  #   bynary_data =
+  #     @type.to_s(2).rjust(8, "0") +
+  #     @code.to_s(2).rjust(8, "0") +
+  #     checksum.to_s(2).rjust(16, "0") +
+  #     @id.to_s(2).rjust(16, "0") +
+  #     @seq_number.to_s(2).rjust(16, "0")
+
+  #   data_byte_arr = bynary_data.scan(/.{1,8}/)
+  #   data_byte_arr.map! { |byte| byte.to_i(2).chr } # TO ASCII
+  #   data_byte_arr.join + @data
+  # end
+
+  def ping
+    socket = Socket.new(Socket::AF_INET, Socket::SOCK_RAW, Socket::IPPROTO_ICMP)
+
+    socket.connect(Socket.pack_sockaddr_in(0, "8.8.8.8"))
+
+    socket.send("hoge", 0)
+  end
+end
+
+pinger = Ping.new
+pinger.ping