1*993229b6Sjkunz /*
2*993229b6Sjkunz * File: DataSourceImager.cpp
3*993229b6Sjkunz *
4*993229b6Sjkunz * Copyright (c) Freescale Semiconductor, Inc. All rights reserved.
5*993229b6Sjkunz * See included license file for license details.
6*993229b6Sjkunz */
7*993229b6Sjkunz
8*993229b6Sjkunz #include "DataSourceImager.h"
9*993229b6Sjkunz #include <stdlib.h>
10*993229b6Sjkunz #include <string.h>
11*993229b6Sjkunz
12*993229b6Sjkunz using namespace elftosb;
13*993229b6Sjkunz
DataSourceImager()14*993229b6Sjkunz DataSourceImager::DataSourceImager()
15*993229b6Sjkunz : Blob(),
16*993229b6Sjkunz m_fill(0),
17*993229b6Sjkunz m_baseAddress(0),
18*993229b6Sjkunz m_isBaseAddressSet(false)
19*993229b6Sjkunz {
20*993229b6Sjkunz }
21*993229b6Sjkunz
setBaseAddress(uint32_t address)22*993229b6Sjkunz void DataSourceImager::setBaseAddress(uint32_t address)
23*993229b6Sjkunz {
24*993229b6Sjkunz m_baseAddress = address;
25*993229b6Sjkunz m_isBaseAddressSet = true;
26*993229b6Sjkunz }
27*993229b6Sjkunz
setFillPattern(uint8_t pattern)28*993229b6Sjkunz void DataSourceImager::setFillPattern(uint8_t pattern)
29*993229b6Sjkunz {
30*993229b6Sjkunz m_fill = pattern;
31*993229b6Sjkunz }
32*993229b6Sjkunz
reset()33*993229b6Sjkunz void DataSourceImager::reset()
34*993229b6Sjkunz {
35*993229b6Sjkunz clear();
36*993229b6Sjkunz
37*993229b6Sjkunz m_fill = 0;
38*993229b6Sjkunz m_baseAddress = 0;
39*993229b6Sjkunz m_isBaseAddressSet = false;
40*993229b6Sjkunz }
41*993229b6Sjkunz
42*993229b6Sjkunz //! \param dataSource Pointer to an instance of a concrete data source subclass.
43*993229b6Sjkunz //!
addDataSource(DataSource * source)44*993229b6Sjkunz void DataSourceImager::addDataSource(DataSource * source)
45*993229b6Sjkunz {
46*993229b6Sjkunz unsigned segmentCount = source->getSegmentCount();
47*993229b6Sjkunz unsigned index = 0;
48*993229b6Sjkunz for (; index < segmentCount; ++index)
49*993229b6Sjkunz {
50*993229b6Sjkunz addDataSegment(source->getSegmentAt(index));
51*993229b6Sjkunz }
52*993229b6Sjkunz }
53*993229b6Sjkunz
54*993229b6Sjkunz //! \param segment The segment to add. May be any type of data segment, including
55*993229b6Sjkunz //! a pattern segment.
addDataSegment(DataSource::Segment * segment)56*993229b6Sjkunz void DataSourceImager::addDataSegment(DataSource::Segment * segment)
57*993229b6Sjkunz {
58*993229b6Sjkunz DataSource::PatternSegment * patternSegment = dynamic_cast<DataSource::PatternSegment*>(segment);
59*993229b6Sjkunz
60*993229b6Sjkunz unsigned segmentLength = segment->getLength();
61*993229b6Sjkunz bool segmentHasLocation = segment->hasNaturalLocation();
62*993229b6Sjkunz uint32_t segmentAddress = segment->getBaseAddress();
63*993229b6Sjkunz
64*993229b6Sjkunz uint8_t * toPtr = NULL;
65*993229b6Sjkunz unsigned addressDelta;
66*993229b6Sjkunz unsigned newLength;
67*993229b6Sjkunz
68*993229b6Sjkunz // If a pattern segment's length is 0 then make it as big as the fill pattern.
69*993229b6Sjkunz // This needs to be done before the buffer is adjusted.
70*993229b6Sjkunz if (patternSegment && segmentLength == 0)
71*993229b6Sjkunz {
72*993229b6Sjkunz SizedIntegerValue & pattern = patternSegment->getPattern();
73*993229b6Sjkunz segmentLength = pattern.getSize();
74*993229b6Sjkunz }
75*993229b6Sjkunz
76*993229b6Sjkunz if (segmentLength)
77*993229b6Sjkunz {
78*993229b6Sjkunz if (segmentHasLocation)
79*993229b6Sjkunz {
80*993229b6Sjkunz // Make sure a base address is set.
81*993229b6Sjkunz if (!m_isBaseAddressSet)
82*993229b6Sjkunz {
83*993229b6Sjkunz m_baseAddress = segmentAddress;
84*993229b6Sjkunz m_isBaseAddressSet = true;
85*993229b6Sjkunz }
86*993229b6Sjkunz
87*993229b6Sjkunz // The segment is located before our buffer's first address.
88*993229b6Sjkunz // toPtr is not set in this if, but in the else branch of the next if.
89*993229b6Sjkunz // Unless the segment completely overwrite the current data.
90*993229b6Sjkunz if (segmentAddress < m_baseAddress)
91*993229b6Sjkunz {
92*993229b6Sjkunz addressDelta = m_baseAddress - segmentAddress;
93*993229b6Sjkunz
94*993229b6Sjkunz uint8_t * newData = (uint8_t *)malloc(m_length + addressDelta);
95*993229b6Sjkunz memcpy(&newData[addressDelta], m_data, m_length);
96*993229b6Sjkunz free(m_data);
97*993229b6Sjkunz
98*993229b6Sjkunz m_data = newData;
99*993229b6Sjkunz m_length += addressDelta;
100*993229b6Sjkunz m_baseAddress = segmentAddress;
101*993229b6Sjkunz }
102*993229b6Sjkunz
103*993229b6Sjkunz // This segment is located or extends outside of our buffer.
104*993229b6Sjkunz if (segmentAddress + segmentLength > m_baseAddress + m_length)
105*993229b6Sjkunz {
106*993229b6Sjkunz newLength = segmentAddress + segmentLength - m_baseAddress;
107*993229b6Sjkunz
108*993229b6Sjkunz m_data = (uint8_t *)realloc(m_data, newLength);
109*993229b6Sjkunz
110*993229b6Sjkunz // Clear any bytes between the old data and the new segment.
111*993229b6Sjkunz addressDelta = segmentAddress - (m_baseAddress + m_length);
112*993229b6Sjkunz if (addressDelta)
113*993229b6Sjkunz {
114*993229b6Sjkunz memset(m_data + m_length, 0, addressDelta);
115*993229b6Sjkunz }
116*993229b6Sjkunz
117*993229b6Sjkunz toPtr = m_data + (segmentAddress - m_baseAddress);
118*993229b6Sjkunz m_length = newLength;
119*993229b6Sjkunz }
120*993229b6Sjkunz else
121*993229b6Sjkunz {
122*993229b6Sjkunz toPtr = m_data + (segmentAddress - m_baseAddress);
123*993229b6Sjkunz }
124*993229b6Sjkunz }
125*993229b6Sjkunz // Segment has no natural location, so just append it to the end of our buffer.
126*993229b6Sjkunz else
127*993229b6Sjkunz {
128*993229b6Sjkunz newLength = m_length + segmentLength;
129*993229b6Sjkunz m_data = (uint8_t *)realloc(m_data, newLength);
130*993229b6Sjkunz toPtr = m_data + m_length;
131*993229b6Sjkunz m_length = newLength;
132*993229b6Sjkunz }
133*993229b6Sjkunz }
134*993229b6Sjkunz
135*993229b6Sjkunz // A loop is used because getData() may fill in less than the requested
136*993229b6Sjkunz // number of bytes per call.
137*993229b6Sjkunz unsigned offset = 0;
138*993229b6Sjkunz while (offset < segmentLength)
139*993229b6Sjkunz {
140*993229b6Sjkunz offset += segment->getData(offset, segmentLength - offset, toPtr + offset);
141*993229b6Sjkunz }
142*993229b6Sjkunz }
143*993229b6Sjkunz
144