mirror of
https://github.com/bobbimanners/emailler.git
synced 2024-05-28 20:41:33 +00:00
git-svn-id: http://svn.code.sf.net/p/netboot65/code@14 93682198-c243-4bdb-bd91-e943c89aac3b
This commit is contained in:
jonnosan
parent
401effa912
commit
1f9aae4c6b
|
|
@ -1,30 +0,0 @@
|
|||
#
|
||||
# netboot65 server
|
||||
#
|
||||
# Jonno Downes (jonno@jamtronix.com) - January, 2009
|
||||
#
|
||||
#
|
||||
|
||||
Thread.abort_on_exception=true
|
||||
|
||||
def log_msg(msg)
|
||||
puts "#{Time.now.strftime("%Y-%m-%d %H:%M:%S")} #{msg}"
|
||||
end
|
||||
|
||||
$:.unshift(File.dirname(__FILE__)) unless
|
||||
$:.include?(File.dirname(__FILE__)) || $:.include?(File.expand_path(File.dirname(__FILE__)))
|
||||
require 'netboot65_tftp'
|
||||
|
||||
bootfile_dir=File.expand_path(File.dirname(__FILE__)+"//..//boot")
|
||||
tftp_server=Netboot65TFTPServer.new(bootfile_dir)
|
||||
|
||||
tftp_server.start
|
||||
begin
|
||||
loop do
|
||||
sleep(1) #wake up every second to get keyboard input, so we break on ^C
|
||||
end
|
||||
rescue Interrupt
|
||||
log_msg "got interrupt signal - shutting down"
|
||||
end
|
||||
tftp_server.shutdown
|
||||
log_msg "shut down complete."
|
||||
|
|
@ -1,128 +0,0 @@
|
|||
#
|
||||
# minimal TFTP server implementation for use with netboot65
|
||||
#
|
||||
# Jonno Downes (jonno@jamtronix.com) - January, 2009
|
||||
#
|
||||
#
|
||||
|
||||
require 'socket'
|
||||
class Netboot65TFTPServer
|
||||
|
||||
TFTP_OPCODES={
|
||||
1=>'RRQ', #read request
|
||||
2=>'WRQ', #write request
|
||||
3=>'DATA',
|
||||
4=>'ACK',
|
||||
5=>'ERROR',
|
||||
}
|
||||
|
||||
TFTP_ERRORCODES={
|
||||
1 =>'File not found.',
|
||||
2 =>'Access violation.',
|
||||
3 =>'Disk full or allocation exceeded.',
|
||||
4 =>'Illegal TFTP operation.',
|
||||
5 =>'Unknown transfer ID.',
|
||||
6 =>'File already exists.',
|
||||
7 =>'No such user.',
|
||||
}
|
||||
TFTP_MAX_RESENDS=10
|
||||
|
||||
attr_reader :bootfile_dir,:port,:server_thread
|
||||
def initialize(bootfile_dir,port=69)
|
||||
@bootfile_dir=bootfile_dir
|
||||
@port=port
|
||||
@server_thread=nil
|
||||
|
||||
end
|
||||
|
||||
def send_error(client_ip,client_port,error_code,error_msg)
|
||||
packet=[5,error_code,error_msg,0].pack("nnA#{error_msg.length}c")
|
||||
socket=UDPSocket.open.send(packet,0,client_ip,client_port)
|
||||
log_msg("sent error #{error_code}:'#{error_msg}' to #{client_ip}:#{client_port}")
|
||||
end
|
||||
|
||||
def send_file(client_ip,client_port,filename)
|
||||
|
||||
client_sock=UDPSocket.open
|
||||
client_sock.connect(client_ip,client_port)
|
||||
data_to_send=File.open(filename,"rb").read
|
||||
blocks_to_send=(data_to_send.length.to_f/512.0).ceil
|
||||
log_msg("sending #{filename} to #{client_ip}:#{client_port} (#{blocks_to_send} blocks)")
|
||||
blocks_to_send.times do |block_number|
|
||||
block_data=data_to_send[block_number*512,512]
|
||||
packet=[3,block_number+1,block_data].pack("nnA*")
|
||||
got_ack=false
|
||||
TFTP_MAX_RESENDS.times do |attempt_number|
|
||||
log_msg("sending block #{block_number+1}/#{blocks_to_send} of #{filename} to #{client_ip}:#{client_port} - #{block_data.length} bytes - attempt #{attempt_number+1}")
|
||||
client_sock.send(packet,0,client_ip,client_port)
|
||||
if (IO.select([client_sock], nil, nil, 1)) then
|
||||
data,addr_info=client_sock.recvfrom(4096)
|
||||
client_ip=addr_info[3]
|
||||
client_port=addr_info[1]
|
||||
opcode=data[0,2].unpack("n")[0]
|
||||
opcode_description=TFTP_OPCODES[opcode]
|
||||
if opcode==4 then
|
||||
acked_block_number=data[2,2].unpack("n")[0]
|
||||
log_msg "TFTP: ACK from #{client_ip}:#{client_port} - block #{acked_block_number}"
|
||||
got_ack=true if acked_block_number==block_number+1
|
||||
else
|
||||
opcode_description="[UNK]" if opcode_description.nil?
|
||||
log_msg "TFTP: response from #{client_ip}:#{client_port} - opcode #{opcode} : #{opcode_description}"
|
||||
end
|
||||
end
|
||||
break if got_ack
|
||||
end
|
||||
if !got_ack then
|
||||
log_msg "TFTP: timed out waiting for ACK of block #{block_number} from #{client_ip}"
|
||||
break
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
def start()
|
||||
log_msg "TFTP: serving #{bootfile_dir} on port #{port}"
|
||||
Socket.do_not_reverse_lookup = true
|
||||
@server_thread=Thread.start do
|
||||
|
||||
loop do
|
||||
socket=UDPSocket.open
|
||||
socket.setsockopt Socket::SOL_SOCKET, Socket::SO_REUSEADDR, 1
|
||||
log_msg "waiting for TFTP client to connect"
|
||||
socket.bind("",port)
|
||||
data,addr_info=socket.recvfrom(4096)
|
||||
client_ip=addr_info[3]
|
||||
client_port=addr_info[1]
|
||||
opcode=data[0,2].unpack("n")[0]
|
||||
opcode_description=TFTP_OPCODES[opcode]
|
||||
opcode_description="[UNK]" if opcode_description.nil?
|
||||
log_msg "TFTP: connect from #{client_ip}:#{client_port} - opcode #{opcode} : #{opcode_description}"
|
||||
case opcode
|
||||
when 1 : #READ REQUEST
|
||||
opcode,filename_and_mode=data.unpack("nA*")
|
||||
filename,mode=filename_and_mode.split(0.chr)
|
||||
log_msg "RRQ for #{filename} (#{mode})"
|
||||
if filename=~/^\./ || filename=~/\.\./ then #looks like something dodgy - either a dotfile or a directory traversal attempt
|
||||
send_error(client_ip,client_port,1,"'#{filename}' invalid")
|
||||
else
|
||||
full_filename="#{bootfile_dir}/#{filename}"
|
||||
if File.file?(full_filename) then
|
||||
Thread.new {send_file(client_ip,client_port,full_filename)}
|
||||
else
|
||||
send_error(client_ip,client_port,1,"'#{filename}' not found")
|
||||
end
|
||||
end
|
||||
else
|
||||
send_error(client_ip,client_port,4,"opcode #{opcode} not supported")
|
||||
end
|
||||
socket.close
|
||||
end
|
||||
end
|
||||
# @server_thread.join
|
||||
end
|
||||
|
||||
def shutdown()
|
||||
log_msg "TFTP: stopping"
|
||||
server_thread.kill
|
||||
end
|
||||
|
||||
end
|
||||
Loading…
Reference in New Issue
Block a user