GF_Err gf_isom_set_ismacryp_protection(GF_ISOFile *the_file, u32 trackNumber, u32 desc_index, u32 scheme_type,
u32 scheme_version, char *scheme_uri, char *kms_URI,
Bool selective_encryption, u32 KI_length, u32 IV_length)
{
u32 original_format;
GF_Err e;
GF_SampleEntryBox *sea;
GF_TrackBox *trak = gf_isom_get_track_from_file(the_file, trackNumber);
if (!trak) return GF_BAD_PARAM;
e = Media_GetSampleDesc(trak->Media, desc_index, &sea, NULL);
if (e) return e;
/* Replacing the Media Type */
switch (sea->type) {
case GF_ISOM_BOX_TYPE_MP4A:
case GF_ISOM_BOX_TYPE_DAMR:
case GF_ISOM_BOX_TYPE_DEVC:
case GF_ISOM_BOX_TYPE_DQCP:
case GF_ISOM_BOX_TYPE_DSMV:
original_format = sea->type;
sea->type = GF_ISOM_BOX_TYPE_ENCA;
break;
case GF_ISOM_BOX_TYPE_MP4V:
case GF_ISOM_BOX_TYPE_D263:
original_format = sea->type;
sea->type = GF_ISOM_BOX_TYPE_ENCV;
break;
/*special case for AVC1*/
case GF_ISOM_BOX_TYPE_AVC1:
original_format = GF_4CC('2','6','4','b');
sea->type = GF_ISOM_BOX_TYPE_ENCV;
break;
case GF_ISOM_BOX_TYPE_MP4S:
original_format = sea->type;
sea->type = GF_ISOM_BOX_TYPE_ENCS;
break;
default:
return GF_BAD_PARAM;
}
sea->protection_info = (GF_ProtectionInfoBox *)sinf_New();
sea->protection_info->scheme_type = (GF_SchemeTypeBox *)schm_New();
sea->protection_info->scheme_type->scheme_type = scheme_type;
sea->protection_info->scheme_type->scheme_version = scheme_version;
if (scheme_uri) {
sea->protection_info->scheme_type->flags |= 0x000001;
sea->protection_info->scheme_type->URI = strdup(scheme_uri);
}
sea->protection_info->original_format = (GF_OriginalFormatBox *)frma_New();
sea->protection_info->original_format->data_format = original_format;
sea->protection_info->info = (GF_SchemeInformationBox *)schi_New();
sea->protection_info->info->ikms = (GF_ISMAKMSBox *)iKMS_New();
sea->protection_info->info->ikms->URI = strdup(kms_URI);
sea->protection_info->info->isfm = (GF_ISMASampleFormatBox *)iSFM_New();
sea->protection_info->info->isfm->selective_encryption = selective_encryption;
sea->protection_info->info->isfm->key_indicator_length = KI_length;
sea->protection_info->info->isfm->IV_length = IV_length;
return GF_OK;
}
GF_Err gf_isom_set_oma_protection(GF_ISOFile *the_file, u32 trackNumber, u32 desc_index,
char *contentID, char *kms_URI, u32 encryption_type, u64 plainTextLength, char *textual_headers, u32 textual_headers_len,
Bool selective_encryption, u32 KI_length, u32 IV_length)
{
u32 original_format;
GF_Err e;
GF_SampleEntryBox *sea;
GF_TrackBox *trak = gf_isom_get_track_from_file(the_file, trackNumber);
if (!trak) return GF_BAD_PARAM;
e = Media_GetSampleDesc(trak->Media, desc_index, &sea, NULL);
if (e) return e;
/* Replacing the Media Type */
switch (sea->type) {
case GF_ISOM_BOX_TYPE_MP4A:
case GF_ISOM_BOX_TYPE_DAMR:
case GF_ISOM_BOX_TYPE_DEVC:
case GF_ISOM_BOX_TYPE_DQCP:
case GF_ISOM_BOX_TYPE_DSMV:
original_format = sea->type;
sea->type = GF_ISOM_BOX_TYPE_ENCA;
break;
case GF_ISOM_BOX_TYPE_MP4V:
case GF_ISOM_BOX_TYPE_AVC1:
case GF_ISOM_BOX_TYPE_D263:
original_format = sea->type;
sea->type = GF_ISOM_BOX_TYPE_ENCV;
break;
case GF_ISOM_BOX_TYPE_MP4S:
original_format = sea->type;
sea->type = GF_ISOM_BOX_TYPE_ENCS;
break;
default:
return GF_BAD_PARAM;
}
sea->protection_info = (GF_ProtectionInfoBox *)sinf_New();
sea->protection_info->scheme_type = (GF_SchemeTypeBox *)schm_New();
sea->protection_info->scheme_type->scheme_type = GF_4CC('o','d','k','m');
sea->protection_info->scheme_type->scheme_version = 0x00000200;
sea->protection_info->original_format = (GF_OriginalFormatBox *)frma_New();
sea->protection_info->original_format->data_format = original_format;
sea->protection_info->info = (GF_SchemeInformationBox *)schi_New();
sea->protection_info->info->okms = (GF_OMADRMKMSBox *)odkm_New();
sea->protection_info->info->okms->fmt = (GF_OMADRMAUFormatBox*)gf_isom_box_new(GF_ISOM_BOX_TYPE_ODAF);
sea->protection_info->info->okms->fmt->selective_encryption = selective_encryption;
sea->protection_info->info->okms->fmt->key_indicator_length = KI_length;
sea->protection_info->info->okms->fmt->IV_length = IV_length;
sea->protection_info->info->okms->hdr = (GF_OMADRMCommonHeaderBox*)ohdr_New();
sea->protection_info->info->okms->hdr->EncryptionMethod = encryption_type;
sea->protection_info->info->okms->hdr->PaddingScheme = (encryption_type==0x01) ? 1 : 0;
sea->protection_info->info->okms->hdr->PlaintextLength = plainTextLength;
if (contentID) sea->protection_info->info->okms->hdr->ContentID = strdup(contentID);
if (kms_URI) sea->protection_info->info->okms->hdr->RightsIssuerURL = strdup(kms_URI);
if (textual_headers) {
sea->protection_info->info->okms->hdr->TextualHeaders = malloc(sizeof(char)*textual_headers_len);
memcpy(sea->protection_info->info->okms->hdr->TextualHeaders, textual_headers, sizeof(char)*textual_headers_len);
sea->protection_info->info->okms->hdr->TextualHeadersLen = textual_headers_len;
}
return GF_OK;
}
static GF_Err gf_isom_set_protected_entry(GF_ISOFile *the_file, u32 trackNumber, u32 desc_index, u8 version, u32 flags,
u32 scheme_type, u32 scheme_version, char *scheme_uri, Bool is_isma, GF_ProtectionInfoBox **out_sinf)
{
u32 original_format;
GF_Err e;
GF_SampleEntryBox *sea;
GF_ProtectionInfoBox *sinf;
GF_TrackBox *trak = gf_isom_get_track_from_file(the_file, trackNumber);
if (!trak) return GF_BAD_PARAM;
e = Media_GetSampleDesc(trak->Media, desc_index, &sea, NULL);
if (e) return e;
original_format = sea->type;
/* Replacing the Media Type */
switch (sea->type) {
case GF_ISOM_BOX_TYPE_MP4A:
case GF_ISOM_BOX_TYPE_DAMR:
case GF_ISOM_BOX_TYPE_DEVC:
case GF_ISOM_BOX_TYPE_DQCP:
case GF_ISOM_BOX_TYPE_DSMV:
case GF_ISOM_BOX_TYPE_AC3:
sea->type = GF_ISOM_BOX_TYPE_ENCA;
break;
case GF_ISOM_BOX_TYPE_MP4V:
case GF_ISOM_BOX_TYPE_D263:
sea->type = GF_ISOM_BOX_TYPE_ENCV;
break;
/*special case for AVC1*/
case GF_ISOM_BOX_TYPE_AVC1:
case GF_ISOM_BOX_TYPE_AVC2:
case GF_ISOM_BOX_TYPE_AVC3:
case GF_ISOM_BOX_TYPE_AVC4:
case GF_ISOM_BOX_TYPE_SVC1:
if (is_isma)
original_format = GF_4CC('2','6','4','b');
sea->type = GF_ISOM_BOX_TYPE_ENCV;
break;
case GF_ISOM_BOX_TYPE_HVC1:
case GF_ISOM_BOX_TYPE_HEV1:
case GF_ISOM_BOX_TYPE_HVC2:
case GF_ISOM_BOX_TYPE_HEV2:
case GF_ISOM_BOX_TYPE_SHC1:
case GF_ISOM_BOX_TYPE_SHV1:
case GF_ISOM_BOX_TYPE_HVT1:
if (is_isma)
original_format = GF_4CC('2','6','5','b');
sea->type = GF_ISOM_BOX_TYPE_ENCV;
break;
case GF_ISOM_BOX_TYPE_MP4S:
case GF_ISOM_BOX_TYPE_LSR1:
sea->type = GF_ISOM_BOX_TYPE_ENCS;
break;
case GF_ISOM_BOX_TYPE_STXT:
case GF_ISOM_BOX_TYPE_WVTT:
case GF_ISOM_BOX_TYPE_STPP:
sea->type = GF_ISOM_BOX_TYPE_ENCT;
break;
default:
return GF_BAD_PARAM;
}
sinf = (GF_ProtectionInfoBox *)sinf_New();
gf_list_add(sea->protections, sinf);
sinf->scheme_type = (GF_SchemeTypeBox *)schm_New();
sinf->scheme_type->version = version;
sinf->scheme_type->flags = flags;
sinf->scheme_type->scheme_type = scheme_type;
sinf->scheme_type->scheme_version = scheme_version;
if (sinf->scheme_type->flags == 1) {
sinf->scheme_type->URI = (char *)gf_malloc(sizeof(char)*strlen(scheme_uri));
memmove(sinf->scheme_type->URI, scheme_uri, strlen(scheme_uri));
}
sinf->original_format = (GF_OriginalFormatBox *)frma_New();
sinf->original_format->data_format = original_format;
//common to isma, cenc and oma
sinf->info = (GF_SchemeInformationBox *)schi_New();
*out_sinf = sinf;
return GF_OK;
}
