mirror of
https://gitee.com/openharmony/xts_dcts
synced 2024-11-23 07:39:59 +00:00
265 lines
7.4 KiB
C++
265 lines
7.4 KiB
C++
/*
|
||
* Copyright (C) 2022 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 <cstring>
|
||
|
||
#include "shm_utils.h"
|
||
|
||
using namespace std;
|
||
const int MAX_DATA_LENGTH = 1024;
|
||
const int PERMISSION = 0666;
|
||
const int CODE_HEAD = 4;
|
||
const int STR_KEY = 5;
|
||
const int SHARED_DATA_LEN = 5;
|
||
const int WAITTIME = 2;
|
||
const int DECIM_TEN = 10;
|
||
const int CALCU_TWO = 2;
|
||
const int CALCU_FOUR_AIGHT = 48;
|
||
const int RES_FAIL = -1;
|
||
const char LOGSTR[20] = "LOG_SHM::----";
|
||
static void* shm = nullptr; // 分配的共享内存的原始首地址
|
||
static struct shared_use_st* shared; // 指向shm
|
||
static int shmid; // 共享内存标识符
|
||
|
||
int createShm(int key)
|
||
{
|
||
// 创建共享内存
|
||
LOG("%s createShm begin...", LOGSTR);
|
||
int a = sizeof(struct shared_use_st);
|
||
LOG("a=%d", a);
|
||
shmid = shmget(static_cast<key_t>(key), sizeof(struct shared_use_st), PERMISSION | IPC_CREAT);
|
||
LOG("shmget shmid=%d", shmid);
|
||
if (shmid == -1) {
|
||
LOG("%s err: shmget, shmid = -1", LOGSTR);
|
||
return -1;
|
||
}
|
||
// 将共享内存连接到当前进程的地址空间
|
||
shm = shmat(shmid, nullptr, 0);
|
||
int n = -1;
|
||
if (shm == static_cast<void*>(&n)) {
|
||
LOG("%s err: shmat, shm = -1", LOGSTR);
|
||
return -1;
|
||
}
|
||
|
||
shared = (struct shared_use_st*)shm;
|
||
initShm();
|
||
LOG("%s createShm end...", LOGSTR);
|
||
return 0;
|
||
}
|
||
|
||
void initShm(void)
|
||
{
|
||
LOG("%s initShm begin...", LOGSTR);
|
||
memset_s(shm, sizeof(struct shared_use_st), 0, sizeof(struct shared_use_st));
|
||
if (shared == nullptr) {
|
||
LOG("%s err:initShm shared = nullptr", LOGSTR);
|
||
return;
|
||
}
|
||
shared->written = 0;
|
||
memset_s(shared->data, MAX_DATA_LENGTH, 0, MAX_DATA_LENGTH);
|
||
LOG("%s initShm end...", LOGSTR);
|
||
return;
|
||
}
|
||
|
||
int readDataFromShm(char* buf)
|
||
{
|
||
LOG("readDataFromShm begin...");
|
||
if (shared == nullptr) {
|
||
LOG("%s err:readDataFromShm shared = nullptr", LOGSTR);
|
||
return -1;
|
||
}
|
||
|
||
if (shared->written != 0) {
|
||
strcpy_s(buf, strlen(shared->data) + 1, shared->data);
|
||
LOG("readDataFromShm buf= %s", buf);
|
||
LOG("readDataFromShm shared->data= %s", shared->data);
|
||
shared->written = 0;
|
||
memset_s(shared->data, MAX_DATA_LENGTH, 0, MAX_DATA_LENGTH);
|
||
} else {
|
||
return -1;
|
||
}
|
||
LOG("SUCCESS: readDataFromShm return 0");
|
||
return 0;
|
||
}
|
||
|
||
int readDataFromShmNoClear(char* buf)
|
||
{
|
||
LOG("readDataFromShmNoClear begin...");
|
||
if (shared == nullptr) {
|
||
return -1;
|
||
}
|
||
if (shared->written != 0) {
|
||
strcpy_s(buf, strlen(shared->data) + 1, shared->data);
|
||
LOG("readDataFromShmNoClear buf= %s", buf);
|
||
LOG("readDataFromShmNoClear shared->data= %s", shared->data);
|
||
} else {
|
||
return -1;
|
||
}
|
||
LOG("SUCCESS: readDataFromShmNoClear return 0");
|
||
return 0;
|
||
}
|
||
|
||
int waitDataWithCode(char* code, char* data)
|
||
{
|
||
LOG("waitDataWithCode begin...");
|
||
int i = 0;
|
||
int timeout = 10;
|
||
char str[MAX_DATA_LENGTH] = { 0 };
|
||
if (code == nullptr || data == nullptr) {
|
||
LOG("code = nullptr ,data = nullptr");
|
||
return RES_FAIL;
|
||
}
|
||
while (i < timeout) {
|
||
if (readDataFromShmNoClear(str) != 0 || strncmp(code, str, CODE_HEAD) != 0) {
|
||
i++;
|
||
sleep(1);
|
||
LOG("while: waitDataWithCode 9999 str= %s, i=%d", str, i);
|
||
continue;
|
||
}
|
||
|
||
if (readDataFromShm(str) == 0 && strncmp(code, str, CODE_HEAD) == 0) {
|
||
if (strncpy_s(data, strlen("0") + 1, str + STR_KEY, 1) != EOK) {
|
||
LOG("ERR:strncpy_s");
|
||
return RES_FAIL;
|
||
}
|
||
LOG("waitDataWithCode 9999 str= %s", str);
|
||
LOG("waitDataWithCode 9999 data= %s", data);
|
||
LOG("SUCCESS:waitDataWithCode return 0");
|
||
return 0;
|
||
}
|
||
i++;
|
||
sleep(1);
|
||
LOG("while: waitDataWithCode 9999 str= %s, i=%d", str, i);
|
||
}
|
||
LOG("ERR :waitDataWithCode ");
|
||
return RES_FAIL;
|
||
}
|
||
|
||
int writeCodeDataToShm(int code, char* buf)
|
||
{
|
||
LOG("writeCodeDataToShm, begin");
|
||
char* str = (char*)malloc(MAX_DATA_LENGTH);
|
||
if (str == nullptr) {
|
||
LOG("malloc fail");
|
||
return -1;
|
||
}
|
||
(void)memset_s(str, MAX_DATA_LENGTH, 0, MAX_DATA_LENGTH);
|
||
|
||
char codeStr[5] = { 0 };
|
||
char* str2 = Int2String(code, codeStr);
|
||
if (str2 == nullptr) {
|
||
LOG("ERROR: str2 == nullptr");
|
||
return -1;
|
||
}
|
||
|
||
if (strcpy_s(str, MAX_DATA_LENGTH, codeStr) != EOK) {
|
||
LOG("ERROR: strcpy_s != EOK");
|
||
return -1;
|
||
}
|
||
if (strcat_s(str, MAX_DATA_LENGTH, ":") != EOK) {
|
||
LOG("ERROR: 1. strcat_s!= EOK ");
|
||
return -1;
|
||
}
|
||
|
||
if (buf == nullptr) {
|
||
LOG("ERROR:buf == nullptr ");
|
||
return -1;
|
||
}
|
||
if (strcat_s(str, MAX_DATA_LENGTH, buf) != EOK) {
|
||
LOG("ERROR:2. strcat_s != EOK");
|
||
return -1;
|
||
}
|
||
int nres = writeDataToShm(str);
|
||
if (nres == -1) {
|
||
return -1;
|
||
}
|
||
return 0;
|
||
}
|
||
int writeDataToShm(char* buf)
|
||
{
|
||
LOG("%s writeDataToShm, begin", LOGSTR);
|
||
if (shared == nullptr) {
|
||
LOG("%s err:writeDataToShm shared == nullptr", LOGSTR);
|
||
return -1;
|
||
}
|
||
if (buf == nullptr) {
|
||
LOG("%s err: writeDataToShm, buf == nullptr", LOGSTR);
|
||
return -1;
|
||
}
|
||
while (shared->written == 1) {
|
||
sleep(1);
|
||
}
|
||
|
||
LOG("writeDataToShm %s", buf);
|
||
memset_s(shared->data, SHARED_DATA_LEN, 0, SHARED_DATA_LEN);
|
||
strcpy_s(shared->data, strlen(buf) + 1, buf);
|
||
shared->written = 1;
|
||
LOG("writeDataToShm shared->data= %s", shared->data);
|
||
LOG("writeDataToShm shared->written= %d", shared->written);
|
||
|
||
sleep(WAITTIME);
|
||
return 0;
|
||
}
|
||
|
||
int disconnectShm(void)
|
||
{
|
||
if (shmdt(shm) == -1) {
|
||
return -1;
|
||
}
|
||
return 0;
|
||
}
|
||
|
||
int deleteShm(void)
|
||
{
|
||
if (shmctl(shmid, IPC_RMID, nullptr) == -1) {
|
||
return -1;
|
||
}
|
||
return 0;
|
||
}
|
||
|
||
char* Int2String(int num, char* str) // 10进制
|
||
{
|
||
if (str == nullptr) {
|
||
return nullptr;
|
||
}
|
||
int i = 0; // 指示填充str
|
||
if (num < 0) {
|
||
num = -num;
|
||
str[i++] = '-';
|
||
}
|
||
// 转换
|
||
do {
|
||
str[i++] = num % DECIM_TEN + CALCU_FOUR_AIGHT;
|
||
num /= DECIM_TEN;
|
||
} while (num); // num不为0继续循环
|
||
|
||
str[i] = '\0';
|
||
|
||
// 确定开始调整的位置
|
||
int j = 0;
|
||
if (str[0] == '-') {
|
||
j = 1; // 从第二位开始调整
|
||
++i; // 由于有负号，所以交换的对称轴也要后移一位
|
||
}
|
||
// 对称交换
|
||
for (; j < i / CALCU_TWO; j++) {
|
||
// 对称交换两端的值 其实就是省下中间变量交换a+b的值：a=a+b;b=a-b;a=a-b;
|
||
str[j] = str[j] + str[i - 1 - j];
|
||
str[i - 1 - j] = str[j] - str[i - 1 - j];
|
||
str[j] = str[j] - str[i - 1 - j];
|
||
}
|
||
|
||
return str; // 返回转换后的值
|
||
} |