DFGFunctionWhitelist.cpp [plain text]
#include "config.h"
#include "DFGFunctionWhitelist.h"
#if ENABLE(DFG_JIT)
#include "CodeBlock.h"
#include "Options.h"
#include <stdio.h>
#include <string.h>
#include <wtf/NeverDestroyed.h>
#include <wtf/text/StringBuilder.h>
namespace JSC { namespace DFG {
FunctionWhitelist& FunctionWhitelist::ensureGlobalWhitelist()
{
static LazyNeverDestroyed<FunctionWhitelist> functionWhitelist;
static std::once_flag initializeWhitelistFlag;
std::call_once(initializeWhitelistFlag, [] {
const char* functionWhitelistFile = Options::dfgWhitelist();
functionWhitelist.construct(functionWhitelistFile);
});
return functionWhitelist;
}
FunctionWhitelist::FunctionWhitelist(const char* filename)
{
parseFunctionNamesInFile(filename);
}
void FunctionWhitelist::parseFunctionNamesInFile(const char* filename)
{
if (!filename)
return;
FILE* f = fopen(filename, "r");
if (!f) {
dataLogF("Failed to open file %s. Did you add the file-read-data entitlement to WebProcess.sb?\n", filename);
return;
}
char* line;
char buffer[BUFSIZ];
while ((line = fgets(buffer, sizeof(buffer), f))) {
if (strstr(line, "//") == line)
continue;
size_t length = strlen(line);
if (line[length - 1] == '\n') {
line[length - 1] = '\0';
length--;
}
if (!length)
continue;
m_entries.add(String(line, length));
}
int result = fclose(f);
if (result)
dataLogF("Failed to close file %s: %s\n", filename, strerror(errno));
}
bool FunctionWhitelist::contains(CodeBlock* codeBlock) const
{
ASSERT(!isCompilationThread());
if (!Options::dfgWhitelist())
return true;
if (m_entries.isEmpty())
return false;
String name = String::fromUTF8(codeBlock->inferredName());
if (m_entries.contains(name))
return true;
String hash = String::fromUTF8(codeBlock->hashAsStringIfPossible());
if (m_entries.contains(hash))
return true;
return m_entries.contains(name + '#' + hash);
}
} }
#endif // ENABLE(DFG_JIT)