1

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

     2

 *

     3

 *  $Id$

     4

 *

     5

 *  Copyright (C) 2006-2009  Florian Pose, Ingenieurgemeinschaft IgH

     6

 *

     7

 *  This file is part of the IgH EtherCAT Master.

     8

 *

     9

 *  The IgH EtherCAT Master is free software; you can redistribute it and/or

    10

 *  modify it under the terms of the GNU General Public License version 2, as

    11

 *  published by the Free Software Foundation.

    12

 *

    13

 *  The IgH EtherCAT Master is distributed in the hope that it will be useful,

    14

 *  but WITHOUT ANY WARRANTY; without even the implied warranty of

    15

 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General

    16

 *  Public License for more details.

    17

 *

    18

 *  You should have received a copy of the GNU General Public License along

    19

 *  with the IgH EtherCAT Master; if not, write to the Free Software

    20

 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

    21

 *

    22

 *  ---

    23

 *

    24

 *  The license mentioned above concerns the source code only. Using the

    25

 *  EtherCAT technology and brand is only permitted in compliance with the

    26

 *  industrial property and similar rights of Beckhoff Automation GmbH.

    27

 *

    28

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

    29

    30

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

    31

#include <string.h>

    32

    33

#include <iostream>

    34

#include <iomanip>

    35

#include <fstream>

    36

using namespace std;

    37

    38

#include "CommandFoeWrite.h"

    39

#include "foe.h"

    40

#include "MasterDevice.h"

    41

    42

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

    43

    44

CommandFoeWrite::CommandFoeWrite():

    45

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

    46

{

    47

}

    48

    49

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

    50

    51

string CommandFoeWrite::helpString() const

    52

{

    53

    stringstream str;

    54

    55

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

    56

        << endl

    57

        << getBriefDescription() << endl

    58

        << endl

    59

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

    60

        << endl

    61

        << "Arguments:" << endl

    62

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

    63

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

    64

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

    65

        << endl

    66

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

    67

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

    68

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

    69

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

    70

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

    71

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

    72

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

    73

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

    74

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

    75

        << endl

    76

        << numericInfo();

    77

    78

    return str.str();

    79

}

    80

    81

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

    82

    83

void CommandFoeWrite::execute(const StringVector &args)

    84

{

    86

    stringstream err;

    87

    ec_ioctl_slave_foe_t data;

    88

    ifstream file;

    89

    SlaveList slaves;

    90

    string storeFileName;

    91

    92

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

    93

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

    94

        throwInvalidUsageException(err);

    95

    }

    96

    99

        err << getName() << " requires to select a single master!";

   100

        throwInvalidUsageException(err);

   101

    }

   103

   104

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

   105

        loadFoeData(&data, cin);

   106

        if (getOutputFile().empty()) {

   107

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

   108

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

   109

            throwCommandException(err);

   110

        } else {

   111

            storeFileName = getOutputFile();

   112

        }

   113

    } else {

   114

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

   115

        if (file.fail()) {

   116

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

   117

            throwCommandException(err);

   118

        }

   119

        loadFoeData(&data, file);

   120

        file.close();

   121

        if (getOutputFile().empty()) {

   122

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

   123

                                                 // the string contents

   124

            storeFileName = basename(cpy);

   125

            free(cpy);

   126

        } else {

   127

            storeFileName = getOutputFile();

   128

        }

   129

    }

   130

   131

    try {

   132

        m.open(MasterDevice::ReadWrite);

   133

    } catch (MasterDeviceException &e) {

   134

        if (data.buffer_size)

   135

            delete [] data.buffer;

   136

        throw e;

   137

    }

   138

   139

    slaves = selectedSlaves(m);

   140

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

   141

        if (data.buffer_size)

   142

            delete [] data.buffer;

   143

        throwSingleSlaveRequired(slaves.size());

   144

    }

   145

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

   146

   147

    // write data via foe to the slave

   148

    data.offset = 0;

   149

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

   150

   151

    try {

   152

        m.writeFoe(&data);

   153

    } catch (MasterDeviceException &e) {

   154

        if (data.buffer_size)

   155

            delete [] data.buffer;

   156

        if (data.result) {

   157

            if (data.result == FOE_OPCODE_ERROR) {

   158

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

   159

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

   160

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

   161

            } else {

   162

                err << "Failed to write via FoE: "

   163

                    << resultText(data.result);

   164

            }

   165

            throwCommandException(err);

   166

        } else {

   167

            throw e;

   168

        }

   169

    }

   170

   171

    if (getVerbosity() == Verbose) {

   172

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

   173

    }

   174

   175

    if (data.buffer_size)

   176

        delete [] data.buffer;

   177

}

   178

   179

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

   180

   181

void CommandFoeWrite::loadFoeData(

   182

        ec_ioctl_slave_foe_t *data,

   183

        const istream &in

   184

        )

   185

{

   186

    stringstream err;

   187

    ostringstream tmp;

   188

   189

    tmp << in.rdbuf();

   190

    string const &contents = tmp.str();

   191

   192

    if (getVerbosity() == Verbose) {

   193

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

   194

    }

   195

   196

    data->buffer_size = contents.size();

   197

   198

    if (data->buffer_size) {

   199

        // allocate buffer and read file into buffer

   200

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

   201

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

   202

    }

   203

}

   204

   205

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