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

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

    34

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

    35

        << endl

    36

        << "Arguments:" << endl

    37

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

    38

        << "           positive number of words." << endl

    39

        << endl

    40

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

    41

        << "  --slave -s <index>  Positive numerical ring position" << endl

    42

        << "                      (mandatory)." << endl

    43

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

    44

        << endl

    45

        << numericInfo();

    46

    47

    return str.str();

    48

}

    49

    50

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

    51

    52

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

    53

{

    55

    stringstream err;

    56

    ec_ioctl_slave_sii_t data;

    57

    ifstream file;

    58

    unsigned int byte_size;

    59

    const uint16_t *categoryHeader;

    60

    uint16_t categoryType, categorySize;

    61

    uint8_t crc;

    62

    68

        throwInvalidUsageException(err);

    69

    }

    71

    72

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

    73

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

    74

        throwInvalidUsageException(err);

    75

    }

    76

    77

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

    78

    if (file.fail()) {

    79

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

    80

        throwCommandException(err);

    81

    }

    82

    83

    // get length of file

    84

    file.seekg(0, ios::end);

    85

    byte_size = file.tellg();

    86

    file.seekg(0, ios::beg);

    87

    88

    if (!byte_size || byte_size % 2) {

    89

        err << "Invalid file size! Must be non-zero and even.";

    90

        throwCommandException(err);

    91

    }

    92

    93

    data.nwords = byte_size / 2;

    94

    if (data.nwords < 0x0041 && !force) {

    95

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

    96

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

    97

        throwCommandException(err);

    98

    }

    99

   100

    // allocate buffer and read file into buffer

   101

    data.words = new uint16_t[data.nwords];

   102

    file.read((char *) data.words, byte_size);

   103

    file.close();

   104

   105

    if (!force) {

   106

        // calculate checksum over words 0 to 6

   107

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

   108

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

   109

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

   110

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

   111

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

   112

            throwCommandException(err);

   113

        }

   114

   115

        // cycle through categories to detect corruption

   116

        categoryHeader = data.words + 0x0040U;

   117

        categoryType = le16tocpu(*categoryHeader);

   118

        while (categoryType != 0xffff) {

   119

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

   120

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

   121

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

   122

                throwCommandException(err);

   123

            }

   124

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

   125

            if (categoryHeader + 2 + categorySize + 1

   126

                    > data.words + data.nwords) {

   127

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

   128

                    "Use --force to write anyway.";

   129

                throwCommandException(err);

   130

            }

   131

            categoryHeader += 2 + categorySize;

   132

            categoryType = le16tocpu(*categoryHeader);

   133

        }

   134

    }

   135

   136

    // send data to master

   137

    m.open(MasterDevice::ReadWrite);

   138

    data.offset = 0;

   139

	m.writeSii(&data);

   140

}

   141

   142

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