Module: RubySMB::Server::ServerClient::TreeConnect

Included in:
RubySMB::Server::ServerClient
Defined in:
lib/ruby_smb/server/server_client/tree_connect.rb

Instance Method Summary collapse

Instance Method Details

#do_tree_connect_smb1(request, session) ⇒ Object 



6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
 
# File 'lib/ruby_smb/server/server_client/tree_connect.rb', line 6

def do_tree_connect_smb1(request, session)
  # see: https://docs.microsoft.com/en-us/openspecs/windows_protocols/ms-cifs/b062f3e3-1b65-4a9a-854a-0ee432499d8f
  response = RubySMB::SMB1::Packet::TreeConnectResponse.new

  share_name = RubySMB::Utils.safe_encode(request.data_block.path, 'UTF-8').split('\\', 4).last
  share_provider = @server.shares.transform_keys(&:downcase)[share_name.downcase]
  if share_provider.nil?
    logger.warn("Received TREE_CONNECT request for non-existent share: #{share_name}")
    response.smb_header.nt_status = WindowsError::NTStatus::STATUS_OBJECT_PATH_NOT_FOUND
    return response
  end
  logger.debug("Received TREE_CONNECT request for share: #{share_name}")

  tree_id = rand(1..0xfffe)
  tree_id = rand(1..0xfffe) while session.tree_connect_table.include?(tree_id)

  response.smb_header.tid = tree_id
  session.tree_connect_table[tree_id] = share_processor = share_provider.new_processor(self, session)
  response.parameter_block.access_rights = share_processor.maximal_access

  response
end

#do_tree_connect_smb2(request, session) ⇒ Object 



43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
 
# File 'lib/ruby_smb/server/server_client/tree_connect.rb', line 43

def do_tree_connect_smb2(request, session)
  response = RubySMB::SMB2::Packet::TreeConnectResponse.new
  response.smb2_header.credits = 1
  if session.tree_connect_table.length >= MAX_TREE_CONNECTIONS
    logger.warn('Connection has reached the maximum number of tree connections')
    response.smb2_header.nt_status = WindowsError::NTStatus::STATUS_INSUFFICIENT_RESOURCES
    return response
  end

  share_name = RubySMB::Utils.safe_encode(request.path, 'UTF-8').split('\\', 4).last
  share_provider = @server.shares.transform_keys(&:downcase)[share_name.downcase]

  if share_provider.nil?
    logger.warn("Received TREE_CONNECT request for non-existent share: #{share_name}")
    response.smb2_header.nt_status = WindowsError::NTStatus::STATUS_BAD_NETWORK_NAME
    return response
  end
  logger.debug("Received TREE_CONNECT request for share: #{share_name}")

  response.share_type = case share_provider.type
  when Share::TYPE_DISK
    RubySMB::SMB2::Packet::TreeConnectResponse::SMB2_SHARE_TYPE_DISK
  when Share::TYPE_PIPE
    RubySMB::SMB2::Packet::TreeConnectResponse::SMB2_SHARE_TYPE_PIPE
  when Share::TYPE_PRINT
    RubySMB::SMB2::Packet::TreeConnectResponse::SMB2_SHARE_TYPE_PRINT
  end

  tree_id = rand(1..0xfffffffe)
  tree_id = rand(1..0xfffffffe) while session.tree_connect_table.include?(tree_id)

  response.smb2_header.tree_id = tree_id
  session.tree_connect_table[tree_id] = share_processor = share_provider.new_processor(self, session)
  response.maximal_access = share_processor.maximal_access

  response
end

#do_tree_disconnect_smb1(request, session) ⇒ Object 



29
30
31
32
33
34
35
36
37
38
39
40
41
 
# File 'lib/ruby_smb/server/server_client/tree_connect.rb', line 29

def do_tree_disconnect_smb1(request, session)
  share_processor = session.tree_connect_table.delete(request.smb_header.tid)
  if share_processor.nil?
    response = RubySMB::SMB1::Packet::EmptyPacket.new
    response.smb_header.nt_status = WindowsError::NTStatus::STATUS_NETWORK_NAME_DELETED
    return response
  end

  logger.debug("Received TREE_DISCONNECT request for share: #{share_processor.provider.name}")
  share_processor.disconnect!
  response = RubySMB::SMB1::Packet::TreeDisconnectResponse.new
  response
end

#do_tree_disconnect_smb2(request, session) ⇒ Object 



81
82
83
84
85
86
87
88
89
90
91
92
93
 
# File 'lib/ruby_smb/server/server_client/tree_connect.rb', line 81

def do_tree_disconnect_smb2(request, session)
  share_processor = session.tree_connect_table.delete(request.smb2_header.tree_id)
  if share_processor.nil?
    response = SMB2::Packet::ErrorPacket.new
    response.smb2_header.nt_status = WindowsError::NTStatus::STATUS_NETWORK_NAME_DELETED
    return response
  end

  logger.debug("Received TREE_DISCONNECT request for share: #{share_processor.provider.name}")
  share_processor.disconnect!
  response = RubySMB::SMB2::Packet::TreeDisconnectResponse.new
  response
end