mirror of
https://github.com/davidgiven/fluxengine.git
synced 2025-10-31 11:17:01 -07:00
38 lines
990 B
C++
38 lines
990 B
C++
#include "globals.h"
|
|
#include "flags.h"
|
|
#include "dataspec.h"
|
|
#include "fluxsource/fluxsource.h"
|
|
#include "lib/config.pb.h"
|
|
|
|
static bool ends_with(const std::string& value, const std::string& ending)
|
|
{
|
|
if (ending.size() > value.size())
|
|
return false;
|
|
return std::equal(ending.rbegin(), ending.rend(), value.rbegin());
|
|
}
|
|
|
|
std::unique_ptr<FluxSource> FluxSource::create(const FluxSourceProto& config)
|
|
{
|
|
switch (config.source_case())
|
|
{
|
|
case FluxSourceProto::kFluxfile:
|
|
return createSqliteFluxSource(config.fluxfile());
|
|
|
|
case FluxSourceProto::kDrive:
|
|
return createHardwareFluxSource(config.drive());
|
|
|
|
case FluxSourceProto::kErase:
|
|
return createEraseFluxSource(config.erase());
|
|
|
|
case FluxSourceProto::kKryoflux:
|
|
return createKryofluxFluxSource(config.kryoflux());
|
|
|
|
case FluxSourceProto::kTestPattern:
|
|
return createTestPatternFluxSource(config.test_pattern());
|
|
}
|
|
|
|
Error() << "bad input disk configuration";
|
|
return std::unique_ptr<FluxSource>();
|
|
}
|
|
|