Your Name
3 years ago
19 changed files with 690 additions and 42 deletions
Binary file not shown.
@ -0,0 +1,34 @@ |
|||||||
|
FROM ubuntu:16.04 |
||||||
|
|
||||||
|
MAINTAINER <lishengyin, 1348640163@qq.com> |
||||||
|
|
||||||
|
|
||||||
|
ARG with_satellite_code=yes |
||||||
|
|
||||||
|
RUN apt-get update && \ |
||||||
|
DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \ |
||||||
|
cmake \ |
||||||
|
libmysqlclient-dev \ |
||||||
|
libssl-dev \ |
||||||
|
libgoogle-glog-dev \ |
||||||
|
libsqlite3-dev \ |
||||||
|
sqlite3 \ |
||||||
|
uuid-dev |
||||||
|
|
||||||
|
# 拷贝文件 |
||||||
|
COPY ./3rdparty/ZLToolKit.tar.gz /root |
||||||
|
COPY ./3rdparty/opencv.tar.gz /root |
||||||
|
COPY ./3rdparty/kafka.tar.gz /root |
||||||
|
|
||||||
|
# install zltoolkit |
||||||
|
RUN cd /root/ && tar -zxvf ZLToolKit.tar.gz |
||||||
|
RUN cp -rf /root/ZLToolKit/lib/* /usr/lib/ |
||||||
|
RUN cp -rf /root/ZLToolKit/include/* /usr/include/ |
||||||
|
|
||||||
|
RUN if [ "$with_cnstream_code" = "yes" ]; then \ |
||||||
|
cd /root && git clone http://120.79.157.137:80/lishengyin/Satellite.git Satellite && \ |
||||||
|
cd /root/Satellite && mkdir build && cd build &&\ |
||||||
|
cmake ..&& make -j; \ |
||||||
|
fi |
||||||
|
|
||||||
|
WORKDIR /root |
||||||
@ -0,0 +1,20 @@ |
|||||||
|
version: '2' |
||||||
|
|
||||||
|
services: |
||||||
|
satellite: |
||||||
|
build: |
||||||
|
context: $PWD/ |
||||||
|
dockerfile: Dockerfile.satellite |
||||||
|
volumes: |
||||||
|
- /tmp/.X11-unix:/tmp/.X11-unix |
||||||
|
- /usr/share/zoneinfo/Asia/Shanghai:/etc/localtime |
||||||
|
network_mode: "host" |
||||||
|
privileged: true |
||||||
|
ports: |
||||||
|
- "7400:7400" |
||||||
|
tty: true |
||||||
|
environment: |
||||||
|
- DISPLAY:$DISPLAY |
||||||
|
ipc: "host" |
||||||
|
pid: "host" |
||||||
|
command: /bin/bash -c "cd /root/Satellite/build && ./main" |
||||||
Binary file not shown.
@ -0,0 +1,76 @@ |
|||||||
|
/*
|
||||||
|
* @Description:
|
||||||
|
* @Version: 1.0 |
||||||
|
* @Autor: lishengyin |
||||||
|
* @Date: 2022-04-12 08:37:45 |
||||||
|
* @LastEditors: lishengyin |
||||||
|
* @LastEditTime: 2022-04-12 10:16:08 |
||||||
|
*/ |
||||||
|
#ifndef __DISPOSITION_HPP_ |
||||||
|
#define __DISPOSITION_HPP_ |
||||||
|
|
||||||
|
#include <iostream> |
||||||
|
#include "NonCopyable.hpp" |
||||||
|
#include <map> |
||||||
|
#include "login.hpp" |
||||||
|
|
||||||
|
#include "Util/logger.h" |
||||||
|
#include "Util/NoticeCenter.h" |
||||||
|
#include "Poller/EventPoller.h" |
||||||
|
#include "Util/SqlPool.h" |
||||||
|
#include "Network/TcpClient.h" |
||||||
|
#include "Poller/Timer.h" |
||||||
|
#include <vector> |
||||||
|
|
||||||
|
using namespace std; |
||||||
|
using namespace toolkit; |
||||||
|
|
||||||
|
|
||||||
|
class Disposition:private NonCopyable |
||||||
|
{ |
||||||
|
private: |
||||||
|
Disposition(){} |
||||||
|
|
||||||
|
public: |
||||||
|
using Ptr = std::shared_ptr<Disposition>; |
||||||
|
|
||||||
|
~Disposition(){} |
||||||
|
|
||||||
|
/**
|
||||||
|
* @description: 获取ptr |
||||||
|
* @param {*} |
||||||
|
* @return {*} |
||||||
|
*/
|
||||||
|
static std::shared_ptr<Disposition> getPtr(); |
||||||
|
|
||||||
|
/**
|
||||||
|
* @description: 初始化 |
||||||
|
* @param {*} |
||||||
|
* @return {*} |
||||||
|
*/
|
||||||
|
bool Init(); |
||||||
|
|
||||||
|
/**
|
||||||
|
* @description: 获取用户合法 |
||||||
|
* @param {*} |
||||||
|
* @return {*} |
||||||
|
*/
|
||||||
|
bool getUserValidity(Login& login); |
||||||
|
|
||||||
|
/**
|
||||||
|
* @description: 获取用户数据 |
||||||
|
* @param {*} |
||||||
|
* @return {*} |
||||||
|
*/
|
||||||
|
bool getUserInfos(); |
||||||
|
|
||||||
|
protected: |
||||||
|
std::string mysqlIP = "101.35.200.152"; |
||||||
|
int port = 13306; |
||||||
|
map<std::string, std::string> LoginInfos; |
||||||
|
}; |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
#endif |
||||||
@ -0,0 +1,74 @@ |
|||||||
|
/*
|
||||||
|
* @Description:
|
||||||
|
* @Version: 1.0 |
||||||
|
* @Autor: lishengyin |
||||||
|
* @Date: 2022-04-12 08:37:49 |
||||||
|
* @LastEditors: lishengyin |
||||||
|
* @LastEditTime: 2022-04-12 10:02:49 |
||||||
|
*/ |
||||||
|
#include "Disposition.hpp" |
||||||
|
#include "md5.hpp" |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @description: 获取ptr |
||||||
|
* @param {*} |
||||||
|
* @return {*} |
||||||
|
*/ |
||||||
|
std::shared_ptr<Disposition> Disposition::getPtr(){ |
||||||
|
static std::shared_ptr<Disposition> m_Disposition = nullptr;
|
||||||
|
if(m_Disposition == nullptr) m_Disposition = std::shared_ptr<Disposition>(new Disposition); |
||||||
|
return m_Disposition; |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @description: 初始化 |
||||||
|
* @param {*} |
||||||
|
* @return {*} |
||||||
|
*/ |
||||||
|
bool Disposition::Init(){ |
||||||
|
// 链接Mysql
|
||||||
|
#if defined(SUPPORT_DYNAMIC_TEMPLATE) |
||||||
|
//初始化数据
|
||||||
|
SqlPool::Instance().Init(this->mysqlIP,this->port,"","root",""/*,character*/); |
||||||
|
#else |
||||||
|
//由于需要编译器对可变参数模板的支持，所以gcc5.0以下一般都不支持，否则编译报错
|
||||||
|
ErrorL << "your compiler does not support variable parameter templates!" << endl; |
||||||
|
return false; |
||||||
|
#endif //defined(SUPPORT_DYNAMIC_TEMPLATE)
|
||||||
|
// 初始化数据库连接池
|
||||||
|
SqlPool::Instance().setSize(2 + thread::hardware_concurrency()); |
||||||
|
|
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* @description: 用户合法性 |
||||||
|
* @param {Login&} login |
||||||
|
* @return {*} |
||||||
|
*/ |
||||||
|
bool Disposition::getUserValidity(Login& login){ |
||||||
|
// 查询表列表
|
||||||
|
this->getUserInfos(); |
||||||
|
auto iter = LoginInfos.find(login.usr); |
||||||
|
if(iter == LoginInfos.end()) return false; |
||||||
|
// md5加密后与数据库的密码做比较
|
||||||
|
if(md5(login.usr+login.pwd) == iter->second) return true; |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
/**
|
||||||
|
* @description: 获取用户数据 |
||||||
|
* @param {*} |
||||||
|
* @return {*} |
||||||
|
*/ |
||||||
|
bool Disposition::getUserInfos(){ |
||||||
|
vector<vector<string>> sqlRet; |
||||||
|
SqlWriter sqlSelect("SELECT usr,pwd FROM SatelliteConfigDB.`UserInfos`"); |
||||||
|
sqlSelect << sqlRet; |
||||||
|
for(auto& line : sqlRet){ |
||||||
|
LoginInfos[line[0]] = line[1]; |
||||||
|
} |
||||||
|
return true; |
||||||
|
} |
||||||
@ -0,0 +1,39 @@ |
|||||||
|
/*
|
||||||
|
* @Description:
|
||||||
|
* @Version: 1.0 |
||||||
|
* @Autor: lishengyin |
||||||
|
* @Date: 2022-04-12 10:07:55 |
||||||
|
* @LastEditors: lishengyin |
||||||
|
* @LastEditTime: 2022-04-12 10:08:56 |
||||||
|
*/ |
||||||
|
#ifndef __NONCOPYABLE_HPP_ |
||||||
|
#define __NONCOPYABLE_HPP_ |
||||||
|
|
||||||
|
#include <iostream> |
||||||
|
using namespace std; |
||||||
|
|
||||||
|
// NonCopyAble
|
||||||
|
class NonCopyable { |
||||||
|
protected: |
||||||
|
NonCopyable() {}; |
||||||
|
~NonCopyable() {}; |
||||||
|
|
||||||
|
public: |
||||||
|
NonCopyable(const NonCopyable&) = delete; |
||||||
|
NonCopyable(NonCopyable&&) = delete; |
||||||
|
NonCopyable& operator=(const NonCopyable&) = delete; |
||||||
|
NonCopyable& operator=(NonCopyable&&) = delete; |
||||||
|
|
||||||
|
static std::string getDataTime(){ |
||||||
|
char ctime[80]; |
||||||
|
time_t rawtime; |
||||||
|
struct tm *info; |
||||||
|
time(&rawtime); |
||||||
|
info = localtime(&rawtime); |
||||||
|
strftime(ctime, 80, "%Y-%m-%d %H:%M:%S", info); |
||||||
|
std::string DataTime = ctime; |
||||||
|
return DataTime; |
||||||
|
} |
||||||
|
}; |
||||||
|
|
||||||
|
#endif |
||||||
@ -0,0 +1,61 @@ |
|||||||
|
#ifndef BZF_MD5_H |
||||||
|
#define BZF_MD5_H |
||||||
|
|
||||||
|
#include <cstring> |
||||||
|
#include <iostream> |
||||||
|
|
||||||
|
|
||||||
|
// a small class for calculating MD5 hashes of strings or byte arrays
|
||||||
|
// it is not meant to be fast or secure
|
||||||
|
//
|
||||||
|
// usage: 1) feed it blocks of uchars with update()
|
||||||
|
// 2) finalize()
|
||||||
|
// 3) get hexdigest() string
|
||||||
|
// or
|
||||||
|
// MD5(std::string).hexdigest()
|
||||||
|
//
|
||||||
|
// assumes that char is 8 bit and int is 32 bit
|
||||||
|
class MD5 |
||||||
|
{ |
||||||
|
public: |
||||||
|
typedef unsigned int size_type; // must be 32bit
|
||||||
|
|
||||||
|
MD5(); |
||||||
|
MD5(const std::string& text); |
||||||
|
void update(const unsigned char *buf, size_type length); |
||||||
|
void update(const char *buf, size_type length); |
||||||
|
MD5& finalize(); |
||||||
|
std::string hexdigest() const; |
||||||
|
friend std::ostream& operator<<(std::ostream&, MD5 md5); |
||||||
|
|
||||||
|
private: |
||||||
|
void init(); |
||||||
|
typedef unsigned char uint1; // 8bit
|
||||||
|
typedef unsigned int uint4; // 32bit
|
||||||
|
enum {blocksize = 64}; // VC6 won't eat a const static int here
|
||||||
|
|
||||||
|
void transform(const uint1 block[blocksize]); |
||||||
|
static void decode(uint4 output[], const uint1 input[], size_type len); |
||||||
|
static void encode(uint1 output[], const uint4 input[], size_type len); |
||||||
|
|
||||||
|
bool finalized; |
||||||
|
uint1 buffer[blocksize]; // bytes that didn't fit in last 64 byte chunk
|
||||||
|
uint4 count[2]; // 64bit counter for number of bits (lo, hi)
|
||||||
|
uint4 state[4]; // digest so far
|
||||||
|
uint1 digest[16]; // the result
|
||||||
|
|
||||||
|
// low level logic operations
|
||||||
|
static inline uint4 F(uint4 x, uint4 y, uint4 z); |
||||||
|
static inline uint4 G(uint4 x, uint4 y, uint4 z); |
||||||
|
static inline uint4 H(uint4 x, uint4 y, uint4 z); |
||||||
|
static inline uint4 I(uint4 x, uint4 y, uint4 z); |
||||||
|
static inline uint4 rotate_left(uint4 x, int n); |
||||||
|
static inline void FF(uint4 &a, uint4 b, uint4 c, uint4 d, uint4 x, uint4 s, uint4 ac); |
||||||
|
static inline void GG(uint4 &a, uint4 b, uint4 c, uint4 d, uint4 x, uint4 s, uint4 ac); |
||||||
|
static inline void HH(uint4 &a, uint4 b, uint4 c, uint4 d, uint4 x, uint4 s, uint4 ac); |
||||||
|
static inline void II(uint4 &a, uint4 b, uint4 c, uint4 d, uint4 x, uint4 s, uint4 ac); |
||||||
|
}; |
||||||
|
|
||||||
|
std::string md5(const std::string str); |
||||||
|
|
||||||
|
#endif |
||||||
@ -0,0 +1,330 @@ |
|||||||
|
/* interface header */ |
||||||
|
#include "md5.hpp" |
||||||
|
|
||||||
|
/* system implementation headers */ |
||||||
|
#include <cstdio> |
||||||
|
|
||||||
|
|
||||||
|
// Constants for MD5Transform routine.
|
||||||
|
#define S11 7 |
||||||
|
#define S12 12 |
||||||
|
#define S13 17 |
||||||
|
#define S14 22 |
||||||
|
#define S21 5 |
||||||
|
#define S22 9 |
||||||
|
#define S23 14 |
||||||
|
#define S24 20 |
||||||
|
#define S31 4 |
||||||
|
#define S32 11 |
||||||
|
#define S33 16 |
||||||
|
#define S34 23 |
||||||
|
#define S41 6 |
||||||
|
#define S42 10 |
||||||
|
#define S43 15 |
||||||
|
#define S44 21 |
||||||
|
|
||||||
|
///////////////////////////////////////////////
|
||||||
|
|
||||||
|
// F, G, H and I are basic MD5 functions.
|
||||||
|
inline MD5::uint4 MD5::F(uint4 x, uint4 y, uint4 z) { |
||||||
|
return (x&y) | (~x&z); |
||||||
|
} |
||||||
|
|
||||||
|
inline MD5::uint4 MD5::G(uint4 x, uint4 y, uint4 z) { |
||||||
|
return (x&z) | (y&~z); |
||||||
|
} |
||||||
|
|
||||||
|
inline MD5::uint4 MD5::H(uint4 x, uint4 y, uint4 z) { |
||||||
|
return x^y^z; |
||||||
|
} |
||||||
|
|
||||||
|
inline MD5::uint4 MD5::I(uint4 x, uint4 y, uint4 z) { |
||||||
|
return y ^ (x | ~z); |
||||||
|
} |
||||||
|
|
||||||
|
// rotate_left rotates x left n bits.
|
||||||
|
inline MD5::uint4 MD5::rotate_left(uint4 x, int n) { |
||||||
|
return (x << n) | (x >> (32-n)); |
||||||
|
} |
||||||
|
|
||||||
|
// FF, GG, HH, and II transformations for rounds 1, 2, 3, and 4.
|
||||||
|
// Rotation is separate from addition to prevent recomputation.
|
||||||
|
inline void MD5::FF(uint4 &a, uint4 b, uint4 c, uint4 d, uint4 x, uint4 s, uint4 ac) { |
||||||
|
a = rotate_left(a+ F(b,c,d) + x + ac, s) + b; |
||||||
|
} |
||||||
|
|
||||||
|
inline void MD5::GG(uint4 &a, uint4 b, uint4 c, uint4 d, uint4 x, uint4 s, uint4 ac) { |
||||||
|
a = rotate_left(a + G(b,c,d) + x + ac, s) + b; |
||||||
|
} |
||||||
|
|
||||||
|
inline void MD5::HH(uint4 &a, uint4 b, uint4 c, uint4 d, uint4 x, uint4 s, uint4 ac) { |
||||||
|
a = rotate_left(a + H(b,c,d) + x + ac, s) + b; |
||||||
|
} |
||||||
|
|
||||||
|
inline void MD5::II(uint4 &a, uint4 b, uint4 c, uint4 d, uint4 x, uint4 s, uint4 ac) { |
||||||
|
a = rotate_left(a + I(b,c,d) + x + ac, s) + b; |
||||||
|
} |
||||||
|
|
||||||
|
//////////////////////////////////////////////
|
||||||
|
|
||||||
|
// default ctor, just initailize
|
||||||
|
MD5::MD5() |
||||||
|
{ |
||||||
|
init(); |
||||||
|
} |
||||||
|
|
||||||
|
//////////////////////////////////////////////
|
||||||
|
|
||||||
|
// nifty shortcut ctor, compute MD5 for string and finalize it right away
|
||||||
|
MD5::MD5(const std::string &text) |
||||||
|
{ |
||||||
|
init(); |
||||||
|
update(text.c_str(), text.length()); |
||||||
|
finalize(); |
||||||
|
} |
||||||
|
|
||||||
|
//////////////////////////////
|
||||||
|
|
||||||
|
void MD5::init() |
||||||
|
{ |
||||||
|
finalized=false; |
||||||
|
|
||||||
|
count[0] = 0; |
||||||
|
count[1] = 0; |
||||||
|
|
||||||
|
// load magic initialization constants.
|
||||||
|
state[0] = 0x67452301; |
||||||
|
state[1] = 0xefcdab89; |
||||||
|
state[2] = 0x98badcfe; |
||||||
|
state[3] = 0x10325476; |
||||||
|
} |
||||||
|
|
||||||
|
//////////////////////////////
|
||||||
|
|
||||||
|
// decodes input (unsigned char) into output (uint4). Assumes len is a multiple of 4.
|
||||||
|
void MD5::decode(uint4 output[], const uint1 input[], size_type len) |
||||||
|
{ |
||||||
|
for (unsigned int i = 0, j = 0; j < len; i++, j += 4) |
||||||
|
output[i] = ((uint4)input[j]) | (((uint4)input[j+1]) << 8) | |
||||||
|
(((uint4)input[j+2]) << 16) | (((uint4)input[j+3]) << 24); |
||||||
|
} |
||||||
|
|
||||||
|
//////////////////////////////
|
||||||
|
|
||||||
|
// encodes input (uint4) into output (unsigned char). Assumes len is
|
||||||
|
// a multiple of 4.
|
||||||
|
void MD5::encode(uint1 output[], const uint4 input[], size_type len) |
||||||
|
{ |
||||||
|
for (size_type i = 0, j = 0; j < len; i++, j += 4) { |
||||||
|
output[j] = input[i] & 0xff; |
||||||
|
output[j+1] = (input[i] >> 8) & 0xff; |
||||||
|
output[j+2] = (input[i] >> 16) & 0xff; |
||||||
|
output[j+3] = (input[i] >> 24) & 0xff; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
//////////////////////////////
|
||||||
|
|
||||||
|
// apply MD5 algo on a block
|
||||||
|
void MD5::transform(const uint1 block[blocksize]) |
||||||
|
{ |
||||||
|
uint4 a = state[0], b = state[1], c = state[2], d = state[3], x[16]; |
||||||
|
decode (x, block, blocksize); |
||||||
|
|
||||||
|
/* Round 1 */ |
||||||
|
FF (a, b, c, d, x[ 0], S11, 0xd76aa478); /* 1 */ |
||||||
|
FF (d, a, b, c, x[ 1], S12, 0xe8c7b756); /* 2 */ |
||||||
|
FF (c, d, a, b, x[ 2], S13, 0x242070db); /* 3 */ |
||||||
|
FF (b, c, d, a, x[ 3], S14, 0xc1bdceee); /* 4 */ |
||||||
|
FF (a, b, c, d, x[ 4], S11, 0xf57c0faf); /* 5 */ |
||||||
|
FF (d, a, b, c, x[ 5], S12, 0x4787c62a); /* 6 */ |
||||||
|
FF (c, d, a, b, x[ 6], S13, 0xa8304613); /* 7 */ |
||||||
|
FF (b, c, d, a, x[ 7], S14, 0xfd469501); /* 8 */ |
||||||
|
FF (a, b, c, d, x[ 8], S11, 0x698098d8); /* 9 */ |
||||||
|
FF (d, a, b, c, x[ 9], S12, 0x8b44f7af); /* 10 */ |
||||||
|
FF (c, d, a, b, x[10], S13, 0xffff5bb1); /* 11 */ |
||||||
|
FF (b, c, d, a, x[11], S14, 0x895cd7be); /* 12 */ |
||||||
|
FF (a, b, c, d, x[12], S11, 0x6b901122); /* 13 */ |
||||||
|
FF (d, a, b, c, x[13], S12, 0xfd987193); /* 14 */ |
||||||
|
FF (c, d, a, b, x[14], S13, 0xa679438e); /* 15 */ |
||||||
|
FF (b, c, d, a, x[15], S14, 0x49b40821); /* 16 */ |
||||||
|
|
||||||
|
/* Round 2 */ |
||||||
|
GG (a, b, c, d, x[ 1], S21, 0xf61e2562); /* 17 */ |
||||||
|
GG (d, a, b, c, x[ 6], S22, 0xc040b340); /* 18 */ |
||||||
|
GG (c, d, a, b, x[11], S23, 0x265e5a51); /* 19 */ |
||||||
|
GG (b, c, d, a, x[ 0], S24, 0xe9b6c7aa); /* 20 */ |
||||||
|
GG (a, b, c, d, x[ 5], S21, 0xd62f105d); /* 21 */ |
||||||
|
GG (d, a, b, c, x[10], S22, 0x2441453); /* 22 */ |
||||||
|
GG (c, d, a, b, x[15], S23, 0xd8a1e681); /* 23 */ |
||||||
|
GG (b, c, d, a, x[ 4], S24, 0xe7d3fbc8); /* 24 */ |
||||||
|
GG (a, b, c, d, x[ 9], S21, 0x21e1cde6); /* 25 */ |
||||||
|
GG (d, a, b, c, x[14], S22, 0xc33707d6); /* 26 */ |
||||||
|
GG (c, d, a, b, x[ 3], S23, 0xf4d50d87); /* 27 */ |
||||||
|
GG (b, c, d, a, x[ 8], S24, 0x455a14ed); /* 28 */ |
||||||
|
GG (a, b, c, d, x[13], S21, 0xa9e3e905); /* 29 */ |
||||||
|
GG (d, a, b, c, x[ 2], S22, 0xfcefa3f8); /* 30 */ |
||||||
|
GG (c, d, a, b, x[ 7], S23, 0x676f02d9); /* 31 */ |
||||||
|
GG (b, c, d, a, x[12], S24, 0x8d2a4c8a); /* 32 */ |
||||||
|
|
||||||
|
/* Round 3 */ |
||||||
|
HH (a, b, c, d, x[ 5], S31, 0xfffa3942); /* 33 */ |
||||||
|
HH (d, a, b, c, x[ 8], S32, 0x8771f681); /* 34 */ |
||||||
|
HH (c, d, a, b, x[11], S33, 0x6d9d6122); /* 35 */ |
||||||
|
HH (b, c, d, a, x[14], S34, 0xfde5380c); /* 36 */ |
||||||
|
HH (a, b, c, d, x[ 1], S31, 0xa4beea44); /* 37 */ |
||||||
|
HH (d, a, b, c, x[ 4], S32, 0x4bdecfa9); /* 38 */ |
||||||
|
HH (c, d, a, b, x[ 7], S33, 0xf6bb4b60); /* 39 */ |
||||||
|
HH (b, c, d, a, x[10], S34, 0xbebfbc70); /* 40 */ |
||||||
|
HH (a, b, c, d, x[13], S31, 0x289b7ec6); /* 41 */ |
||||||
|
HH (d, a, b, c, x[ 0], S32, 0xeaa127fa); /* 42 */ |
||||||
|
HH (c, d, a, b, x[ 3], S33, 0xd4ef3085); /* 43 */ |
||||||
|
HH (b, c, d, a, x[ 6], S34, 0x4881d05); /* 44 */ |
||||||
|
HH (a, b, c, d, x[ 9], S31, 0xd9d4d039); /* 45 */ |
||||||
|
HH (d, a, b, c, x[12], S32, 0xe6db99e5); /* 46 */ |
||||||
|
HH (c, d, a, b, x[15], S33, 0x1fa27cf8); /* 47 */ |
||||||
|
HH (b, c, d, a, x[ 2], S34, 0xc4ac5665); /* 48 */ |
||||||
|
|
||||||
|
/* Round 4 */ |
||||||
|
II (a, b, c, d, x[ 0], S41, 0xf4292244); /* 49 */ |
||||||
|
II (d, a, b, c, x[ 7], S42, 0x432aff97); /* 50 */ |
||||||
|
II (c, d, a, b, x[14], S43, 0xab9423a7); /* 51 */ |
||||||
|
II (b, c, d, a, x[ 5], S44, 0xfc93a039); /* 52 */ |
||||||
|
II (a, b, c, d, x[12], S41, 0x655b59c3); /* 53 */ |
||||||
|
II (d, a, b, c, x[ 3], S42, 0x8f0ccc92); /* 54 */ |
||||||
|
II (c, d, a, b, x[10], S43, 0xffeff47d); /* 55 */ |
||||||
|
II (b, c, d, a, x[ 1], S44, 0x85845dd1); /* 56 */ |
||||||
|
II (a, b, c, d, x[ 8], S41, 0x6fa87e4f); /* 57 */ |
||||||
|
II (d, a, b, c, x[15], S42, 0xfe2ce6e0); /* 58 */ |
||||||
|
II (c, d, a, b, x[ 6], S43, 0xa3014314); /* 59 */ |
||||||
|
II (b, c, d, a, x[13], S44, 0x4e0811a1); /* 60 */ |
||||||
|
II (a, b, c, d, x[ 4], S41, 0xf7537e82); /* 61 */ |
||||||
|
II (d, a, b, c, x[11], S42, 0xbd3af235); /* 62 */ |
||||||
|
II (c, d, a, b, x[ 2], S43, 0x2ad7d2bb); /* 63 */ |
||||||
|
II (b, c, d, a, x[ 9], S44, 0xeb86d391); /* 64 */ |
||||||
|
|
||||||
|
state[0] += a; |
||||||
|
state[1] += b; |
||||||
|
state[2] += c; |
||||||
|
state[3] += d; |
||||||
|
|
||||||
|
// Zeroize sensitive information.
|
||||||
|
memset(x, 0, sizeof x); |
||||||
|
} |
||||||
|
|
||||||
|
//////////////////////////////
|
||||||
|
|
||||||
|
// MD5 block update operation. Continues an MD5 message-digest
|
||||||
|
// operation, processing another message block
|
||||||
|
void MD5::update(const unsigned char input[], size_type length) |
||||||
|
{ |
||||||
|
// compute number of bytes mod 64
|
||||||
|
size_type index = count[0] / 8 % blocksize; |
||||||
|
|
||||||
|
// Update number of bits
|
||||||
|
if ((count[0] += (length << 3)) < (length << 3)) |
||||||
|
count[1]++; |
||||||
|
count[1] += (length >> 29); |
||||||
|
|
||||||
|
// number of bytes we need to fill in buffer
|
||||||
|
size_type firstpart = 64 - index; |
||||||
|
|
||||||
|
size_type i; |
||||||
|
|
||||||
|
// transform as many times as possible.
|
||||||
|
if (length >= firstpart) |
||||||
|
{ |
||||||
|
// fill buffer first, transform
|
||||||
|
memcpy(&buffer[index], input, firstpart); |
||||||
|
transform(buffer); |
||||||
|
|
||||||
|
// transform chunks of blocksize (64 bytes)
|
||||||
|
for (i = firstpart; i + blocksize <= length; i += blocksize) |
||||||
|
transform(&input[i]); |
||||||
|
|
||||||
|
index = 0; |
||||||
|
} |
||||||
|
else |
||||||
|
i = 0; |
||||||
|
|
||||||
|
// buffer remaining input
|
||||||
|
memcpy(&buffer[index], &input[i], length-i); |
||||||
|
} |
||||||
|
|
||||||
|
//////////////////////////////
|
||||||
|
|
||||||
|
// for convenience provide a verson with signed char
|
||||||
|
void MD5::update(const char input[], size_type length) |
||||||
|
{ |
||||||
|
update((const unsigned char*)input, length); |
||||||
|
} |
||||||
|
|
||||||
|
//////////////////////////////
|
||||||
|
|
||||||
|
// MD5 finalization. Ends an MD5 message-digest operation, writing the
|
||||||
|
// the message digest and zeroizing the context.
|
||||||
|
MD5& MD5::finalize() |
||||||
|
{ |
||||||
|
static unsigned char padding[64] = { |
||||||
|
0x80, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, |
||||||
|
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, |
||||||
|
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 |
||||||
|
}; |
||||||
|
|
||||||
|
if (!finalized) { |
||||||
|
// Save number of bits
|
||||||
|
unsigned char bits[8]; |
||||||
|
encode(bits, count, 8); |
||||||
|
|
||||||
|
// pad out to 56 mod 64.
|
||||||
|
size_type index = count[0] / 8 % 64; |
||||||
|
size_type padLen = (index < 56) ? (56 - index) : (120 - index); |
||||||
|
update(padding, padLen); |
||||||
|
|
||||||
|
// Append length (before padding)
|
||||||
|
update(bits, 8); |
||||||
|
|
||||||
|
// Store state in digest
|
||||||
|
encode(digest, state, 16); |
||||||
|
|
||||||
|
// Zeroize sensitive information.
|
||||||
|
memset(buffer, 0, sizeof buffer); |
||||||
|
memset(count, 0, sizeof count); |
||||||
|
|
||||||
|
finalized=true; |
||||||
|
} |
||||||
|
|
||||||
|
return *this; |
||||||
|
} |
||||||
|
|
||||||
|
//////////////////////////////
|
||||||
|
|
||||||
|
// return hex representation of digest as string
|
||||||
|
std::string MD5::hexdigest() const |
||||||
|
{ |
||||||
|
if (!finalized) |
||||||
|
return ""; |
||||||
|
|
||||||
|
char buf[33]; |
||||||
|
for (int i=0; i<16; i++) |
||||||
|
sprintf(buf+i*2, "%02x", digest[i]); |
||||||
|
buf[32]=0; |
||||||
|
|
||||||
|
return std::string(buf); |
||||||
|
} |
||||||
|
|
||||||
|
//////////////////////////////
|
||||||
|
|
||||||
|
std::ostream& operator<<(std::ostream& out, MD5 md5) |
||||||
|
{ |
||||||
|
return out << md5.hexdigest(); |
||||||
|
} |
||||||
|
|
||||||
|
//////////////////////////////
|
||||||
|
|
||||||
|
std::string md5(const std::string str) |
||||||
|
{ |
||||||
|
MD5 md5 = MD5(str); |
||||||
|
|
||||||
|
return md5.hexdigest(); |
||||||
|
} |
||||||
Loading…
Reference in new issue