1

/*****************************************************************************

     2

 *

     3

 * $Id:$

     4

 *

     5

 ****************************************************************************/

     6

     7

#include <libgen.h> // basename()

     8

     9

#include <iostream>

    10

#include <iomanip>

    11

#include <fstream>

    12

using namespace std;

    13

    14

#include "CommandFoeWrite.h"

    15

#include "byteorder.h"

    16

    17

/*****************************************************************************/

    18

    19

CommandFoeWrite::CommandFoeWrite():

    20

    FoeCommand("foe_write", "Store a file on a slave via FoE.")

    21

{

    22

}

    23

    24

/*****************************************************************************/

    25

    26

string CommandFoeWrite::helpString() const

    27

{

    28

    stringstream str;

    29

    30

    str << getName() << " [OPTIONS] <FILENAME>" << endl

    31

        << endl

    32

        << getBriefDescription() << endl

    33

        << endl

    34

        << "This command requires a single slave to be selected." << endl

    35

    	<< endl

    36

        << "Arguments:" << endl

    37

        << "  FILENAME can either be a path to a file, or '-'. In" << endl

    38

        << "           the latter case, data are read from stdin and" << endl

    39

        << "           the --output-file option has to be specified." << endl

    40

        << endl

    41

        << "Command-specific options:" << endl

    42

        << "  --output-file -o <file>   Target filename on the slave." << endl

    43

        << "                            If the FILENAME argument is" << endl

    44

        << "                            '-', this is mandatory." << endl

    45

        << "                            Otherwise, the basename() of" << endl

    46

        << "                            FILENAME is used by default." << endl

    47

        << "  --alias       -a <alias>" << endl

    48

        << "  --position    -p <pos>    Slave selection. See the help" << endl

    49

        << "                            of the 'slaves' command." << endl

    50

        << endl

    51

        << numericInfo();

    52

    53

    return str.str();

    54

}

    55

    56

/****************************************************************************/

    57

    58

void CommandFoeWrite::execute(MasterDevice &m, const StringVector &args)

    59

{

    60

    stringstream err;

    61

    ec_ioctl_slave_foe_t data;

    62

    ifstream file;

    63

    SlaveList slaves;

    64

    string storeFileName;

    65

    66

    if (args.size() != 1) {

    67

        err << "'" << getName() << "' takes exactly one argument!";

    68

        throwInvalidUsageException(err);

    69

    }

    70

    71

    if (args[0] == "-") {

    72

        loadFoeData(&data, cin);

    73

        if (getOutputFile().empty()) {

    74

            err << "Please specify a filename for the slave side"

    75

                << " with --output-file!";

    76

            throwCommandException(err);

    77

        } else {

    78

            storeFileName = getOutputFile();

    79

        }

    80

    } else {

    81

        file.open(args[0].c_str(), ifstream::in | ifstream::binary);

    82

        if (file.fail()) {

    83

            err << "Failed to open '" << args[0] << "'!";

    84

            throwCommandException(err);

    85

        }

    86

        loadFoeData(&data, file);

    87

        file.close();

    88

        if (getOutputFile().empty()) {

    89

            char *cpy = strdup(args[0].c_str()); // basename can modify

    90

                                                 // the string contents

    91

            storeFileName = basename(cpy);

    92

            free(cpy);

    93

        } else {

    94

            storeFileName = getOutputFile();

    95

        }

    96

    }

    97

    98

    try {

    99

        m.open(MasterDevice::ReadWrite);

   100

    } catch (MasterDeviceException &e) {

   101

        if (data.buffer_size)

   102

            delete [] data.buffer;

   103

        throw e;

   104

    }

   105

   106

    slaves = selectedSlaves(m);

   107

    if (slaves.size() != 1) {

   108

        if (data.buffer_size)

   109

            delete [] data.buffer;

   110

        throwSingleSlaveRequired(slaves.size());

   111

    }

   112

    data.slave_position = slaves.front().position;

   113

   114

    // write data via foe to the slave

   115

    data.offset = 0;

   116

    strncpy(data.file_name, storeFileName.c_str(), sizeof(data.file_name));

   117

   118

    try {

   119

        m.writeFoe(&data);

   120

    } catch (MasterDeviceException &e) {

   121

        if (data.buffer_size)

   122

            delete [] data.buffer;

   123

        if (data.abort_code) {

   124

            err << "Failed to write via FoE: "

   125

                << errorString(data.abort_code);

   126

            throwCommandException(err);

   127

        } else {

   128

            throw e;

   129

        }

   130

    }

   131

   132

    if (getVerbosity() == Verbose) {

   133

        cerr << "FoE writing finished." << endl;

   134

    }

   135

   136

    if (data.buffer_size)

   137

        delete [] data.buffer;

   138

}

   139

   140

/*****************************************************************************/

   141

   142

void CommandFoeWrite::loadFoeData(

   143

        ec_ioctl_slave_foe_t *data,

   144

        const istream &in

   145

        )

   146

{

   147

    stringstream err;

   148

    ostringstream tmp;

   149

   150

    tmp << in.rdbuf();

   151

    string const &contents = tmp.str();

   152

   153

    if (getVerbosity() == Verbose) {

   154

        cerr << "Read " << contents.size() << " bytes of FoE data." << endl;

   155

    }

   156

   157

    data->buffer_size = contents.size();

   158

   159

    if (data->buffer_size) {

   160

        // allocate buffer and read file into buffer

   161

        data->buffer = new uint8_t[data->buffer_size];

   162

        contents.copy((char *) data->buffer, contents.size());

   163

    }

   164

}

   165

   166

/*****************************************************************************/