116 lines
2.9 KiB
C++
116 lines
2.9 KiB
C++
/*
|
|
* Copyright (C) 2020-2025 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "shared/source/helpers/string.h"
|
|
#include "shared/source/utilities/io_functions.h"
|
|
|
|
#include <cstdint>
|
|
#include <cstring>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
|
|
namespace NEO {
|
|
namespace IoFunctions {
|
|
extern uint32_t mockFopenCalled;
|
|
extern uint32_t failAfterNFopenCount;
|
|
extern FILE *mockFopenReturned;
|
|
extern uint32_t mockVfptrinfCalled;
|
|
extern uint32_t mockFcloseCalled;
|
|
extern uint32_t mockGetenvCalled;
|
|
extern uint32_t mockFseekCalled;
|
|
extern uint32_t mockFtellCalled;
|
|
extern long int mockFtellReturn;
|
|
extern uint32_t mockRewindCalled;
|
|
extern uint32_t mockFreadCalled;
|
|
extern size_t mockFreadReturn;
|
|
extern uint32_t mockFwriteCalled;
|
|
extern size_t mockFwriteReturn;
|
|
extern char *mockFwriteBuffer;
|
|
extern char *mockFreadBuffer;
|
|
extern bool mockVfptrinfUseStdioFunction;
|
|
|
|
extern std::unordered_map<std::string, std::string> *mockableEnvValues;
|
|
|
|
inline FILE *mockFopen(const char *filename, const char *mode) {
|
|
mockFopenCalled++;
|
|
if (failAfterNFopenCount > 0 && failAfterNFopenCount < mockFopenCalled) {
|
|
return NULL;
|
|
}
|
|
return mockFopenReturned;
|
|
}
|
|
|
|
inline int mockVfptrinf(FILE *stream, const char *format, va_list arg) {
|
|
mockVfptrinfCalled++;
|
|
if (stream == stdout || stream == stderr || mockVfptrinfUseStdioFunction) {
|
|
return vfprintf(stream, format, arg);
|
|
}
|
|
return 0x10;
|
|
}
|
|
|
|
inline int mockFclose(FILE *stream) {
|
|
mockFcloseCalled++;
|
|
return 0;
|
|
}
|
|
extern const char *openCLDriverName;
|
|
|
|
inline char *mockGetenv(const char *name) noexcept {
|
|
mockGetenvCalled++;
|
|
if (strcmp(name, "OpenCLDriverName") == 0) {
|
|
return const_cast<char *>(openCLDriverName);
|
|
}
|
|
if (mockableEnvValues != nullptr && mockableEnvValues->find(name) != mockableEnvValues->end()) {
|
|
return const_cast<char *>(mockableEnvValues->find(name)->second.c_str());
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
inline int mockFseek(FILE *stream, long int offset, int origin) {
|
|
mockFseekCalled++;
|
|
return 0;
|
|
}
|
|
|
|
inline long int mockFtell(FILE *stream) {
|
|
mockFtellCalled++;
|
|
return mockFtellReturn;
|
|
}
|
|
|
|
inline void mockRewind(FILE *stream) {
|
|
mockRewindCalled++;
|
|
}
|
|
|
|
inline size_t mockFread(void *ptr, size_t size, size_t count, FILE *stream) {
|
|
mockFreadCalled++;
|
|
if (mockFreadBuffer) {
|
|
memcpy_s(ptr, mockFreadReturn, mockFreadBuffer, size * count);
|
|
}
|
|
return mockFreadReturn;
|
|
}
|
|
|
|
inline size_t mockFwrite(const void *ptr, size_t size, size_t nmemb, FILE *stream) {
|
|
mockFwriteCalled++;
|
|
|
|
if (mockFwriteBuffer) {
|
|
memcpy_s(mockFwriteBuffer, mockFwriteReturn, ptr, size * nmemb);
|
|
}
|
|
|
|
return mockFwriteReturn;
|
|
}
|
|
|
|
inline int mockFflush(FILE *stream) {
|
|
if (stream == stdout || stream == stderr)
|
|
return fflush(stream);
|
|
|
|
return 0;
|
|
}
|
|
|
|
inline int mockMkdir(const char *filename) {
|
|
return 0;
|
|
}
|
|
|
|
} // namespace IoFunctions
|
|
} // namespace NEO
|