com.intel.analytics.bigdl.ppml.utils.EHSMParams.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of bigdl-ppml-spark_3.1.2 Show documentation
The newest version!
/*
 * Copyright 2016 The BigDL Authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.intel.analytics.bigdl.ppml.utils

import com.intel.analytics.bigdl.dllib.utils.Log4Error

import scala.collection.mutable.HashMap
import javax.crypto.Mac
import javax.crypto.spec.SecretKeySpec
import java.util.Base64

class EHSMParams(
      appId: String,
      apiKey: String,
      timeStamp: String) {

  protected val payLoad = new HashMap[String, String]

  def addPayloadElement(payloadElementKey: String, payloadElementVal: String): Unit = {
    payLoad(payloadElementKey) = payloadElementVal
  }

  def getPostJSONString(): String = {

    var postJSONString: String = "{\"appid\":\"" + appId + "\""
    postJSONString = postJSONString + ",\"payload\":{"
    for ((payloadElementKey, payloadElementVal) <- payLoad) {
      if (payloadElementKey == "keylen") {
        postJSONString = postJSONString + "\"" + payloadElementKey + "\":" +
          payloadElementVal + ','
      } else {
        postJSONString = postJSONString + "\"" + payloadElementKey + "\":\"" +
          payloadElementVal + "\","
      }
    }
    postJSONString = postJSONString.dropRight(1)
    val signCiphertextString: String = getSignCiphertextString()
    postJSONString = postJSONString + "},\"timestamp\":\"" + timeStamp +
      "\",\"sign\":\"" + signCiphertextString + "\"}"
    postJSONString
  }


  private def getSignCiphertextString(): String = {
    val secret = new SecretKeySpec(apiKey.getBytes("UTF-8"), "SHA256")
    val mac = Mac.getInstance("HmacSHA256")
    mac.init(secret)
    val signPlaintextString: String = getSignPlaintextString()
    val signCiphertextString: String = Base64.getEncoder.encodeToString(
      mac.doFinal(signPlaintextString.getBytes("UTF-8")))
    signCiphertextString
  }

  private def getSignPlaintextString(): String = {

    Log4Error.invalidInputError(appId != "" && apiKey != "" && timeStamp != ""
      && !payLoad.isEmpty, "Lack necessary param or payload!")
    var signString: String = s"appid=$appId&payload="
    val tmp = Map(payLoad.toSeq.sortWith(_._1 < _._1): _*)
    for ((payloadElementKey, payloadElementVal) <- tmp) {
      signString = signString + s"$payloadElementKey=$payloadElementVal&"
    }
    signString = signString + s"timestamp=$timeStamp"
    signString
  }

}