1

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

     2

 *

     3

 * $Id$

     4

 *

     5

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

     6

     7

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

     8

#include <string.h>

     9

    10

#include <iostream>

    11

#include <iomanip>

    12

#include <fstream>

    13

using namespace std;

    14

    15

#include "CommandFoeWrite.h"

    16

#include "foe.h"

    17

    18

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

    19

    20

CommandFoeWrite::CommandFoeWrite():

    21

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

    22

{

    23

}

    24

    25

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

    26

    27

string CommandFoeWrite::helpString() const

    28

{

    29

    stringstream str;

    30

    31

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

    32

        << endl

    33

        << getBriefDescription() << endl

    34

        << endl

    35

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

    36

    	<< endl

    37

        << "Arguments:" << endl

    38

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

    39

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

    40

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

    41

        << endl

    42

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

    43

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

    44

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

    45

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

    46

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

    47

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

    48

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

    49

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

    50

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

    51

        << endl

    52

        << numericInfo();

    53

    54

    return str.str();

    55

}

    56

    57

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

    58

    59

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

    60

{

    61

    stringstream err;

    62

    ec_ioctl_slave_foe_t data;

    63

    ifstream file;

    64

    SlaveList slaves;

    65

    string storeFileName;

    66

    67

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

    68

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

    69

        throwInvalidUsageException(err);

    70

    }

    71

    72

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

    73

        loadFoeData(&data, cin);

    74

        if (getOutputFile().empty()) {

    75

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

    76

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

    77

            throwCommandException(err);

    78

        } else {

    79

            storeFileName = getOutputFile();

    80

        }

    81

    } else {

    82

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

    83

        if (file.fail()) {

    84

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

    85

            throwCommandException(err);

    86

        }

    87

        loadFoeData(&data, file);

    88

        file.close();

    89

        if (getOutputFile().empty()) {

    90

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

    91

                                                 // the string contents

    92

            storeFileName = basename(cpy);

    93

            free(cpy);

    94

        } else {

    95

            storeFileName = getOutputFile();

    96

        }

    97

    }

    98

    99

    try {

   100

        m.open(MasterDevice::ReadWrite);

   101

    } catch (MasterDeviceException &e) {

   102

        if (data.buffer_size)

   103

            delete [] data.buffer;

   104

        throw e;

   105

    }

   106

   107

    slaves = selectedSlaves(m);

   108

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

   109

        if (data.buffer_size)

   110

            delete [] data.buffer;

   111

        throwSingleSlaveRequired(slaves.size());

   112

    }

   113

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

   114

   115

    // write data via foe to the slave

   116

    data.offset = 0;

   117

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

   118

   119

    try {

   120

        m.writeFoe(&data);

   121

    } catch (MasterDeviceException &e) {

   122

        if (data.buffer_size)

   123

            delete [] data.buffer;

   124

        if (data.result) {

   125

            if (data.result == FOE_OPCODE_ERROR) {

   126

                err << "FoE write aborted with error code 0x"

   127

                    << setw(8) << setfill('0') << hex << data.error_code

   128

                    << ": " << errorText(data.error_code);

   129

            } else {

   130

                err << "Failed to write via FoE: "

   131

                    << resultText(data.result);

   132

            }

   133

            throwCommandException(err);

   134

        } else {

   135

            throw e;

   136

        }

   137

    }

   138

   139

    if (getVerbosity() == Verbose) {

   140

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

   141

    }

   142

   143

    if (data.buffer_size)

   144

        delete [] data.buffer;

   145

}

   146

   147

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

   148

   149

void CommandFoeWrite::loadFoeData(

   150

        ec_ioctl_slave_foe_t *data,

   151

        const istream &in

   152

        )

   153

{

   154

    stringstream err;

   155

    ostringstream tmp;

   156

   157

    tmp << in.rdbuf();

   158

    string const &contents = tmp.str();

   159

   160

    if (getVerbosity() == Verbose) {

   161

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

   162

    }

   163

   164

    data->buffer_size = contents.size();

   165

   166

    if (data->buffer_size) {

   167

        // allocate buffer and read file into buffer

   168

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

   169

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

   170

    }

   171

}

   172

   173

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