xref: /test/xts/dcts/common/shm_utils.h

/*
 * 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.
 */

#ifndef SHM_UTILS_H
#define SHM_UTILS_H

#include <cerrno>
#include <cstdarg>
#include <cstdint>
#include <cstdio>
#include <cstdlib>
#include <pthread.h>
#include <securec.h>
#include <sys/shm.h>
#include <sys/time.h>
#include <unistd.h>

#define MAX_WAIT_TIMEOUT 10
#define SHM_SEND_KEY 123456
#define SHM_RECV_KEY 123466

struct shared_use_st {
    int written;     // 作为一个标志，非0：表示可读，0表示可写
    char data[1024]; // 记录写入和读取的文本
};

#define LOG(format, ...)                                                        \
    do {                                                                        \
        time_t timeSec;                                                         \
        time(&timeSec);                                                         \
        struct tm tmRst;                                                        \
        localtime_r(&timeSec, &tmRst);                                          \
        char strTime[10];                                                       \
        strftime(strTime, sizeof(strTime), "%H:%M:%S", &tmRst);                 \
        fprintf(stdout, "[shm-utils] %s " format "\n", strTime, ##__VA_ARGS__); \
    } while (0)

int createShm(int key);
int disconnectShm(void);
int writeCodeDataToShm(int code, char* buf);
int waitDataWithCode(char* code, char* data);
int writeDataToShm(char* buf);
int deleteShm(void);
char* Int2String(int num, char* str);
void initShm(void);
int readDataFromShm(char* buf);
int readDataFromShmNoClear(char* buf);
#endif