File: util.rb

package info (click to toggle)
ruby-aliyun-sdk 0.8.0-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 792 kB
  • sloc: ruby: 7,909; ansic: 204; makefile: 4
file content (113 lines) | stat: -rw-r--r-- 3,317 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
# -*- encoding: utf-8 -*-

require 'time'
require 'base64'
require 'openssl'
require 'digest/md5'

module Aliyun
  module OSS

    ##
    # Util functions to help generate formatted Date, signatures,
    # etc.
    #
    module Util

      # Prefix for OSS specific HTTP headers
      HEADER_PREFIX = "x-oss-"

      class << self

        include Common::Logging

        # Calculate request signatures
        def get_signature(key, verb, headers, resources)
          logger.debug("Sign, headers: #{headers}, resources: #{resources}")

          content_md5 = headers['content-md5'] || ""
          content_type = headers['content-type'] || ""
          date = headers['date']

          cano_headers = headers.select { |k, v| k.start_with?(HEADER_PREFIX) }
                         .map { |k, v| [k.downcase.strip, v.strip] }
                         .sort.map { |k, v| [k, v].join(":") + "\n" }.join

          cano_res = resources[:path] || "/"
          sub_res = (resources[:sub_res] || {})
                    .sort.map { |k, v| v ? [k, v].join("=") : k }.join("&")
          cano_res += "?#{sub_res}" unless sub_res.empty?

          string_to_sign =
            "#{verb}\n#{content_md5}\n#{content_type}\n#{date}\n" +
            "#{cano_headers}#{cano_res}"

          Util.sign(key, string_to_sign)
        end

        # Sign a string using HMAC and BASE64
        # @param [String] key the secret key
        # @param [String] string_to_sign the string to sign
        # @return [String] the signature
        def sign(key, string_to_sign)
          logger.debug("String to sign: #{string_to_sign}")

          Base64.strict_encode64(
            OpenSSL::HMAC.digest('sha1', key, string_to_sign))
        end

        # Calculate content md5
        def get_content_md5(content)
          Base64.strict_encode64(OpenSSL::Digest::MD5.digest(content))
        end

        # Symbolize keys in Hash, recursively
        def symbolize(v)
          if v.is_a?(Hash)
            result = {}
            v.each_key { |k| result[k.to_sym] = symbolize(v[k]) }
            result
          elsif v.is_a?(Array)
            result = []
            v.each { |x| result << symbolize(x) }
            result
          else
            v
          end
        end

        # Get a crc value of the data
        def crc(data, init_crc = 0)
          CrcX::crc64(init_crc, data, data.size)
        end

        # Calculate a value of the crc1 combine with crc2. 
        def crc_combine(crc1, crc2, len2)
          CrcX::crc64_combine(crc1, crc2, len2)
        end

        def crc_check(crc_a, crc_b, operation)
          if crc_a.nil? || crc_b.nil? || crc_a.to_i != crc_b.to_i
            logger.error("The crc of #{operation} between client and oss is not inconsistent. crc_a=#{crc_a} crc_b=#{crc_b}")
            fail CrcInconsistentError.new("The crc of #{operation} between client and oss is not inconsistent.")
          end
        end

        def ensure_bucket_name_valid(name)
          unless (name =~ %r|^[a-z0-9][a-z0-9-]{1,61}[a-z0-9]$|)
            fail ClientError, "The bucket name is invalid."
          end
        end  

      end # self
    end # Util
  end # OSS
end # Aliyun

# Monkey patch to support #to_bool
class String
  def to_bool
    return true if self =~ /^true$/i
    false
  end
end