softbus_standard/common/net_trans_common.c

/*
 * Copyright (c) 2021 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "net_trans_common.h"

static int ONE_SECOND = 1;
static int32_t g_currentSessionId4Data = -1;
static int32_t g_currentSessionId4Ctl = -1;
static int32_t g_currentSessionId4Proxy = -1;
static int32_t g_waitFlag = WAIT_DEF_VALUE;
static int32_t g_waitFlag4Data = WAIT_DEF_VALUE;
static int32_t g_waitFlag4Ctl = WAIT_DEF_VALUE;
static int32_t g_waitFlag4CtlClose = WAIT_DEF_VALUE;
static int32_t g_waitFlag4Proxy = WAIT_DEF_VALUE;
static int32_t g_waitFlag4Stream = WAIT_DEF_VALUE;
static int32_t g_waitFlag4DataClose = WAIT_DEF_VALUE;
static int32_t g_passiveOpenRetFlag = SOFTBUS_OK;
static int32_t g_nodeOnlineCount = 0;
static int32_t g_nodeOfflineCount = 0;
static int32_t g_msgCount4Data = 0;
static int32_t g_byteCount4Data = 0;
static int32_t g_sessionCloseCount4Data = 0;
static int32_t g_sessionCloseCount4Ctrl = 0;
static int32_t g_sessionOpenCount4Data = 0;
static int32_t g_sessionOpenCount4Ctrl = 0;
static int32_t g_sessionOpenCount4Proxy = 0;

static SessionAttribute* g_sessionAttr4Data = NULL;
static SessionAttribute* g_sessionAttr4Ctl = NULL;
static SessionAttribute* g_sessionAttr4Pass = NULL;
static SessionAttribute* g_sessionAttr4Perf = NULL;
static SessionAttribute* g_sessionAttr4Proxy = NULL;
static SessionAttribute* g_p2pattribute = NULL;
static SessionAttribute* g_p2pattributeProxy = NULL;
static ISessionListener* g_sessionlistener4Data = NULL;
static ISessionListener* g_sessionlistener4Ctl = NULL;
static ISessionListener* g_sessionlistener4Pass = NULL;
static ISessionListener* g_sessionlistener4Perf = NULL;
static ISessionListener* g_sessionlistener4Proxy = NULL;
static ISessionListener* g_sessionlistener4Stream = NULL;

static IFileSendListener* g_fileSendListener = NULL;
static IFileReceiveListener* g_fileRecvListener = NULL;

static char g_networkId[NETWORK_ID_BUF_LEN] = { 0 };
static INodeStateCb g_defNodeStateCallback;

static int* g_sId4Task2;
static int* g_sId4Task3;
static char g_fillContentChar = 'd';
static unsigned int g_expectDataSize = 0;
static char* g_expectDataContent = NULL;
static char* g_expectMessageContent = NULL;
static int32_t g_recvMsgStat4Control[MAX_SESSION_NUM] = { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 };
static int32_t g_recvByteStat4Control[MAX_SESSION_NUM] = { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 };
static uint64_t g_transTimeEnd;
static uint64_t g_tokenId;
static pthread_barrier_t* g_barrier = NULL;

void sleepn(int n)
{
    for(int i = 0; i < n; i++) {
        sleep(1);
    }
}
int Wait(int timeout)
{
    LOG("start wait,timeout:%d", timeout);
    int count = 0;
    int t = timeout;
    while (t > 0) {
        sleep(1);
        if (g_waitFlag != WAIT_DEF_VALUE) {
            LOG("Wait success[flag:%d] [time:%d]", g_waitFlag, count);
            break;
        }
        t--;
        count++;
    }
    if (g_waitFlag != WAIT_SUCCESS_VALUE) {
        LOG("Wait fail[exp:%d, real:%d][used time:%d]", WAIT_SUCCESS_VALUE, g_waitFlag, count);
        return SOFTBUS_ERR;
    }
    return SOFTBUS_OK;
}

int Wait4Session(int timeout, WaitSessionType type)
{
    int hitFlag = -1;
    int t = timeout;
    while (t > 0) {
        sleep(1);
        switch (type) {
            case SESSION_4CTL:
                if (g_waitFlag4Ctl != WAIT_DEF_VALUE) {
                    LOG("Wait4Session[ctrl] succ, flag:%d", g_waitFlag4Ctl);
                    hitFlag = 1;
                }
                break;
            case SESSION_4DATA:
                if (g_waitFlag4Data != WAIT_DEF_VALUE) {
                    LOG("Wait4Session[data] succ,flag:%d", g_waitFlag4Data);
                    hitFlag = 1;
                }
                break;
            case SESSION_4PROXY:
                if (g_waitFlag4Proxy != WAIT_DEF_VALUE) {
                    LOG("Wait4Session[proxy] succ,flag:%d", g_waitFlag4Proxy);
                    hitFlag = 1;
                }
                break;
            case SESSION_4STREAM:
                if (g_waitFlag4Stream != WAIT_DEF_VALUE) {
                    LOG("Wait4Session[proxy] succ,flag:%d", g_waitFlag4Stream);
                    hitFlag = 1;
                }
                break;
            default:
                LOG("Wait4Session type error");
                hitFlag = 1;
                break;
        }
        if (hitFlag != -1) {
            break;
        }
        t--;
    }
    switch (type) {
        case SESSION_4CTL:
            if (g_waitFlag4Ctl != WAIT_SUCCESS_VALUE) {
                LOG("Wait4Session[ctrl] fail[exp:%d, real:%d]", WAIT_SUCCESS_VALUE, g_waitFlag4Ctl);
                return SOFTBUS_ERR;
            }
            break;
        case SESSION_4DATA:
            if (g_waitFlag4Data != WAIT_SUCCESS_VALUE) {
                LOG("Wait4Session[data] fail[exp:%d, real:%d]", WAIT_SUCCESS_VALUE, g_waitFlag4Data);
                return SOFTBUS_ERR;
            }
            break;
        case SESSION_4PROXY:
            if (g_waitFlag4Proxy != WAIT_SUCCESS_VALUE) {
                LOG("Wait4Session[proxy] fail[exp:%d, real:%d]", WAIT_SUCCESS_VALUE, g_waitFlag4Proxy);
                return SOFTBUS_ERR;
            }
            break;
         case SESSION_4STREAM:
            if (g_waitFlag4Stream != WAIT_SUCCESS_VALUE) {
                LOG("Wait4Session[proxy] fail[exp:%d, real:%d]", WAIT_SUCCESS_VALUE, g_waitFlag4Stream);
                return SOFTBUS_ERR;
            }
            break;
        default:
            return SOFTBUS_ERR;
    }
    return SOFTBUS_OK;
}

uint64_t GetCurrentTimeOfMs(void)
{
    int unit = 1000;
    struct timeval tv;
    gettimeofday(&tv, NULL);
    return (tv.tv_sec * unit + tv.tv_usec / unit);
}

char* GetSoftbusPid(void)
{
    FILE* file = NULL;
    int buffSize = 20;
    char* buffer = (char*)malloc(buffSize);
    if (buffer == NULL) {
        LOG("malloc fail");
        return NULL;
    }

    file = popen(CMD_PIDOF_SOFTBUS, "r");
    if (file == NULL) {
        LOG("call popen fail");
        free(buffer);
        return NULL;
    }
    (void)fgets(buffer, buffSize, file);
    LOG("##################Softbus PID:%s", buffer);
    pclose(file);

    return buffer;
}

static int OnReceiveFileStarted(int sessionId, const char* files, int fileCnt)
{
    LOG("[recv file]start,sid:%d, fileCnt:%d", sessionId, fileCnt);
    return 0;
}

static int OnReceiveFileProcess(int sessionId, const char* firstFile, uint64_t bytesUpload, uint64_t bytesTotal)
{
    return 0;
}

static void OnReceiveFileFinished(int sessionId, const char* files, int fileCnt)
{
    LOG("[recv file]finish,sid:%d, fileCnt:%d", sessionId, fileCnt);
}

static void OnRecvFileTransError(int sessionId)
{
    LOG("[recv file]trans error,sid:%d", sessionId);
}

static int OnSendFileProcess(int sessionId, uint64_t bytesUpload, uint64_t bytesTotal)
{
    return 0;
}

static int OnSendFileFinished(int sessionId, const char* firstFile)
{
    g_waitFlag = WAIT_SUCCESS_VALUE;
    LOG("[send file]finish,sid:%d, firstFile:%s\n", sessionId, firstFile);
    return 0;
}

static void OnSendFileTransError(int sessionId)
{
    g_waitFlag = WAIT_FAIL_VALUE;
    LOG("[send file]trans error,sid = %d\n", sessionId);
}

static void OnDefNodeOnline(NodeBasicInfo* info)
{
    if (info == NULL) {
        LOG("[cb]Online: info is null");
        return;
    }
    (void)strncpy_s(g_networkId, NETWORK_ID_BUF_LEN, info->networkId, NETWORK_ID_BUF_LEN);
    LOG("[cb]Online id:%s, name:%s, type id:%u", info->networkId, info->deviceName, info->deviceTypeId);
    g_nodeOnlineCount++;
}

static void OnDefNodeOffline(NodeBasicInfo* info)
{
    if (info == NULL) {
        LOG("[cb]Offline: info is null");
        return;
    }
    LOG("[cb]Offline id:%s, name:%s, type id:%u", info->networkId, info->deviceName, info->deviceTypeId);
    g_nodeOfflineCount++;
}

static void OnDefNodeBasicInfoChanged(NodeBasicInfoType type, NodeBasicInfo* info)
{
    if (info == NULL) {
        LOG("[cb]InfoChanged: info is null, type[%d]", type);
        return;
    }
    LOG("[cb]InfoChanged id: %s,name: %s", info->networkId, info->deviceName);
}

static void onDefNodeStatusChanged(NodeStatusType type, NodeStatus *status)
{
    if (status == NULL) {
        LOG("[cb]StatusChanged: info is null, type[%d]", type);
        return;
    }
    LOG("[cb]StatusChanged id: %s,status: %d", status->basicInfo.networkId, status->authStatus);
}

static int DataSessionOpened(int sessionId, int result)
{
    // wait 1s, ensure set current session id
    sleep(1);
    g_sessionOpenCount4Data++;
    if (result == SOFTBUS_OK) {
        if (sessionId == g_currentSessionId4Data) {
            LOG("[cb][data]open session check success sid[%d]", sessionId);
            g_waitFlag4Data = WAIT_SUCCESS_VALUE;
        } else {
            LOG("[cb][data]open session callback sid[%d] not match open "
                "sid[%d]",
                sessionId, g_currentSessionId4Data);
            g_waitFlag4Data = WAIT_FAIL_VALUE;
        }
    } else {
        g_waitFlag4Data = WAIT_FAIL_VALUE;
    }
    return SOFTBUS_OK;
}

static void DataSessionClosed(int sessionId)
{
    g_sessionCloseCount4Data++;
    if (sessionId == g_currentSessionId4Data) {
        LOG("[cb][data]closed session,check sid[%d] success", sessionId);
        g_currentSessionId4Data = -1;
        g_waitFlag4DataClose = WAIT_SUCCESS_VALUE;
    } else {
        LOG("[cb][data]closed session, callback sid[%d] not match open sid[%d]", sessionId, g_currentSessionId4Data);
        g_waitFlag4DataClose = WAIT_FAIL_VALUE;
    }
}

static void CheckReceiveData(int sessionId, const void* data, unsigned int dataLen)
{
    // check session id
    if (sessionId == g_currentSessionId4Data && dataLen == g_expectDataSize) {
        LOG("[check]sid/size ok[sid:%d,size:%u]", sessionId, dataLen);
    } else {
        LOG("[check]sid/size fail[sid exp:%d,real:%d][size exp:%u,real:%u]", g_currentSessionId4Data, sessionId,
            g_expectDataSize, dataLen);
        g_waitFlag = WAIT_FAIL_VALUE;
        return;
    }

    // check data content
    int cmpRst = strncmp(g_expectDataContent, (char*)data, dataLen);
    if (cmpRst == 0) {
        LOG("[check] cmp content ok");
        g_waitFlag = WAIT_SUCCESS_VALUE;
    } else {
        LOG("[check] cmp content fail[exp:%s,real:%s]", g_expectDataContent, (char*)data);
        g_waitFlag = WAIT_FAIL_VALUE;
    }
}

static void DataBytesReceived(int sessionId, const void* data, unsigned int dataLen)
{
    LOG("[cb][data]ByteRec start");
    CheckReceiveData(sessionId, data, dataLen);
    g_byteCount4Data++;
    LOG("[cb][data]ByteRec end");
}

static void DataMessageReceived(int sessionId, const void* data, unsigned int dataLen)
{
    LOG("[cb][data]MessageRec start");
    CheckReceiveData(sessionId, data, dataLen);
    g_msgCount4Data++;
    LOG("[cb][data]MessageRec end");
}

static int ControlSessionOpened(int sessionId, int result)
{
    // wait 1s, ensure set current session id
    sleep(1);
    g_sessionOpenCount4Ctrl++;
    if (result == SOFTBUS_OK) {
        if (sessionId == g_currentSessionId4Ctl) {
            LOG("[cb][ctrl]open session check success sid[%d]", sessionId);
            g_waitFlag4Ctl = WAIT_SUCCESS_VALUE;
        } else {
            LOG("[cb][ctrl]open session callback sid[%d] not match open "
                "sid[%d]",
                sessionId, g_currentSessionId4Ctl);
            g_waitFlag4Ctl = WAIT_FAIL_VALUE;
        }
    } else {
        g_waitFlag4Ctl = WAIT_FAIL_VALUE;
    }
    return SOFTBUS_OK;
}

static void ControlSessionClosed(int sessionId)
{
    g_sessionCloseCount4Ctrl++;
    if (sessionId == g_currentSessionId4Ctl) {
        LOG("[cb][ctrl]closed session check sid[%d] success", sessionId);
        g_currentSessionId4Ctl = -1;
        g_waitFlag4CtlClose = WAIT_SUCCESS_VALUE;
    } else {
        LOG("[cb][ctrl]closed session callback sid[%d] not match open sid[%d]", sessionId, g_currentSessionId4Ctl);
        g_waitFlag4CtlClose = WAIT_FAIL_VALUE;
    }
}

static void ControlBytesReceived(int sessionId, const void* data, unsigned int dataLen)
{
    LOG("[cb][ctrl]ByteRec sid:%d, data len:%u", sessionId, dataLen);
    if (sessionId < 0 || sessionId > MAX_SESSION_NUM) {
        LOG("[cb][ctrl]ByteRec invalid session sid[%d]", sessionId);
        return;
    }
    if (data == NULL) {
        LOG("[cb][ctrl]ByteRec invalid data=null sid[%d]", sessionId);
        return;
    }
    int sid = sessionId % MAX_SESSION_NUM;
    g_recvByteStat4Control[sid]++;
}

static void ControlMessageReceived(int sessionId, const void* data, unsigned int dataLen)
{
    LOG("[cb][ctrl]MessageRec sid:%d, data len:%u", sessionId, dataLen);
    if (sessionId < 0 || sessionId > MAX_SESSION_NUM) {
        LOG("[cb][ctrl]MessageRec invalid session sid[%d]", sessionId);
        return;
    }
    if (data == NULL) {
        LOG("[cb][ctrl]MessageRec invalid data=null sid[%d]", sessionId);
        return;
    }
    int sid = sessionId % MAX_SESSION_NUM;
    g_recvMsgStat4Control[sid]++;
}

static int PassiveSessionOpened(int sessionId, int result)
{
    LOG("[cb][pass]open session passive sid[%d],rst[%d]", sessionId, result);
    if (result == SOFTBUS_OK) {
        g_waitFlag = WAIT_SUCCESS_VALUE;
    } else {
        g_waitFlag = WAIT_FAIL_VALUE;
    }
    if (g_passiveOpenRetFlag == SOFTBUS_ERR) {
        LOG("[cb][pass]return fail");
        g_passiveOpenRetFlag = SOFTBUS_OK;
        return SOFTBUS_ERR;
    } else {
        return SOFTBUS_OK;
    }
}

static void PassiveSessionClosed(int sessionId)
{
    LOG("[cb][pass]close session passive sid[%d]", sessionId);
    g_waitFlag = WAIT_SUCCESS_VALUE;
}

static void PassiveBytesReceived(int sessionId, const void* data, unsigned int dataLen)
{
    LOG("[cb][pass]ByteRec sid:%d, data len:%u", sessionId, dataLen);
    if (data == NULL) {
        LOG("[cb][pass]ByteRec invalid data=null sid[%d]", sessionId);
        g_waitFlag = WAIT_FAIL_VALUE;
        return;
    }
    int ret = SendBytes(sessionId, data, dataLen);
    LOG("[cb][pass]ByteRec send back ret[%d]", ret);
    if (ret == SOFTBUS_OK) {
        g_waitFlag = WAIT_SUCCESS_VALUE;
    } else {
        g_waitFlag = WAIT_FAIL_VALUE;
    }
}

static void PassiveMessageReceived(int sessionId, const void* data, unsigned int dataLen)
{
    LOG("[cb][pass]MessageRec sid:%d, data len:%u", sessionId, dataLen);
    if (data == NULL) {
        LOG("[cb][pass]MessageRec invalid data=null sid[%d]", sessionId);
        g_waitFlag = WAIT_FAIL_VALUE;
        return;
    }
    int ret = SendMessage(sessionId, data, dataLen);
    LOG("[cb][pass]MessageRec send back ret[%d]", ret);
    if (ret == SOFTBUS_OK) {
        g_waitFlag = WAIT_SUCCESS_VALUE;
    } else {
        g_waitFlag = WAIT_FAIL_VALUE;
    }
}

static int PerfSessionOpened(int sessionId, int result)
{
    if (result == SOFTBUS_OK) {
        g_waitFlag = WAIT_SUCCESS_VALUE;
    } else {
        LOG("[cb][perf]open session result fail, ret:%d", result);
        g_waitFlag = WAIT_FAIL_VALUE;
    }
    return SOFTBUS_OK;
}

static void PerfSessionClosed(int sessionId)
{
    LOG("[cb][perf]closed session,sid[%d]", sessionId);
}

static void PerfBytesReceived(int sessionId, const void* data, unsigned int dataLen)
{
    g_transTimeEnd = GetCurrentTimeOfMs();

    LOG("[cb][perf]Byte recv");
    g_waitFlag = WAIT_SUCCESS_VALUE;
}

static void PerfMessageReceived(int sessionId, const void* data, unsigned int dataLen)
{
    LOG("[cb][perf]Message recv");
}

static int ProxySessionOpened(int sessionId, int result)
{
    LOG("[cb][proxy]open session proxy sid[%d],rst[%d]", sessionId, result);
    if (result == SOFTBUS_OK) {
        g_waitFlag4Proxy = WAIT_SUCCESS_VALUE;
    } else {
        g_waitFlag4Proxy = WAIT_FAIL_VALUE;
    }
    return SOFTBUS_OK;
}

static void ProxySessionClosed(int sessionId)
{
    LOG("[cb][proxy]close session proxy sid[%d]", sessionId);
    g_waitFlag4Proxy = WAIT_SUCCESS_VALUE;
}

static void ProxyBytesReceived(int sessionId, const void* data, unsigned int dataLen)
{
    LOG("[cb][proxy]ByteRec sid:%d, data len:%u", sessionId, dataLen);
    if (data == NULL) {
        LOG("[cb][proxy]ByteRec invalid data=null sid:[%d]", sessionId);
        g_waitFlag4Proxy = WAIT_FAIL_VALUE;
    } else {
        g_waitFlag4Proxy = WAIT_SUCCESS_VALUE;
    }
}

static void ProxyMessageReceived(int sessionId, const void* data, unsigned int dataLen)
{
    LOG("[cb][proxy]MessageRec sid:%d, data len:%u", sessionId, dataLen);
    if (data == NULL) {
        LOG("[cb][proxy]MessageRec invalid data=null sid[%d]", sessionId);
        g_waitFlag4Proxy = WAIT_FAIL_VALUE;
    } else {
        g_waitFlag4Proxy = WAIT_SUCCESS_VALUE;
    }
}

static int StreamSessionOpened(int sessionId, int result)
{
    LOG("[cb][stream]open session sessionId[%d],rst[%d]", sessionId, result);
    if (result == SOFTBUS_OK) {
        g_waitFlag4Stream = WAIT_SUCCESS_VALUE;
    } else {
        g_waitFlag4Stream = WAIT_FAIL_VALUE;
    }
    return SOFTBUS_OK;
}

static void StreamSessionClosed(int sessionId)
{
    LOG("[cb][stream]close session sessionId[%d]", sessionId);
}


static void StreamReceived(int sessionId, const StreamData *data, const StreamData *ext, const StreamFrameInfo *frame)
{
    if (data != NULL) {
        LOG("[cb][stream]ByteRec sessionId:%d, data= %.*s.\n", sessionId, data->bufLen, data->buf);
        g_waitFlag4Stream = WAIT_SUCCESS_VALUE;
    }
}

void ResetWaitFlag(void)
{
    g_waitFlag = WAIT_DEF_VALUE;
}

void ResetWaitFlag4Data(void)
{
    g_waitFlag4Data = WAIT_DEF_VALUE;
    g_waitFlag4DataClose  = WAIT_DEF_VALUE;
}

void ResetWaitFlag4Ctl(void)
{
    g_waitFlag4Ctl = WAIT_DEF_VALUE;
    g_waitFlag4CtlClose = WAIT_DEF_VALUE;
}

void ResetWaitFlag4Proxy(void)
{
    g_waitFlag4Proxy = WAIT_DEF_VALUE;
}

void ResetWaitFlag4Stream(void)
{
    g_waitFlag4Stream = WAIT_DEF_VALUE;
}

void ResetwaitCount4Online(void)
{
    g_nodeOnlineCount = 0;
}

void ResetwaitCount4Offline(void)
{
    g_nodeOfflineCount = 0;
}

int RegisterDeviceStateDefCallback(void)
{
    return RegNodeDeviceStateCb(DEF_PKG_NAME, &g_defNodeStateCallback);
}

int UnRegisterDeviceStateDefCallback(void)
{
    return UnregNodeDeviceStateCb(&g_defNodeStateCallback);
}

int CreateSsAndOpenSession4Data(void)
{
    int ret;
    int timeout = 10;
    ret = CreateSessionServer(DEF_PKG_NAME, SESSION_NAME_DATA, g_sessionlistener4Data);
    if (ret != SOFTBUS_OK) {
        LOG("call CreateSessionServer[data] fail, ret:%d", ret);
        return ret;
    }

    ResetWaitFlag4Data();
    int sessionId;
    sessionId = OpenSession(SESSION_NAME_DATA, SESSION_NAME_DATA, g_networkId, DEF_GROUP_ID, g_sessionAttr4Data);
    if (sessionId < SESSION_ID_MIN) {
        LOG("call OpenSession[data] fail, ret sid:%d,netId:%s", sessionId, g_networkId);
        ret = RemoveSessionServer(DEF_PKG_NAME, SESSION_NAME_DATA);
        if (ret != SOFTBUS_OK) {
            LOG("RemoveSessionServer[data] fail, ret:%d", ret);
        }
        return SOFTBUS_ERR;
    }
    SetCurrentSessionId4Data(sessionId);
    LOG("call OpenSession[data] success,sid:%d", sessionId);

    ret = Wait4Session(timeout, SESSION_4DATA);
    if (ret != SOFTBUS_OK) {
        LOG("OpenSession[data] fail");
        ret = RemoveSessionServer(DEF_PKG_NAME, SESSION_NAME_DATA);
        if (ret != SOFTBUS_OK) {
            LOG("RemoveSessionServer[data] fail, ret:%d", ret);
        }
        return SOFTBUS_ERR;
    }
    return ret;
}

int CreateSsAndOpenSession4Ctl(void)
{
    int ret;
    int timeout = 10;
    ret = CreateSessionServer(DEF_PKG_NAME, SESSION_NAME_CTL, g_sessionlistener4Ctl);
    if (ret != SOFTBUS_OK) {
        LOG("call CreateSessionServer[ctl] fail, ret:%d", ret);
        return ret;
    }

    int sessionId;
    ResetWaitFlag4Ctl();
    sessionId = OpenSession(SESSION_NAME_CTL, SESSION_NAME_CTL, g_networkId, DEF_GROUP_ID, g_sessionAttr4Ctl);
    if (sessionId < SESSION_ID_MIN) {
        LOG("call OpenSession[ctl] fail, ret sid:%d", sessionId);
        ret = RemoveSessionServer(DEF_PKG_NAME, SESSION_NAME_CTL);
        if (ret != SOFTBUS_OK) {
            LOG("RemoveSessionServer[ctl] fail, ret:%d", ret);
        }
        return SOFTBUS_ERR;
    }
    SetCurrentSessionId4Ctl(sessionId);
    LOG("call OpenSession[ctl] success,sid:%d", sessionId);

    ret = Wait4Session(timeout, SESSION_4CTL);
    if (ret != SOFTBUS_OK) {
        LOG("OpenSession[ctl] fail");
        ret = RemoveSessionServer(DEF_PKG_NAME, SESSION_NAME_CTL);
        if (ret != SOFTBUS_OK) {
            LOG("RemoveSessionServer[ctl] fail, ret:%d", ret);
        }
        return SOFTBUS_ERR;
    }
    return ret;
}

int CreateSsAndOpenSession4Proxy(void)
{
    int ret;
    int timeout = 10;
    ret = CreateSessionServer(DEF_PKG_NAME, SESSION_NAME_PROXY, g_sessionlistener4Proxy);
    if (ret != SOFTBUS_OK) {
        LOG("call CreateSessionServer[Proxy] fail, ret:%d", ret);
        return ret;
    }

    int sessionId;
    ResetWaitFlag4Proxy();
    sessionId = OpenSession(SESSION_NAME_PROXY, SESSION_NAME_PROXY, g_networkId, DEF_GROUP_ID, g_sessionAttr4Proxy);
    if (sessionId < SESSION_ID_MIN) {
        LOG("call OpenSession[Proxy] fail, ret sid:%d", sessionId);
        return SOFTBUS_ERR;
    }
    SetCurrentSessionId4Proxy(sessionId);
    LOG("call OpenSession[Proxy] success,sid:%d", sessionId);

    ret = Wait4Session(timeout, SESSION_4PROXY);
    if (ret != SOFTBUS_OK) {
        LOG("OpenSession[Proxy] fail");
        return SOFTBUS_ERR;
    }
    return ret;
}

int SendData4Data(DataType type, int size)
{
    int ret;
    if (size > 0) {
        g_expectDataContent = (char*)calloc(1, size);
        if (g_expectDataContent == NULL) {
            LOG("[send data]calloc fail");
            return SOFTBUS_ERR;
        }
        (void)memset_s(g_expectDataContent, size, g_fillContentChar, size);
    } else {
        LOG("[send data]invalid param[size>=1]");
        return SOFTBUS_ERR;
    }

    g_expectDataSize = size;
    ResetWaitFlag();
    if (type == DATA_TYPE_MSG) {
        ret = SendMessage(g_currentSessionId4Data, g_expectDataContent, size);
    } else if (type == DATA_TYPE_BYTE) {
        ret = SendBytes(g_currentSessionId4Data, g_expectDataContent, size);
    } else {
        LOG("[send data]invalid param[DataType]");
        free(g_expectDataContent);
        return SOFTBUS_ERR;
    }
    if (ret != SOFTBUS_OK) {
        LOG("[send data]call SendX fail, ret:%d", ret);
        free(g_expectDataContent);
        return SOFTBUS_ERR;
    }
    LOG("[send data]call SendX success");
    int sleeptime = 90;
    int timeout = 30;
    if (size > SIZE_1K * SIZE_1K) {
        timeout = sleeptime;
    }
    ret = Wait(timeout);
    free(g_expectDataContent);
    g_expectDataContent = NULL;
    return ret;
}

int SendData4Message(DataType type, int size)
{
    int ret;
    if (size > 0) {
        g_expectMessageContent = (char*)calloc(1, size);
        if (g_expectMessageContent == NULL) {
            LOG("[send data]calloc fail");
            return SOFTBUS_ERR;
        }
        (void)memset_s(g_expectMessageContent, size, g_fillContentChar, size);
    } else {
        LOG("[send data]invalid param[size>=1]");
        return SOFTBUS_ERR;
    }

    g_expectDataSize = size;
    ResetWaitFlag();
    if (type == DATA_TYPE_MSG) {
        ret = SendMessage(g_currentSessionId4Proxy, g_expectMessageContent, size);
    } else if (type == DATA_TYPE_BYTE) {
        ret = SendBytes(g_currentSessionId4Proxy, g_expectMessageContent, size);
    } else {
        LOG("[send data]invalid param[DataType]");
        free(g_expectMessageContent);
        return SOFTBUS_ERR;
    }
    if (ret != SOFTBUS_OK) {
        LOG("[send data]call SendX fail, ret:%d", ret);
        free(g_expectMessageContent);
        return SOFTBUS_ERR;
    }
    free(g_expectMessageContent);
    g_expectMessageContent = NULL;
    return ret;
}

int CloseSessionAndRemoveSs4Data(void)
{
    int ret4Close;
    int timeout = 2;
    // 主动close不回调
    ResetWaitFlag4Data();
    CloseSession(g_currentSessionId4Data);
    ret4Close = Wait4Session(timeout, SESSION_4DATA);
    if (ret4Close == SOFTBUS_OK) {
        LOG("CloseSession[data] recv callback");
    }

    int ret4Ss;
    ret4Ss = RemoveSessionServer(DEF_PKG_NAME, SESSION_NAME_DATA);
    if (ret4Ss != SOFTBUS_OK) {
        LOG("RemoveSessionServer[data] fail, ret:%d", ret4Ss);
    }

    if (ret4Ss != SOFTBUS_OK || ret4Close == SOFTBUS_OK) {
        return SOFTBUS_ERR;
    } else {
        return SOFTBUS_OK;
    }
}

int CloseSessionAndRemoveSs4Ctl(void)
{
    int ret4Close;
    int timeout = 2;
    ResetWaitFlag4Ctl();
    CloseSession(g_currentSessionId4Ctl);
    ret4Close = Wait4Session(timeout, SESSION_4CTL);
    if (ret4Close == SOFTBUS_OK) {
        LOG("CloseSession[ctrl] recv callback");
    }

    int ret4Ss;
    ret4Ss = RemoveSessionServer(DEF_PKG_NAME, SESSION_NAME_CTL);
    if (ret4Ss != SOFTBUS_OK) {
        LOG("RemoveSessionServer[ctl] fail, ret:%d", ret4Ss);
    }

    if (ret4Ss != SOFTBUS_OK || ret4Close == SOFTBUS_OK) {
        return SOFTBUS_ERR;
    } else {
        return SOFTBUS_OK;
    }
}

int CloseSessionAndRemoveSs4Proxy(void)
{
    int ret4Close;
    int timeout = 2;
    ResetWaitFlag();
    CloseSession(g_currentSessionId4Proxy);
    ret4Close = Wait(timeout);
    if (ret4Close == SOFTBUS_OK) {
        LOG("CloseSession[Proxy] recv callback");
    }

    int ret4Ss;
    ret4Ss = RemoveSessionServer(DEF_PKG_NAME, SESSION_NAME_PROXY);
    if (ret4Ss != SOFTBUS_OK) {
        LOG("RemoveSessionServer[Proxy] fail, ret:%d", ret4Ss);
    }

    if (ret4Ss != SOFTBUS_OK || ret4Close == SOFTBUS_OK) {
        return SOFTBUS_ERR;
    } else {
        return SOFTBUS_OK;
    }
}

int OpenSession4DataByP2p(void)
{
    int ret;
    int timeout = 10;
    ResetWaitFlag4Data();
    int sId = OpenSession(SESSION_NAME_DATA, SESSION_NAME_DATA, g_networkId, DEF_GROUP_ID, g_p2pattribute);
    if (sId < SESSION_ID_MIN) {
        LOG("openSession[data] fail,id:%d, netId:%s", sId, g_networkId);
        return  SOFTBUS_ERR;
    }
    SetCurrentSessionId4Data(sId);
    ret = Wait4Session(timeout, SESSION_4DATA);
    if (ret != SOFTBUS_OK) {
        LOG("Wait4Session[data] fail");
        return SOFTBUS_ERR;
    }
    return ret;
}

int OpenSession4ProxyByP2p(void)
{
    int ret;
    int timeout = 10;
    ResetWaitFlag4Proxy();
    int sId = OpenSession(SESSION_NAME_PROXY, SESSION_NAME_PROXY, g_networkId, DEF_GROUP_ID, g_p2pattributeProxy);
    if (sId < SESSION_ID_MIN) {
        LOG("openSession[data] fail,id:%d, netId:%s", sId, g_networkId);
        return SOFTBUS_ERR;
    }
    SetCurrentSessionId4Proxy(sId);
    ret = Wait4Session(timeout, SESSION_4PROXY);
    if (ret != SOFTBUS_OK) {
        LOG("Wait4Session[data] fail");
        return SOFTBUS_ERR;
    }

    return ret;
}

int OpenSessionBatch4Data(char groupId[][GROUP_ID_LEN], int* sessionId, int count)
{
    int ret;
    int timeout = 10;
    int rstFlag = SOFTBUS_OK;
    for (int i = 0; i < count; i++) {
        ResetWaitFlag4Data();
        int sId = OpenSession(SESSION_NAME_DATA, SESSION_NAME_DATA, g_networkId, groupId[i], g_sessionAttr4Data);
        if (sId < SESSION_ID_MIN) {
            LOG("call OpenSession[data] fail,sid:%d,i:%d", sId, i);
            rstFlag = SOFTBUS_ERR;
            continue;
        }
        SetCurrentSessionId4Data(sId);
        *(sessionId + i) = sId;
        ret = Wait4Session(timeout, SESSION_4DATA);
        if (ret != SOFTBUS_OK) {
            LOG("Wait4Session[data] fail,i:%d", i);
            rstFlag = SOFTBUS_ERR;
        }
    }
    return rstFlag;
}

int OpenSessionBatch4Ctl(char groupId[][GROUP_ID_LEN], int* sessionId, int count)
{
    int ret;
    int timeout = 10;
    int rstFlag = SOFTBUS_OK;
    for (int i = 0; i < count; i++) {
        ResetWaitFlag4Ctl();
        int sId = OpenSession(SESSION_NAME_CTL, SESSION_NAME_CTL, g_networkId, groupId[i], g_sessionAttr4Ctl);
        if (sId < SESSION_ID_MIN) {
            LOG("call OpenSession[ctrl] fail,sid:%d,i:%d", sId, i);
            rstFlag = SOFTBUS_ERR;
            continue;
        }
        SetCurrentSessionId4Ctl(sId);
        *(sessionId + i) = sId;
        ret = Wait4Session(timeout, SESSION_4CTL);
        if (ret != SOFTBUS_OK) {
            LOG("Wait4Session[ctrl] fail,i:%d", i);
            rstFlag = SOFTBUS_ERR;
        }
    }
    return rstFlag;
}

int CloseSessionBatch4Data(int* sessionId, int count)
{
    for (int i = 0; i < count; i++) {
        int sid = *(sessionId + i);
        LOG("close session[data] sid:%d", sid);

        SetCurrentSessionId4Data(sid);
        CloseSession(sid);
    }
    return SOFTBUS_OK;
}

int CloseSessionBatch4Ctl(int* sessionId, int count)
{
    for (int i = 0; i < count; i++) {
        int sid = *(sessionId + i);
        LOG("close session[ctrl] sid:%d", sid);

        SetCurrentSessionId4Ctl(sid);
        CloseSession(sid);
    }
    return SOFTBUS_OK;
}

int WaitNodeCount(int timeout, WaitNodeStateType state, int expectCount)
{
    LOG("Wait4Node,timeout:%d, type:%d, exp count:%d", timeout, state, expectCount);
    int hitFlag = -1;
    while (timeout > 0) {
        sleep(ONE_SECOND);
        switch (state) {
            case STATE_ONLINE:
                if (g_nodeOnlineCount == expectCount) {
                    LOG("Wait4Node[online] succ,timeout:%d", timeout);
                    hitFlag = 1;
                }
                break;
            case STATE_OFFLINE:
                if (g_nodeOfflineCount == expectCount) {
                    LOG("Wait4Node[offline] succ,timeout:%d", timeout);
                    hitFlag = 1;
                }
                break;
            default:
                LOG("Wait4Node state error");
                hitFlag = 1;
                break;
        }
        if (hitFlag != -1) {
            break;
        }
        timeout--;
    }
    switch (state) {
        case STATE_ONLINE:
            if (g_nodeOnlineCount != expectCount) {
                LOG("Wait4Node[online] fail[exp:%d, real:%d]", expectCount, g_nodeOnlineCount);
                return SOFTBUS_ERR;
            }
            break;
        case STATE_OFFLINE:
            if (g_nodeOfflineCount != expectCount) {
                LOG("Wait4Node[offline] fail[exp:%d, real:%d]", expectCount, g_nodeOfflineCount);
                return SOFTBUS_ERR;
            }
            break;
        default:
            return SOFTBUS_ERR;
    }
    return SOFTBUS_OK;
}

// set/get function
void ResetClosedSessionCount4Data(void)
{
    g_sessionCloseCount4Data = 0;
}

void ResetClosedSessionCount4Ctrl(void)
{
    g_sessionCloseCount4Ctrl = 0;
}

int GetClosedSessionCount4Data(void)
{
    return g_sessionCloseCount4Data;
}

int GetClosedSessionCount4Ctrl(void)
{
    return g_sessionCloseCount4Ctrl;
}

char* GetNetworkId(void)
{
    return g_networkId;
}

// To one device only
int CheckRemoteDeviceIsNull(int isSetNetId)
{
    int nodeNum = 0;
    NodeBasicInfo* nodeInfo = NULL;
    int ret = GetAllNodeDeviceInfo(DEF_PKG_NAME, &nodeInfo, &nodeNum);
    LOG("[check]get node number is:%d, ret:%d", nodeNum, ret);
    if (nodeInfo != NULL && nodeNum > 0) {
        LOG("[check]get netid is:%s", nodeInfo->networkId);
        if (isSetNetId == BOOL_TRUE) {
            (void)strncpy_s(g_networkId, NETWORK_ID_BUF_LEN, nodeInfo->networkId, NETWORK_ID_BUF_LEN);
        }
        FreeNodeInfo(nodeInfo);
        return SOFTBUS_OK;
    } else {
        LOG("[check]get nodeInfo is null");
        return SOFTBUS_ERR;
    }
}

ISessionListener* GetSessionListenser4Data(void)
{
    return g_sessionlistener4Data;
}

ISessionListener* GetSessionListenser4Ctl(void)
{
    return g_sessionlistener4Ctl;
}

ISessionListener* GetSessionListenser4Proxy(void)
{
    return g_sessionlistener4Proxy;
}

ISessionListener* GetSessionListenser4Stream(void)
{
    return g_sessionlistener4Stream;
}

IFileSendListener* GetSendFileListener(void)
{
    return g_fileSendListener;
}

IFileReceiveListener* GetRecvFileListener(void)
{
    return g_fileRecvListener;
}

SessionAttribute* GetSessionAttr4Data(void)
{
    return g_sessionAttr4Data;
}

SessionAttribute* GetSessionAttr4DataP2p(void)
{
    return g_p2pattribute;
}

SessionAttribute* GetSessionAttr4ProxyP2p(void)
{
    return g_p2pattributeProxy;
}
SessionAttribute* GetSessionAttr4Ctl(void)
{
    return g_sessionAttr4Ctl;
}

SessionAttribute* GetSessionAttr4Proxy(void)
{
    return g_sessionAttr4Proxy;
}

void SetCurrentSessionId4Data(int sessionId)
{
    g_currentSessionId4Data = sessionId;
}

void SetCurrentSessionId4Ctl(int sessionId)
{
    g_currentSessionId4Ctl = sessionId;
}

void SetCurrentSessionId4Proxy(int sessionId)
{
    g_currentSessionId4Proxy = sessionId;
}

int GetCurrentSessionId4Data(void)
{
    return g_currentSessionId4Data;
}

int GetCurrentSessionId4Ctl(void)
{
    return g_currentSessionId4Ctl;
}

int GetCurrentSessionId4Proxy(void)
{
    return g_currentSessionId4Proxy;
}

int GetOpenSessionCount4Proxy(void)
{
    return g_sessionOpenCount4Proxy;
}

uint64_t GetTestTokenId(void)
{
    return g_tokenId;
}

void TestSetUp(void)
{
    g_defNodeStateCallback.events = EVENT_NODE_STATE_MASK;
    g_defNodeStateCallback.onNodeOnline = OnDefNodeOnline;
    g_defNodeStateCallback.onNodeOffline = OnDefNodeOffline;
    g_defNodeStateCallback.onNodeBasicInfoChanged = OnDefNodeBasicInfoChanged;
    g_defNodeStateCallback.onNodeStatusChanged = onDefNodeStatusChanged;
    if (g_sessionlistener4Data == NULL) {
        g_sessionlistener4Data = (ISessionListener*)calloc(1, sizeof(ISessionListener));
        g_sessionlistener4Data->OnSessionOpened = DataSessionOpened;
        g_sessionlistener4Data->OnSessionClosed = DataSessionClosed;
        g_sessionlistener4Data->OnMessageReceived = DataMessageReceived;
        g_sessionlistener4Data->OnBytesReceived = DataBytesReceived;
    }
    if (g_sessionlistener4Ctl == NULL) {
        g_sessionlistener4Ctl = (ISessionListener*)calloc(1, sizeof(ISessionListener));
        g_sessionlistener4Ctl->OnSessionOpened = ControlSessionOpened;
        g_sessionlistener4Ctl->OnSessionClosed = ControlSessionClosed;
        g_sessionlistener4Ctl->OnMessageReceived = ControlMessageReceived;
        g_sessionlistener4Ctl->OnBytesReceived = ControlBytesReceived;
    }
    if (g_sessionlistener4Pass == NULL) {
        g_sessionlistener4Pass = (ISessionListener*)calloc(1, sizeof(ISessionListener));
        g_sessionlistener4Pass->OnSessionOpened = PassiveSessionOpened;
        g_sessionlistener4Pass->OnSessionClosed = PassiveSessionClosed;
        g_sessionlistener4Pass->OnMessageReceived = PassiveMessageReceived;
        g_sessionlistener4Pass->OnBytesReceived = PassiveBytesReceived;
    }
    if (g_sessionlistener4Perf == NULL) {
        g_sessionlistener4Perf = (ISessionListener*)calloc(1, sizeof(ISessionListener));
        g_sessionlistener4Perf->OnSessionOpened = PerfSessionOpened;
        g_sessionlistener4Perf->OnSessionClosed = PerfSessionClosed;
        g_sessionlistener4Perf->OnMessageReceived = PerfMessageReceived;
        g_sessionlistener4Perf->OnBytesReceived = PerfBytesReceived;
    }
    if (g_sessionlistener4Proxy == NULL) {
        g_sessionlistener4Proxy = (ISessionListener*)calloc(1, sizeof(ISessionListener));
        g_sessionlistener4Proxy->OnSessionOpened = ProxySessionOpened;
        g_sessionlistener4Proxy->OnSessionClosed = ProxySessionClosed;
        g_sessionlistener4Proxy->OnMessageReceived = ProxyMessageReceived;
        g_sessionlistener4Proxy->OnBytesReceived = ProxyBytesReceived;
    }
    if (g_sessionlistener4Stream == NULL) {
        g_sessionlistener4Stream = (ISessionListener*)calloc(1, sizeof(ISessionListener));
        g_sessionlistener4Stream->OnSessionOpened = StreamSessionOpened;
        g_sessionlistener4Stream->OnSessionClosed = StreamSessionClosed;
        g_sessionlistener4Stream->OnStreamReceived = StreamReceived;
    }
    if (g_fileSendListener == NULL) {
        g_fileSendListener = (IFileSendListener*)calloc(1, sizeof(IFileSendListener));
        g_fileSendListener->OnSendFileProcess = OnSendFileProcess;
        g_fileSendListener->OnSendFileFinished = OnSendFileFinished;
        g_fileSendListener->OnFileTransError = OnSendFileTransError;
    };
    if (g_fileRecvListener == NULL) {
        g_fileRecvListener = (IFileReceiveListener*)calloc(1, sizeof(IFileReceiveListener));
        g_fileRecvListener->OnReceiveFileStarted = OnReceiveFileStarted;
        g_fileRecvListener->OnReceiveFileProcess = OnReceiveFileProcess;
        g_fileRecvListener->OnReceiveFileFinished = OnReceiveFileFinished;
        g_fileRecvListener->OnFileTransError = OnRecvFileTransError;
    }
    if (g_sessionAttr4Data == NULL) {
        g_sessionAttr4Data = (SessionAttribute*)calloc(1, sizeof(SessionAttribute));
        g_sessionAttr4Data->dataType = TYPE_BYTES;
    }
    if (g_sessionAttr4Ctl == NULL) {
        g_sessionAttr4Ctl = (SessionAttribute*)calloc(1, sizeof(SessionAttribute));
        g_sessionAttr4Ctl->dataType = TYPE_BYTES;
    }
    if (g_sessionAttr4Pass == NULL) {
        g_sessionAttr4Pass = (SessionAttribute*)calloc(1, sizeof(SessionAttribute));
        g_sessionAttr4Pass->dataType = TYPE_BYTES;
    }
    if (g_sessionAttr4Perf == NULL) {
        g_sessionAttr4Perf = (SessionAttribute*)calloc(1, sizeof(SessionAttribute));
        g_sessionAttr4Perf->dataType = TYPE_BYTES;
    }
    if (g_sessionAttr4Proxy == NULL) {
        g_sessionAttr4Proxy = (SessionAttribute*)calloc(1, sizeof(SessionAttribute));
        g_sessionAttr4Proxy->dataType = TYPE_MESSAGE;
    }
    if (g_p2pattribute == NULL) {
        g_p2pattribute = (SessionAttribute*)calloc(1, sizeof(SessionAttribute));
        g_p2pattribute->dataType = TYPE_BYTES;
        g_p2pattribute->linkTypeNum = 1;
        g_p2pattribute->linkType[0] = LINK_TYPE_WIFI_P2P;
    }
    if (g_p2pattributeProxy == NULL) {
        g_p2pattributeProxy = (SessionAttribute*)calloc(1, sizeof(SessionAttribute));
        g_p2pattributeProxy->dataType = TYPE_MESSAGE;
        g_p2pattributeProxy->linkTypeNum = 1;
        g_p2pattributeProxy->linkType[0] = LINK_TYPE_WIFI_P2P;
    }
    if (g_barrier == NULL) {
        g_barrier = (pthread_barrier_t*)calloc(1, sizeof(pthread_barrier_t));
    }
    if (g_sId4Task2 == NULL) {
        g_sId4Task2 = (int*)malloc(sizeof(int) * DEF_SEND_DATA_SID_COUNT);
    }
    if (g_sId4Task3 == NULL) {
        g_sId4Task3 = (int*)malloc(sizeof(int) * DEF_SEND_DATA_SID_COUNT);
    }
}

void TestTearDown(void)
{
    if (g_sessionlistener4Data != NULL) {
        free(g_sessionlistener4Data);
        g_sessionlistener4Data = NULL;
    }
    if (g_sessionlistener4Ctl != NULL) {
        free(g_sessionlistener4Ctl);
        g_sessionlistener4Ctl = NULL;
    }
    if (g_sessionlistener4Pass != NULL) {
        free(g_sessionlistener4Pass);
        g_sessionlistener4Pass = NULL;
    }
    if (g_sessionlistener4Perf != NULL) {
        free(g_sessionlistener4Perf);
        g_sessionlistener4Perf = NULL;
    }
    if (g_sessionlistener4Proxy != NULL) {
        free(g_sessionlistener4Proxy);
        g_sessionlistener4Proxy = NULL;
    }
    if (g_sessionlistener4Stream != NULL) {
        free(g_sessionlistener4Stream);
        g_sessionlistener4Stream = NULL;
    }
    if (g_fileRecvListener != NULL) {
        free(g_fileRecvListener);
        g_fileRecvListener = NULL;
    }
    if (g_fileSendListener != NULL) {
        free(g_fileSendListener);
        g_fileSendListener = NULL;
    }
    if (g_sessionAttr4Data != NULL) {
        free(g_sessionAttr4Data);
        g_sessionAttr4Data = NULL;
    }
    if (g_sessionAttr4Ctl != NULL) {
        free(g_sessionAttr4Ctl);
        g_sessionAttr4Ctl = NULL;
    }
    if (g_sessionAttr4Pass != NULL) {
        free(g_sessionAttr4Pass);
        g_sessionAttr4Pass = NULL;
    }
    if (g_sessionAttr4Perf != NULL) {
        free(g_sessionAttr4Perf);
        g_sessionAttr4Perf = NULL;
    }
    if (g_sessionAttr4Proxy != NULL) {
        free(g_sessionAttr4Proxy);
        g_sessionAttr4Proxy = NULL;
    }
    if (g_p2pattribute != NULL) {
        free(g_p2pattribute);
        g_p2pattribute = NULL;
    }
    if (g_p2pattributeProxy != NULL) {
        free(g_p2pattributeProxy);
        g_p2pattributeProxy = NULL;
    }
    if (g_barrier != NULL) {
        free(g_barrier);
        g_barrier = NULL;
    }
    if (g_sId4Task2 != NULL) {
        free(g_sId4Task2);
        g_sId4Task2 = NULL;
    }
    if (g_sId4Task3 != NULL) {
        free(g_sId4Task3);
        g_sId4Task3 = NULL;
    }
}

void AddPermission(void)
{
    const char *perms[2];
    perms[0] = OHOS_PERMISSION_DISTRIBUTED_SOFTBUS_CENTER;
    perms[1] = OHOS_PERMISSION_DISTRIBUTED_DATASYNC;
    NativeTokenInfoParams infoTnstance = {
        .dcapsNum = 0,
        .permsNum = 2,
        .aclsNum = 0,
        .dcaps = NULL,
        .perms = perms,
        .acls = NULL,
        .processName = "dsoftbus_test_service",
        .aplStr = "system_core",
    };
    g_tokenId = GetAccessTokenId(&infoTnstance);
    SetSelfTokenID(g_tokenId);
}