CMakeLists.txt 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106
  1. set(SRC_COMMON
  2. util/crypto/basic.cpp
  3. util/system_utils.cpp
  4. util/io_utils.cpp
  5. util/crypto/aes_gcm.cpp
  6. util/crypto/sha256_utils.cpp
  7. util/crypto/base64.cpp)
  8. set(SRC_ENCRYPT
  9. safeapi/paddle_model_encrypt.cpp
  10. ${SRC_COMMON}
  11. )
  12. set(SRC_DECRYPT
  13. safeapi/paddle_model_decrypt.cpp
  14. ${SRC_COMMON})
  15. # encrypt: libpmodel-encrypt.so
  16. add_library(pmodel-encrypt SHARED
  17. ${SRC_ENCRYPT})
  18. IF (CMAKE_SYSTEM_NAME MATCHES "Windows")
  19. target_link_libraries(pmodel-encrypt
  20. ${OPENSSL_LIBS}
  21. )
  22. ELSEIF (CMAKE_SYSTEM_NAME MATCHES "Linux")
  23. target_link_libraries(pmodel-encrypt
  24. ${OPENSSL_LIBS}
  25. -ldl -lpthread
  26. )
  27. ENDIF ()
  28. IF(WITH_PADDLE)
  29. # decrypt: libpmodel-decrypt.so
  30. add_library(pmodel-decrypt SHARED
  31. ${SRC_DECRYPT})
  32. IF (CMAKE_SYSTEM_NAME MATCHES "Windows")
  33. target_link_libraries(pmodel-decrypt
  34. ${OPENSSL_LIBS}
  35. ${PADDLE_LIBS}
  36. )
  37. ELSEIF (CMAKE_SYSTEM_NAME MATCHES "Linux")
  38. target_link_libraries(pmodel-decrypt
  39. ${OPENSSL_LIBS}
  40. -ldl -lpthread
  41. )
  42. ENDIF ()
  43. ENDIF ()
  44. # decrypt: libpstream-decrypt.so
  45. add_library(pstream-decrypt SHARED
  46. ${SRC_COMMON}
  47. safeapi/paddle_stream_decrypt.cpp)
  48. IF (CMAKE_SYSTEM_NAME MATCHES "Windows")
  49. target_link_libraries(pstream-decrypt
  50. ${OPENSSL_LIBS}
  51. )
  52. ELSEIF (CMAKE_SYSTEM_NAME MATCHES "Linux")
  53. target_link_libraries(pstream-decrypt
  54. ${OPENSSL_LIBS}
  55. -ldl -lpthread
  56. )
  57. ENDIF ()
  58. # tool: paddle_encrypt_tool
  59. add_executable(paddle_encrypt_tool
  60. ../sample/paddle_encrypt_tool.cpp
  61. safeapi/paddle_model_encrypt.cpp
  62. util/crypto/basic.cpp
  63. util/system_utils.cpp
  64. util/io_utils.cpp
  65. util/crypto/aes_gcm.cpp
  66. util/crypto/sha256_utils.cpp
  67. util/crypto/base64.cpp)
  68. IF (CMAKE_SYSTEM_NAME MATCHES "Windows")
  69. target_link_libraries(paddle_encrypt_tool
  70. ${OPENSSL_LIBS}
  71. )
  72. ELSEIF (CMAKE_SYSTEM_NAME MATCHES "Linux")
  73. target_link_libraries(paddle_encrypt_tool
  74. ${OPENSSL_LIBS}
  75. -ldl -lpthread
  76. )
  77. ENDIF ()
  78. # test: test stream encryption and decryption
  79. add_executable(test_enc_dec
  80. ../sample/paddle_safe_stream_test.cpp
  81. safeapi/paddle_stream_decrypt.cpp
  82. safeapi/paddle_model_encrypt.cpp
  83. util/crypto/basic.cpp
  84. util/system_utils.cpp
  85. util/io_utils.cpp
  86. util/crypto/aes_gcm.cpp
  87. util/crypto/sha256_utils.cpp
  88. util/crypto/base64.cpp)
  89. IF (CMAKE_SYSTEM_NAME MATCHES "Windows")
  90. target_link_libraries(test_enc_dec
  91. ${OPENSSL_LIBS}
  92. )
  93. ELSEIF (CMAKE_SYSTEM_NAME MATCHES "Linux")
  94. target_link_libraries(test_enc_dec
  95. ${OPENSSL_LIBS}
  96. -ldl -lpthread
  97. )
  98. ENDIF ()