1

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

     2

 *

     3

 * $Id$

     4

 *

     5

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

     6

     7

#include <iostream>

     8

#include <iomanip>

     9

#include <fstream>

    10

using namespace std;

    11

    12

#include "CommandSiiWrite.h"

    13

#include "sii_crc.h"

    14

#include "byteorder.h"

    15

    16

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

    17

    18

CommandSiiWrite::CommandSiiWrite():

    19

    Command("sii_write", "Write SII contents to a slave.")

    20

{

    21

}

    22

    23

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

    24

    25

string CommandSiiWrite::helpString() const

    26

{

    27

    stringstream str;

    28

    29

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

    30

        << endl 

    31

        << getBriefDescription() << endl

    32

        << endl

    33

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

    34

    	<< endl

    35

        << "The file contents are checked for validity and integrity." << endl

    36

        << "These checks can be overridden with the --force option." << endl

    37

        << endl

    38

        << "Arguments:" << endl

    39

        << "  FILENAME must be a path to a file that contains a" << endl

    40

        << "           positive number of words. If it is '-'," << endl

    41

        << "           data are read from stdin." << endl

    42

        << endl

    43

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

    44

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

    45

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

    46

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

    47

        << "  --force    -f          Override validity checks." << endl

    48

        << endl

    49

        << numericInfo();

    50

    51

    return str.str();

    52

}

    53

    54

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

    55

    56

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

    57

{

    58

    stringstream err;

    59

    ec_ioctl_slave_sii_t data;

    60

    ifstream file;

    61

    SlaveList slaves;

    62

    63

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

    64

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

    65

        throwInvalidUsageException(err);

    66

    }

    67

    68

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

    69

        loadSiiData(&data, cin);

    70

    } else {

    71

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

    72

        if (file.fail()) {

    73

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

    74

            throwCommandException(err);

    75

        }

    76

        loadSiiData(&data, file);

    77

        file.close();

    78

    }

    79

    80

    if (!getForce()) {

    81

        try {

    82

            checkSiiData(&data);

    83

        } catch (CommandException &e) {

    84

            delete [] data.words;

    85

            throw e;

    86

        }

    87

    }

    88

    89

    try {

    90

        m.open(MasterDevice::ReadWrite);

    91

    } catch (MasterDeviceException &e) {

    92

        delete [] data.words;

    93

        throw e;

    94

    }

    95

    96

    slaves = selectedSlaves(m);

    97

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

    98

        delete [] data.words;

    99

        throwSingleSlaveRequired(slaves.size());

   100

    }

   101

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

   102

   103

    // send data to master

   104

    data.offset = 0;

   105

    try {

   106

        m.writeSii(&data);

   107

    } catch (MasterDeviceException &e) {

   108

        delete [] data.words;

   109

        throw e;

   110

    }

   111

   112

    if (getVerbosity() == Verbose) {

   113

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

   114

    }

   115

   116

    delete [] data.words;

   117

}

   118

   119

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

   120

   121

void CommandSiiWrite::loadSiiData(

   122

        ec_ioctl_slave_sii_t *data,

   123

        const istream &in

   124

        )

   125

{

   126

    stringstream err;

   127

    ostringstream tmp;

   128

   129

    tmp << in.rdbuf();

   130

    string const &contents = tmp.str();

   131

   132

    if (getVerbosity() == Verbose) {

   133

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

   134

    }

   135

   136

    if (!contents.size() || contents.size() % 2) {

   137

        err << "Invalid data size " << contents.size() << "! "

   138

            << "Must be non-zero and even.";

   139

        throwCommandException(err);

   140

    }

   141

    data->nwords = contents.size() / 2;

   142

   143

    // allocate buffer and read file into buffer

   144

    data->words = new uint16_t[data->nwords];

   145

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

   146

}

   147

   148

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

   149

   150

void CommandSiiWrite::checkSiiData(

   151

        const ec_ioctl_slave_sii_t *data

   152

        )

   153

{

   154

    stringstream err;

   155

    const uint16_t *categoryHeader;

   156

    uint16_t categoryType, categorySize;

   157

    uint8_t crc;

   158

   159

    if (data->nwords < 0x0041) {

   160

        err << "SII data too short (" << data->nwords << " words)! Mimimum is"

   161

            " 40 fixed words + 1 delimiter. Use --force to write anyway.";

   162

        throwCommandException(err);

   163

    }

   164

   165

    // calculate checksum over words 0 to 6

   166

    crc = calcSiiCrc((const uint8_t *) data->words, 14);

   167

    if (crc != ((const uint8_t *) data->words)[14]) {

   168

        err << "CRC incorrect. Must be 0x"

   169

            << hex << setfill('0') << setw(2) << (unsigned int) crc

   170

            << ". Use --force to write anyway.";

   171

        throwCommandException(err);

   172

    }

   173

   174

    // cycle through categories to detect corruption

   175

    categoryHeader = data->words + 0x0040U;

   176

    categoryType = le16tocpu(*categoryHeader);

   177

    while (categoryType != 0xffff) {

   178

        if (categoryHeader + 1 > data->words + data->nwords) {

   179

            err << "SII data seem to be corrupted! "

   180

                << "Use --force to write anyway.";

   181

            throwCommandException(err);

   182

        }

   183

        categorySize = le16tocpu(*(categoryHeader + 1));

   184

        if (categoryHeader + 2 + categorySize + 1

   185

                > data->words + data->nwords) {

   186

            err << "SII data seem to be corrupted! "

   187

                "Use --force to write anyway.";

   188

            throwCommandException(err);

   189

        }

   190

        categoryHeader += 2 + categorySize;

   191

        categoryType = le16tocpu(*categoryHeader);

   192

    }

   193

}

   194

   195

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