mirror of
https://github.com/chris124567/hulu
synced 2024-11-21 15:47:30 +00:00
Delete cdm.go
This commit is contained in:
Christopher Tarry
GitHub
parent
9bcd331a50
commit
91a73bd1bc
276
widevine/cdm.go
276
widevine/cdm.go
@ -1,276 +0,0 @@
|
|||||||
package widevine
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"crypto"
|
|
||||||
"crypto/aes"
|
|
||||||
"crypto/cipher"
|
|
||||||
"crypto/rsa"
|
|
||||||
"crypto/sha1"
|
|
||||||
"crypto/x509"
|
|
||||||
"encoding/pem"
|
|
||||||
"errors"
|
|
||||||
"math"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/aead/cmac"
|
|
||||||
"google.golang.org/protobuf/proto"
|
|
||||||
"lukechampine.com/frand"
|
|
||||||
)
|
|
||||||
|
|
||||||
type CDM struct {
|
|
||||||
privateKey *rsa.PrivateKey
|
|
||||||
clientID []byte
|
|
||||||
sessionID [32]byte
|
|
||||||
|
|
||||||
widevineCencHeader WidevineCencHeader
|
|
||||||
signedDeviceCertificate SignedDeviceCertificate
|
|
||||||
privacyMode bool
|
|
||||||
}
|
|
||||||
|
|
||||||
type Key struct {
|
|
||||||
ID []byte
|
|
||||||
Type License_KeyContainer_KeyType
|
|
||||||
Value []byte
|
|
||||||
}
|
|
||||||
|
|
||||||
// Creates a new CDM object with the specified device information.
|
|
||||||
func NewCDM(privateKey, clientID, initData []byte) (CDM, error) {
|
|
||||||
block, _ := pem.Decode(privateKey)
|
|
||||||
if block == nil || (block.Type != "PRIVATE KEY" && block.Type != "RSA PRIVATE KEY") {
|
|
||||||
return CDM{}, errors.New("failed to decode device private key")
|
|
||||||
}
|
|
||||||
|
|
||||||
keyParsed, err := x509.ParsePKCS1PrivateKey(block.Bytes)
|
|
||||||
if err != nil {
|
|
||||||
// if PCKS1 doesn't work, try PCKS8
|
|
||||||
pcks8Key, err := x509.ParsePKCS8PrivateKey(block.Bytes)
|
|
||||||
if err != nil {
|
|
||||||
return CDM{}, err
|
|
||||||
}
|
|
||||||
keyParsed = pcks8Key.(*rsa.PrivateKey)
|
|
||||||
}
|
|
||||||
|
|
||||||
var widevineCencHeader WidevineCencHeader
|
|
||||||
if len(initData) < 32 {
|
|
||||||
return CDM{}, errors.New("initData not long enough")
|
|
||||||
}
|
|
||||||
if err := proto.Unmarshal(initData[32:], &widevineCencHeader); err != nil {
|
|
||||||
return CDM{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
sessionID := func() (s [32]byte) {
|
|
||||||
c := []byte("ABCDEF0123456789")
|
|
||||||
for i := 0; i < 16; i++ {
|
|
||||||
s[i] = c[frand.Intn(len(c))]
|
|
||||||
}
|
|
||||||
s[16] = '0'
|
|
||||||
s[17] = '1'
|
|
||||||
for i := 18; i < 32; i++ {
|
|
||||||
s[i] = '0'
|
|
||||||
}
|
|
||||||
return s
|
|
||||||
}()
|
|
||||||
|
|
||||||
return CDM{
|
|
||||||
privateKey: keyParsed,
|
|
||||||
clientID: clientID,
|
|
||||||
|
|
||||||
widevineCencHeader: widevineCencHeader,
|
|
||||||
|
|
||||||
sessionID: sessionID,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Creates a new CDM object using the default device configuration.
|
|
||||||
func NewDefaultCDM(initData []byte) (CDM, error) {
|
|
||||||
return NewCDM(DefaultPrivateKey, DefaultClientID, initData)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Sets a device certificate. This is makes generating the license request
|
|
||||||
// more complicated but is supported. This is usually not necessary for most
|
|
||||||
// Widevine applications.
|
|
||||||
func (c *CDM) SetServiceCertificate(certData []byte) error {
|
|
||||||
var message SignedMessage
|
|
||||||
if err := proto.Unmarshal(certData, &message); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if err := proto.Unmarshal(message.Msg, &c.signedDeviceCertificate); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
c.privacyMode = true
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Generates the license request data. This is sent to the license server via
|
|
||||||
// HTTP POST and the server in turn returns the license response.
|
|
||||||
func (c *CDM) GetLicenseRequest() ([]byte, error) {
|
|
||||||
var licenseRequest SignedLicenseRequest
|
|
||||||
licenseRequest.Msg = new(LicenseRequest)
|
|
||||||
licenseRequest.Msg.ContentId = new(LicenseRequest_ContentIdentification)
|
|
||||||
licenseRequest.Msg.ContentId.CencId = new(LicenseRequest_ContentIdentification_CENC)
|
|
||||||
|
|
||||||
// this is probably really bad for the GC but protobuf uses pointers for optional
|
|
||||||
// fields so it is necessary and this is not a long running program
|
|
||||||
{
|
|
||||||
v := SignedLicenseRequest_LICENSE_REQUEST
|
|
||||||
licenseRequest.Type = &v
|
|
||||||
}
|
|
||||||
|
|
||||||
licenseRequest.Msg.ContentId.CencId.Pssh = &c.widevineCencHeader
|
|
||||||
|
|
||||||
{
|
|
||||||
v := LicenseType_DEFAULT
|
|
||||||
licenseRequest.Msg.ContentId.CencId.LicenseType = &v
|
|
||||||
}
|
|
||||||
|
|
||||||
licenseRequest.Msg.ContentId.CencId.RequestId = c.sessionID[:]
|
|
||||||
|
|
||||||
{
|
|
||||||
v := LicenseRequest_NEW
|
|
||||||
licenseRequest.Msg.Type = &v
|
|
||||||
}
|
|
||||||
|
|
||||||
{
|
|
||||||
v := uint32(time.Now().Unix())
|
|
||||||
licenseRequest.Msg.RequestTime = &v
|
|
||||||
}
|
|
||||||
|
|
||||||
{
|
|
||||||
v := ProtocolVersion_CURRENT
|
|
||||||
licenseRequest.Msg.ProtocolVersion = &v
|
|
||||||
}
|
|
||||||
|
|
||||||
{
|
|
||||||
v := uint32(frand.Uint64n(math.MaxUint32))
|
|
||||||
licenseRequest.Msg.KeyControlNonce = &v
|
|
||||||
}
|
|
||||||
|
|
||||||
if c.privacyMode {
|
|
||||||
pad := func(data []byte, blockSize int) []byte {
|
|
||||||
padlen := blockSize - (len(data) % blockSize)
|
|
||||||
if padlen == 0 {
|
|
||||||
padlen = blockSize
|
|
||||||
}
|
|
||||||
return append(data, bytes.Repeat([]byte{byte(padlen)}, padlen)...)
|
|
||||||
}
|
|
||||||
const blockSize = 16
|
|
||||||
|
|
||||||
var cidKey, cidIV [blockSize]byte
|
|
||||||
frand.Read(cidKey[:])
|
|
||||||
frand.Read(cidIV[:])
|
|
||||||
|
|
||||||
block, err := aes.NewCipher(cidKey[:])
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
paddedClientID := pad(c.clientID, blockSize)
|
|
||||||
encryptedClientID := make([]byte, len(paddedClientID))
|
|
||||||
cipher.NewCBCEncrypter(block, cidIV[:]).CryptBlocks(encryptedClientID, paddedClientID)
|
|
||||||
|
|
||||||
servicePublicKey, err := x509.ParsePKCS1PublicKey(c.signedDeviceCertificate.XDeviceCertificate.PublicKey)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
encryptedCIDKey, err := rsa.EncryptOAEP(sha1.New(), frand.Reader, servicePublicKey, cidKey[:], nil)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
licenseRequest.Msg.EncryptedClientId = new(EncryptedClientIdentification)
|
|
||||||
{
|
|
||||||
v := string(c.signedDeviceCertificate.XDeviceCertificate.ServiceId)
|
|
||||||
licenseRequest.Msg.EncryptedClientId.ServiceId = &v
|
|
||||||
}
|
|
||||||
licenseRequest.Msg.EncryptedClientId.ServiceCertificateSerialNumber = c.signedDeviceCertificate.XDeviceCertificate.SerialNumber
|
|
||||||
licenseRequest.Msg.EncryptedClientId.EncryptedClientId = encryptedClientID
|
|
||||||
licenseRequest.Msg.EncryptedClientId.EncryptedClientIdIv = cidIV[:]
|
|
||||||
licenseRequest.Msg.EncryptedClientId.EncryptedPrivacyKey = encryptedCIDKey
|
|
||||||
} else {
|
|
||||||
licenseRequest.Msg.ClientId = new(ClientIdentification)
|
|
||||||
if err := proto.Unmarshal(c.clientID, licenseRequest.Msg.ClientId); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
{
|
|
||||||
data, err := proto.Marshal(licenseRequest.Msg)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
hash := sha1.Sum(data)
|
|
||||||
if licenseRequest.Signature, err = rsa.SignPSS(frand.Reader, c.privateKey, crypto.SHA1, hash[:], &rsa.PSSOptions{SaltLength: rsa.PSSSaltLengthEqualsHash}); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return proto.Marshal(&licenseRequest)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Retrieves the keys from the license response data. These keys can be
|
|
||||||
// used to decrypt the DASH-MP4.
|
|
||||||
func (c *CDM) GetLicenseKeys(licenseRequest []byte, licenseResponse []byte) (keys []Key, err error) {
|
|
||||||
var license SignedLicense
|
|
||||||
if err = proto.Unmarshal(licenseResponse, &license); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
var licenseRequestParsed SignedLicenseRequest
|
|
||||||
if err = proto.Unmarshal(licenseRequest, &licenseRequestParsed); err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
licenseRequestMsg, err := proto.Marshal(licenseRequestParsed.Msg)
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
sessionKey, err := rsa.DecryptOAEP(sha1.New(), frand.Reader, c.privateKey, license.SessionKey, nil)
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
sessionKeyBlock, err := aes.NewCipher(sessionKey)
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
encryptionKey := []byte{1, 'E', 'N', 'C', 'R', 'Y', 'P', 'T', 'I', 'O', 'N', 0}
|
|
||||||
encryptionKey = append(encryptionKey, licenseRequestMsg...)
|
|
||||||
encryptionKey = append(encryptionKey, []byte{0, 0, 0, 0x80}...)
|
|
||||||
encryptionKeyCmac, err := cmac.Sum(encryptionKey, sessionKeyBlock, sessionKeyBlock.BlockSize())
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
encryptionKeyCipher, err := aes.NewCipher(encryptionKeyCmac)
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
unpad := func(b []byte) []byte {
|
|
||||||
if len(b) == 0 {
|
|
||||||
return b
|
|
||||||
}
|
|
||||||
// pks padding is designed so that the value of all the padding bytes is
|
|
||||||
// the number of padding bytes repeated so to figure out how many
|
|
||||||
// padding bytes there are we can just look at the value of the last
|
|
||||||
// byte
|
|
||||||
// i.e if there are 6 padding bytes then it will look at like
|
|
||||||
// <data> 0x6 0x6 0x6 0x6 0x6 0x6
|
|
||||||
count := int(b[len(b)-1])
|
|
||||||
return b[0 : len(b)-count]
|
|
||||||
}
|
|
||||||
for _, key := range license.Msg.Key {
|
|
||||||
decrypter := cipher.NewCBCDecrypter(encryptionKeyCipher, key.Iv)
|
|
||||||
decryptedKey := make([]byte, len(key.Key))
|
|
||||||
decrypter.CryptBlocks(decryptedKey, key.Key)
|
|
||||||
keys = append(keys, Key{
|
|
||||||
ID: key.Id,
|
|
||||||
Type: *key.Type,
|
|
||||||
Value: unpad(decryptedKey),
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
Loading…
Reference in New Issue
Block a user