encrypt all syntax elements

Author: whamidou

CMakeLists.txt

@@ -353,7 +353,7 @@ else()
 endif()
 add_library (LibOpenHevcWrapper ${libfilenames} ${YASM_OBJECTS})
 
-target_link_libraries(LibOpenHevcWrapper m)
+target_link_libraries(LibOpenHevcWrapper cryptopp)
 
 include_directories(. gpac/modules/openhevc_dec/ platform/x86/)
 

libavcodec/crypto.cpp

@@ -116,5 +116,5 @@ void DecryptC(Crypto_Handle hdl, const unsigned char *in_stream, int size_bits,
 }
 
 void DeleteCryptoC(Crypto_Handle hdl) {
-    DeleteCryptoC(hdl);
+    DeleteCrypto((AESDecoder *)hdl);
 }

libavcodec/hevc.c

@@ -4072,7 +4072,7 @@ static av_cold int hevc_decode_free(AVCodecContext *avctx)
     int i;
 
     pic_arrays_free(s);
-
+    DeleteCryptoC(s->HEVClc->dbs_g);
     av_freep(&s->md5_ctx);
 
     for(i=0; i < s->nals_allocated; i++) {
@@ -4148,7 +4148,7 @@ static av_cold int hevc_init_context(AVCodecContext *avctx)
     s->dynamic_alloc += HEVC_CONTEXTS;
     if (!s->cabac_state)
         goto fail;
-
+     s->HEVClc->dbs_g = InitC();
     s->tmp_frame = av_frame_alloc();
     s->dynamic_alloc += sizeof(AVFrame); 
     if (!s->tmp_frame)
@@ -4367,6 +4367,8 @@ static av_cold int hevc_decode_init(AVCodecContext *avctx)
         return ret;
 
     s->picture_struct = 0;
+    s->prev_pos = 0;
+    s->encrypt_params = HEVC_CRYPTO_MV_SIGNS | HEVC_CRYPTO_MVs | HEVC_CRYPTO_TRANSF_COEFF_SIGNS | HEVC_CRYPTO_TRANSF_COEFFS;
 
     if (avctx->extradata_size > 0 && avctx->extradata) {
         ret = hevc_decode_extradata(s);

libavcodec/hevc.h

@@ -36,6 +36,10 @@
 #include "thread.h"
 #include "videodsp.h"
 #include "hevc_defs.h"
+#include "crypto.h"
+
+#define EncryptMVDiffSign 1
+
 
 #define PARALLEL_SLICE   0
 #define PARALLEL_FILTERS 0
@@ -436,6 +440,18 @@ enum ChromaFormat
 #endif
 };
 
+/*
+ Encryption configuration
+ */
+enum hevc_crypto_features {
+    HEVC_CRYPTO_OFF = 0,
+    HEVC_CRYPTO_MVs = (1 << 0),
+    HEVC_CRYPTO_MV_SIGNS = (1 << 1),
+    HEVC_CRYPTO_TRANSF_COEFFS = (1 << 2),
+    HEVC_CRYPTO_TRANSF_COEFF_SIGNS = (1 << 3),
+    HEVC_CRYPTO_ON = (1 << 4) - 1,
+};
+
 #if REPN_FORMAT_IN_VPS
 typedef struct RepFormat
 {
@@ -1095,6 +1111,7 @@ typedef struct HEVCLocalContext {
     uint8_t slice_or_tiles_up_boundary;
 
     int ctb_tile_rs;
+    Crypto_Handle       dbs_g;
 
 } HEVCLocalContext;
 
@@ -1263,6 +1280,8 @@ typedef struct HEVCContext {
 
 	uint8_t force_first_slice_in_pic;
 	int64_t last_frame_pts;
+    uint8_t encrypt_params;
+    uint32_t prev_pos;
 } HEVCContext;
 
 int ff_hevc_decode_short_term_rps(HEVCContext *s, ShortTermRPS *rps,