Module: Rmega::Nodes::Uploadable

Includes:
Rmega::Net
Included in:
Expandable
Defined in:
lib/rmega/nodes/uploadable.rb

Instance Method Summary collapse

Methods included from Rmega::Net

#http_get_content, #http_post, #survive

Methods included from Options

included, #options

Methods included from Loggable

included, #logger

Instance Method Details

#encrypt_chunck(start, clean_buffer, aes_key, nonce) ⇒ Object 



23
24
25
26
27
28
29
30
31
32
 
# File 'lib/rmega/nodes/uploadable.rb', line 23

def encrypt_chunck(start, clean_buffer, aes_key, nonce)
  iv = nonce + [start/0x1000000000, start/0x10].pack('l>*')
  enc_data = aes_ctr_encrypt(aes_key, clean_buffer, iv)

  # calculate mac
  mac_iv = nonce * 2
  mac = aes_cbc_mac(aes_key, clean_buffer, mac_iv)

  return [enc_data, mac]
end

#read_chunk(file, start, size) ⇒ Object 



18
19
20
21
 
# File 'lib/rmega/nodes/uploadable.rb', line 18

def read_chunk(file, start, size)
  file.seek(start)
  file.read(size)
end

#upload(path) ⇒ Object 



34
35
36
37
38
39
40
41
42
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
80
81
82
83
84
85
86
87
 
# File 'lib/rmega/nodes/uploadable.rb', line 34

def upload(path)
  path = ::File.expand_path(path)
  filesize = ::File.size(path)

  raise "Empty file - #{path}" if filesize == 0

  file = ::File.open(path, 'rb')

  rnd_node_key = NodeKey.random
  file_handle = nil
  base_url = upload_url(filesize)

  pool = Pool.new
  read_mutex = Mutex.new

  progress = Progress.new(filesize, caption: 'Upload', filename: ::File.basename(path))

  chunk_macs = {}

  self.class.each_chunk(filesize) do |start, size|
    pool.process do
      clean_buffer = nil

      read_mutex.synchronize do
        clean_buffer = read_chunk(file, start, size)
      end

      encrypted_buffer, chunk_mac = *encrypt_chunck(start, clean_buffer, rnd_node_key.aes_key, rnd_node_key.ctr_nonce)
      file_handle = upload_chunk(base_url, start, encrypted_buffer)
      chunk_macs[start] = chunk_mac

      progress.increment(size)
    end
  end

  pool.shutdown

  # encrypt attributes
  _attr = serialize_attributes(:n => ::File.basename(path))
  _attr = aes_cbc_encrypt(rnd_node_key.aes_key, _attr)

  # Calculate meta_mac
  file_mac = aes_cbc_mac(rnd_node_key.aes_key, chunk_macs.sort.map(&:last).join, "\x0"*16)
  rnd_node_key.meta_mac = Utils.compact_to_8_bytes(file_mac)
  encrypted_key = aes_ecb_encrypt(session.master_key, rnd_node_key.generate)

  resp = request(a: 'p', t: handle, n: [
    {h: file_handle, t: 0, a: Utils.base64urlencode(_attr), k: Utils.base64urlencode(encrypted_key)}
  ])

  return Nodes::Factory.build(session, resp['f'][0])
ensure
  file.close if file
end

#upload_chunk(base_url, start, buffer) ⇒ Object 



6
7
8
9
10
11
12
13
14
15
16
 
# File 'lib/rmega/nodes/uploadable.rb', line 6

def upload_chunk(base_url, start, buffer)
  size = buffer.length
  stop = start + size - 1
  url = "#{base_url}/#{start}-#{stop}"

  survive do
    response = http_post(url, buffer)
    raise("Upload failed") if response.code.to_i != 200
    return response.body
  end
end