1 // $OpenLDAP$
2 /*
3 * Copyright 2008-2021 The OpenLDAP Foundation, All Rights Reserved.
4 * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
5 */
6
7 #include "LdifWriter.h"
8 #include "StringList.h"
9 #include "LDAPAttribute.h"
10 #include "debug.h"
11 #include <sstream>
12 #include <stdexcept>
13
LdifWriter(std::ostream & output,int version)14 LdifWriter::LdifWriter( std::ostream& output, int version ) :
15 m_ldifstream(output), m_version(version), m_addSeparator(false)
16 {
17 if ( version )
18 {
19 if ( version == 1 )
20 {
21 m_ldifstream << "version: " << version << std::endl;
22 m_addSeparator = true;
23 } else {
24 std::ostringstream err;
25 err << "Unsupported LDIF Version";
26 throw( std::runtime_error(err.str()) );
27 }
28 }
29
30 }
31
writeRecord(const LDAPEntry & le)32 void LdifWriter::writeRecord(const LDAPEntry& le)
33 {
34 std::ostringstream line;
35
36 if ( m_addSeparator )
37 {
38 m_ldifstream << std::endl;
39 } else {
40 m_addSeparator = true;
41 }
42
43 line << "dn: " << le.getDN();
44 this->breakline( line.str(), m_ldifstream );
45
46 const LDAPAttributeList *al = le.getAttributes();
47 LDAPAttributeList::const_iterator i = al->begin();
48 for ( ; i != al->end(); i++ )
49 {
50 StringList values = i->getValues();
51 StringList::const_iterator j = values.begin();
52 for( ; j != values.end(); j++)
53 {
54 // clear output stream
55 line.str("");
56 line << i->getName() << ": " << *j;
57 this->breakline( line.str(), m_ldifstream );
58 }
59 }
60 }
61
writeIncludeRecord(const std::string & target)62 void LdifWriter::writeIncludeRecord( const std::string& target )
63 {
64 DEBUG(LDAP_DEBUG_TRACE, "writeIncludeRecord: " << target << std::endl);
65 std::string scheme = target.substr( 0, sizeof("file:")-1 );
66
67 if ( m_version == 1 )
68 {
69 std::ostringstream err;
70 err << "\"include\" not allowed in LDIF version 1.";
71 throw( std::runtime_error(err.str()) );
72 }
73
74 if ( m_addSeparator )
75 {
76 m_ldifstream << std::endl;
77 } else {
78 m_addSeparator = true;
79 }
80
81 m_ldifstream << "include: ";
82 if ( scheme != "file:" )
83 {
84 m_ldifstream << "file://";
85 }
86
87 m_ldifstream << target << std::endl;
88 }
89
breakline(const std::string & line,std::ostream & out)90 void LdifWriter::breakline( const std::string &line, std::ostream &out )
91 {
92 std::string::size_type pos = 0;
93 std::string::size_type linelength = 76;
94 bool first = true;
95
96 if ( line.length() >= linelength )
97 {
98 while ( pos < line.length() )
99 {
100 if (! first )
101 {
102 out << " ";
103 }
104 out << line.substr(pos, linelength) << std::endl;
105 pos += linelength;
106 if ( first )
107 {
108 first = false;
109 linelength--; //account for the leading space
110 }
111 }
112 } else {
113 out << line << std::endl;
114 }
115 }
116
117