acouzens
/
open5gs
1
0
Fork 1
Code Issues Pull Requests 1 Projects Releases Wiki Activity

update it

This commit is contained in:
Sukchan Lee 2017-03-03 16:22:09 +09:00
parent 3a93815426
commit 5e0a045a0d
6 changed files with 56 additions and 49 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lib/s6a/Makefile.am
View File

@ -5,11 +5,11 @@ SUBDIRS = freeDiameter
noinst_LTLIBRARIES = libs6a.la
libs6a_la_SOURCES = \
milenage.h \
milenage.h 3gpp_kdf.h \
s6a_app.h
nodist_libs6a_la_SOURCES = \
milenage.c \
milenage.c 3gpp_kdf.c \
s6a_init.c s6a_config.c s6a_fd.c s6a_app.c s6a_dict.c \
hss_main.c hss_ctx.c \
mme_init.c

5
lib/s6a/hss_ctx.c
View File

@ -30,7 +30,7 @@ status_t hss_ctx_init(void)
memset(&self, 0, sizeof(hss_ctx_t));
memcpy(self.op, core_ascii_to_hex(OP, strlen(OP), buf), MAX_KEY_LEN);
memcpy(self.amf, core_ascii_to_hex(AMF, strlen(AMF), buf), MAX_KEY_LEN);
memcpy(self.amf, core_ascii_to_hex(AMF, strlen(AMF), buf), MAX_AMF_LEN);
return CORE_OK;
}
@ -55,6 +55,9 @@ ue_ctx_t* hss_ue_ctx_add()
/* Add new eNB context to list */
list_append(&g_ue_list, ue);
memcpy(ue->op, self.op, MAX_KEY_LEN);
memcpy(ue->amf, self.amf, MAX_AMF_LEN);
return ue;
}

6
lib/s6a/hss_ctx.h
View File

@ -21,14 +21,16 @@ typedef struct _ue_ctx_t {
c_uint8_t imsi_len;
c_uint8_t k[MAX_KEY_LEN];
c_uint64_t seq;
c_uint64_t sqn;
c_uint8_t rand[MAX_KEY_LEN];
c_uint8_t opc[MAX_KEY_LEN];
c_uint8_t op[MAX_KEY_LEN];
c_uint8_t amf[MAX_AMF_LEN];
} ue_ctx_t;
typedef struct _hss_ctx_t {
c_uint8_t op[MAX_KEY_LEN];
c_uint8_t amf[MAX_KEY_LEN];
c_uint8_t amf[MAX_AMF_LEN];
} hss_ctx_t;
CORE_DECLARE(hss_ctx_t*) hss_self(void);

61
lib/s6a/hss_main.c
View File

@ -4,11 +4,17 @@
#include "core_pool.h"
#include "core_lib.h"
#include "3gpp_kdf.h"
#include "milenage.h"
#include "hss_ctx.h"
#include "s6a_app.h"
#define MAX_SQN_LEN 6
#define MAX_AK_LEN 6
#define MAX_XRES_LEN 8
#define MAX_KASME_LEN 32
static struct disp_hdl *hdl_fb = NULL; /* handler for fallback cb */
static struct disp_hdl *hdl_air = NULL; /* handler for Auth-Info-Request cb */
@ -32,12 +38,15 @@ static int hss_air_cb( struct msg **msg, struct avp *avp,
union avp_value val;
ue_ctx_t *ue = NULL;
c_uint8_t mac_a[8];
c_uint8_t seq[6];
c_uint8_t xres[8];
c_uint8_t ak[6];
c_uint8_t autn[16];
int i;
c_uint8_t sqn[MAX_SQN_LEN];
c_uint8_t autn[MAX_KEY_LEN];
c_uint8_t ik[MAX_KEY_LEN];
c_uint8_t ck[MAX_KEY_LEN];
c_uint8_t ak[MAX_AK_LEN];
c_uint8_t xres[MAX_XRES_LEN];
c_uint8_t kasme[MAX_KASME_LEN];
size_t xres_len;
c_uint8_t plmn[3];
d_assert(msg, return EINVAL,);
@ -60,17 +69,13 @@ static int hss_air_cb( struct msg **msg, struct avp *avp,
}
core_generate_random_bytes(ue->rand, MAX_KEY_LEN);
milenage_opc(ue->k, hss_self()->op, ue->opc);
milenage_f1(ue->opc, ue->k, ue->rand, core_uint64_to_array(seq, ue->seq),
hss_self()->amf, mac_a, NULL);
milenage_f2345(ue->opc, ue->k, ue->rand, xres, NULL, NULL, ak, NULL);
milenage_opc(ue->k, ue->op, ue->opc);
milenage_generate(ue->opc, ue->amf, ue->k,
core_uint64_to_array(sqn, ue->sqn), ue->rand,
autn, ik, ck, ak, xres, &xres_len);
derive_kasme(ck, ik, plmn, sqn, ak, kasme);
for ( i = 0; i < 6; i++)
autn[i] = seq[i] ^ ak[i];
memcpy(&autn[6], hss_self()->amf, 2);
memcpy(&autn[7], mac_a, 8);
ue->seq = (ue->seq + 32) & 0x7ffffffffff;
ue->sqn = (ue->sqn + 32) & 0x7ffffffffff;
/* Set the Origin-Host, Origin-Realm, andResult-Code AVPs */
d_assert(fd_msg_rescode_set(ans, "DIAMETER_SUCCESS", NULL, NULL, 1) == 0,
@ -86,34 +91,30 @@ static int hss_air_cb( struct msg **msg, struct avp *avp,
d_assert(fd_msg_avp_new(s6a_authentication_info, 0, &avp) == 0, goto out,);
d_assert(fd_msg_avp_new(s6a_e_utran_vector, 0, &avpch1) == 0, goto out,);
#define TEST_RAND "RAND_123456"
d_assert(fd_msg_avp_new(s6a_rand, 0, &avpch2) == 0, goto out,);
val.os.data = (unsigned char*)TEST_RAND;
val.os.len = strlen(TEST_RAND);
val.os.data = ue->rand;
val.os.len = MAX_KEY_LEN;
d_assert(fd_msg_avp_setvalue(avpch2, &val) == 0, goto out,);
d_assert(fd_msg_avp_add(avpch1, MSG_BRW_LAST_CHILD, avpch2) == 0,
goto out,);
#define TEST_XRES "XRES_123456"
d_assert(fd_msg_avp_new(s6a_xres, 0, &avpch2) == 0, goto out,);
val.os.data = (unsigned char*)TEST_XRES;
val.os.len = strlen(TEST_XRES);
val.os.data = xres;
val.os.len = xres_len;
d_assert(fd_msg_avp_setvalue(avpch2, &val) == 0, goto out,);
d_assert(fd_msg_avp_add(avpch1, MSG_BRW_LAST_CHILD, avpch2) == 0,
goto out,);
#define TEST_AUTH "AUTH_123456"
d_assert(fd_msg_avp_new(s6a_autn, 0, &avpch2) == 0, goto out,);
val.os.data = (unsigned char*)TEST_AUTH;
val.os.len = strlen(TEST_AUTH);
val.os.data = autn;
val.os.len = MAX_KEY_LEN;
d_assert(fd_msg_avp_setvalue(avpch2, &val) == 0, goto out,);
d_assert(fd_msg_avp_add(avpch1, MSG_BRW_LAST_CHILD, avpch2) == 0,
goto out,);
#define TEST_KASME "KASME_123456"
d_assert(fd_msg_avp_new(s6a_kasme, 0, &avpch2) == 0, goto out,);
val.os.data = (unsigned char*)TEST_KASME;
val.os.len = strlen(TEST_KASME);
val.os.data = kasme;
val.os.len = MAX_KASME_LEN;
d_assert(fd_msg_avp_setvalue(avpch2, &val) == 0, goto out,);
d_assert(fd_msg_avp_add(avpch1, MSG_BRW_LAST_CHILD, avpch2) == 0,
goto out,);
@ -159,7 +160,7 @@ int hss_init(void)
strcpy((char*)ue->imsi, UE1_IMSI);
ue->imsi_len = strlen(UE1_IMSI);
memcpy(ue->k, core_ascii_to_hex(K, strlen(K), buf), MAX_KEY_LEN);
ue->seq = 32;
ue->sqn = 32;
ue = hss_ue_ctx_add();
d_assert(ue, return -1, "UE context add failed");
@ -167,7 +168,7 @@ int hss_init(void)
strcpy((char*)ue->imsi, UE2_IMSI);
ue->imsi_len = strlen(UE2_IMSI);
memcpy(ue->k, core_ascii_to_hex(K, strlen(K), buf), MAX_KEY_LEN);
ue->seq = 32;
ue->sqn = 32;
}
memset(&data, 0, sizeof(data));

14
lib/s6a/milenage.c
View File

@ -18,6 +18,7 @@
#include "core_debug.h"
#include "core_aes.h"
#include "core_sha2.h"
#include "milenage.h"
@ -190,11 +191,11 @@ int milenage_f2345(const c_uint8_t *opc, const c_uint8_t *k,
*/
void milenage_generate(const c_uint8_t *opc, const c_uint8_t *amf,
const c_uint8_t *k, const c_uint8_t *sqn, const c_uint8_t *_rand,
c_uint8_t *autn, c_uint8_t *ik, c_uint8_t *ck, c_uint8_t *res,
size_t *res_len)
c_uint8_t *autn, c_uint8_t *ik, c_uint8_t *ck, c_uint8_t *ak,
c_uint8_t *res, size_t *res_len)
{
int i;
c_uint8_t mac_a[8], ak[6];
c_uint8_t mac_a[8];
if (*res_len < 8) {
*res_len = 0;
@ -342,17 +343,14 @@ int milenage_check(const c_uint8_t *opc, const c_uint8_t *k,
return 0;
}
int milenage_opc(const c_uint8_t *k, const c_uint8_t *op, c_uint8_t *opc)
void milenage_opc(const c_uint8_t *k, const c_uint8_t *op, c_uint8_t *opc)
{
int i;
if (aes_128_encrypt_block(k, op, opc))
return -1;
aes_128_encrypt_block(k, op, opc);
for (i = 0; i < 16; i++)
{
opc[i] ^= op[i];
}
return 0;
}

15
lib/s6a/milenage.h
View File

@ -6,15 +6,15 @@
* See README for more details.
*/
#ifndef MILENAGE_H
#define MILENAGE_H
#ifndef __MILENAGE_H__
#define __MILENAGE_H__
#include "core.h"
void milenage_generate(const c_uint8_t *opc, const c_uint8_t *amf,
const c_uint8_t *k, const c_uint8_t *sqn, const c_uint8_t *_rand,
c_uint8_t *autn, c_uint8_t *ik, c_uint8_t *ck, c_uint8_t *res,
size_t *res_len);
c_uint8_t *autn, c_uint8_t *ik, c_uint8_t *ck, c_uint8_t *ak,
c_uint8_t *res, size_t *res_len);
int milenage_auts(const c_uint8_t *opc, const c_uint8_t *k,
const c_uint8_t *_rand, const c_uint8_t *auts, c_uint8_t *sqn);
int gsm_milenage(const c_uint8_t *opc, const c_uint8_t *k,
@ -30,6 +30,9 @@ int milenage_f2345(const c_uint8_t *opc, const c_uint8_t *k,
const c_uint8_t *_rand, c_uint8_t *res, c_uint8_t *ck, c_uint8_t *ik,
c_uint8_t *ak, c_uint8_t *akstar);
int milenage_opc(const c_uint8_t *k, const c_uint8_t *op, c_uint8_t *opc);
void milenage_opc(const c_uint8_t *k, const c_uint8_t *op, c_uint8_t *opc);
void derive_kasme(const c_uint8_t *ck, const c_uint8_t *ik,
const c_uint8_t plmn[3], const c_uint8_t *sqn, const c_uint8_t *ak,
c_uint8_t *kasme);
#endif /* MILENAGE_H */
#endif /* __MILENAGE_H__ */