xref: /netbsd-src/external/bsd/elftosb/dist/common/DataSourceImager.cpp (revision 993229b6fea628ff8b1fa09146c80b0cfb2768eb)
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