#include <iostream>
#include <stdint.h>
#include <stdio.h>
#include <string.h>
#include <assert.h>
#include <uuid/uuid.h>
#include <fcntl.h>
#include <errno.h>
#include <_simple.h>
#include <unistd.h>
#include <sys/uio.h>
#include <sys/param.h>
#include <sys/sysctl.h>
#include <sys/resource.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/mman.h>
#include <dirent.h>
#include <mach/mach.h>
#include <mach/machine.h>
#include <mach/mach_time.h>
#include <mach-o/loader.h>
#include <mach-o/nlist.h>
#include <mach-o/fat.h>
#include <pthread.h>
#include <libc_private.h>
#include "Diagnostics.h"
#if BUILDING_CACHE_BUILDER
#include <dispatch/dispatch.h>
dispatch_queue_t sWarningQueue = dispatch_queue_create("com.apple.dyld.cache-builder.warnings", NULL);
#endif
Diagnostics::Diagnostics(bool verbose)
#if BUILDING_CACHE_BUILDER
: _verbose(verbose)
, _prefix("")
#endif
{
}
#if BUILDING_CACHE_BUILDER
Diagnostics::Diagnostics(const std::string& prefix, bool verbose)
: _verbose(verbose)
, _prefix(prefix)
{
}
#endif
Diagnostics::~Diagnostics()
{
clearError();
}
void Diagnostics::error(const char* format, ...)
{
va_list list;
va_start(list, format);
error(format, list);
va_end(list);
}
void Diagnostics::error(const char* format, va_list list)
{
_buffer = _simple_salloc();
_simple_vsprintf(_buffer, format, list);
#if BUILDING_CACHE_BUILDER
if ( !_verbose )
return;
if (_prefix.empty()) {
fprintf(stderr, "%s\n", _simple_string(_buffer));
} else {
fprintf(stderr, "[%s] %s\n", _prefix.c_str(), _simple_string(_buffer));
}
#endif
}
bool Diagnostics::hasError() const
{
return _buffer != nullptr;
}
bool Diagnostics::noError() const
{
return _buffer == nullptr;
}
void Diagnostics::clearError()
{
if ( _buffer )
_simple_sfree(_buffer);
_buffer = nullptr;
}
void Diagnostics::assertNoError() const
{
if ( _buffer != nullptr )
abort_report_np("%s", _simple_string(_buffer));
}
bool Diagnostics::errorMessageContains(const char* subString) const
{
if ( _buffer == nullptr )
return false;
return (strstr(_simple_string(_buffer), subString) != nullptr);
}
#if !BUILDING_CACHE_BUILDER
const char* Diagnostics::errorMessage() const
{
return _simple_string(_buffer);
}
#else
void Diagnostics::warning(const char* format, ...)
{
_SIMPLE_STRING tmp = _simple_salloc();
va_list list;
va_start(list, format);
_simple_vsprintf(tmp, format, list);
va_end(list);
#if BUILDING_CACHE_BUILDER
dispatch_sync(sWarningQueue, ^{
_warnings.insert(_simple_string(tmp));
});
#else
_warnings.insert(_simple_string(tmp));
#endif
_simple_sfree(tmp);
}
void Diagnostics::verbose(const char* format, ...)
{
if ( !_verbose )
return;
char* output_string;
va_list list;
va_start(list, format);
vasprintf(&output_string, format, list);
va_end(list);
if (_prefix.empty()) {
fprintf(stderr, "%s", output_string);
} else {
fprintf(stderr, "[%s] %s", _prefix.c_str(), output_string);
}
free(output_string);
}
const std::string Diagnostics::prefix() const
{
return _prefix;
}
void Diagnostics::copy(const Diagnostics& other)
{
if ( other.hasError() )
error("%s", other.errorMessage().c_str());
for (const std::string& warn : other.warnings())
warning("%s", warn.c_str());
}
std::string Diagnostics::errorMessage() const
{
if ( _buffer != nullptr )
return _simple_string(_buffer);
else
return "";
}
const std::set<std::string> Diagnostics::warnings() const
{
#if BUILDING_CACHE_BUILDER
__block std::set<std::string> retval;
dispatch_sync(sWarningQueue, ^{
retval = _warnings;
});
return retval;
#else
return _warnings;
#endif
}
void Diagnostics::clearWarnings()
{
#if BUILDING_CACHE_BUILDER
dispatch_sync(sWarningQueue, ^{
_warnings.clear();
});
#else
_warnings.clear();
#endif
}
#if BUILDING_CACHE_BUILDER
void TimeRecorder::pushTimedSection() {
openTimings.push_back(mach_absolute_time());
}
void TimeRecorder::recordTime(const char* format, ...) {
uint64_t t = mach_absolute_time();
uint64_t previousTime = openTimings.back();
openTimings.pop_back();
char* output_string = nullptr;
va_list list;
va_start(list, format);
vasprintf(&output_string, format, list);
va_end(list);
if (output_string != nullptr) {
timings.push_back(TimingEntry {
.time = t - previousTime,
.logMessage = std::string(output_string),
.depth = (int)openTimings.size()
});
}
openTimings.push_back(mach_absolute_time());
}
void TimeRecorder::popTimedSection() {
openTimings.pop_back();
}
static inline uint32_t absolutetime_to_milliseconds(uint64_t abstime)
{
return (uint32_t)(abstime/1000/1000);
}
void TimeRecorder::logTimings() {
for (const TimingEntry& entry : timings) {
for (int i = 0 ; i < entry.depth ; i++) {
std::cerr << " ";
}
std::cerr << "time to " << entry.logMessage << " " << absolutetime_to_milliseconds(entry.time) << "ms" << std::endl;
}
timings.clear();
}
#endif
#endif