1 /*
2 * Copyright (c) 2015, 2018, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24 /**
25 * Demonstrate Java object storage and retrieval using an LDAP directory.
26 * The Person object and its associated object and state factory is supplied by
27 * a third-party module. As the Person object does not implement
28 * javax.naming.Referenceable, the classname of its state and object factory
29 * must be specified to the JNDI initial context.
30 */
31
32 package test;
33
34 import java.io.PrintStream;
35 import java.net.*;
36 import java.util.*;
37 import javax.naming.*;
38 import javax.naming.directory.*;
39
40 import org.example.person.Person;
41
42 public class StorePerson {
43
44 static {
45 final PrintStream out = new PrintStream(System.out, true);
46 final PrintStream err = new PrintStream(System.err, true);
47
48 System.setOut(out);
49 System.setErr(err);
50 }
51
52 // LDAP capture file
53 private static final String LDAP_CAPTURE_FILE =
54 System.getProperty("test.src") + "/src/test/test/StorePerson.ldap";
55
56 public static void main(String[] args) throws Exception {
57
58 /*
59 * Process arguments
60 */
61
62 int argc = args.length;
63 if ((argc < 1) ||
64 ((argc == 1) && (args[0].equalsIgnoreCase("-help")))) {
65
66 System.err.println("\nUsage: StorePerson <ldapurl>\n");
67 System.err.println(" <ldapurl> is the LDAP URL of the parent entry\n");
68 System.err.println("example:");
69 System.err.println(" java StorePerson ldap://oasis/o=airius.com");
70 return;
71 }
72
73 /*
74 * Launch the LDAP server with the StorePerson.ldap capture file
75 */
76
77 try (ServerSocket serverSocket = new ServerSocket()) {
78 serverSocket.bind(new InetSocketAddress(InetAddress.getLoopbackAddress(), 0));
79 new Thread(new Runnable() {
80 @Override
81 public void run() {
82 try {
83 new LDAPServer(serverSocket, LDAP_CAPTURE_FILE);
84 } catch (Exception e) {
85 System.out.println("ERROR: unable to launch LDAP server");
86 e.printStackTrace();
87 }
88 }
89 }).start();
90
91 /*
92 * Store Person objects in the LDAP directory
93 */
94
95 Hashtable<String,Object> env = new Hashtable<>();
96 env.put(Context.INITIAL_CONTEXT_FACTORY,
97 "com.sun.jndi.ldap.LdapCtxFactory");
98 URI ldapUri = new URI(args[0]);
99 if (ldapUri.getPort() == -1) {
100 ldapUri = new URI(ldapUri.getScheme(), null, ldapUri.getHost(),
101 serverSocket.getLocalPort(), ldapUri.getPath(), null, null);
102 }
103 env.put(Context.PROVIDER_URL, ldapUri.toString());
104 if (args[args.length - 1].equalsIgnoreCase("-trace")) {
105 env.put("com.sun.jndi.ldap.trace.ber", System.out);
106 }
107
108 // Specify the factory classname explicitly
109 env.put(Context.STATE_FACTORIES, "org.example.person.PersonFactory");
110 env.put(Context.OBJECT_FACTORIES, "org.example.person.PersonFactory");
111
112 System.out.println("StorePerson: connecting to " + ldapUri);
113 DirContext ctx = new InitialDirContext(env);
114 Person person = null;
115 String name = "John Smith";
116 String dn = "cn=" + name;
117
118 try {
119 person = new Person(name, "Smith");
120 person.setMailAddress("jsmith@smith.com");
121 ctx.bind(dn, person);
122 System.out.println("StorePerson: created entry '" + dn + "'");
123 } catch (NameAlreadyBoundException e) {
124 System.err.println("StorePerson: entry '" + dn +
125 "' already exists");
126 cleanup(ctx, (String)null);
127 return;
128 }
129
130 name = "Jill Smyth";
131 String dn2 = "cn=" + name;
132 Person person2 = new Person(name, "Smyth");
133 person2.setMailAddress("jsmyth@smith.com");
134
135 try {
136 ctx.bind(dn2, person2);
137 System.out.println("StorePerson: created entry '" + dn2 + "'");
138 } catch (NameAlreadyBoundException e) {
139 System.err.println("StorePerson: entry '" + dn2 +
140 "' already exists");
141 cleanup(ctx, dn);
142 return;
143 }
144
145 /*
146 * Retrieve Person objects from the LDAP directory
147 */
148
149 try {
150 Person person3 = (Person) ctx.lookup(dn);
151 System.out.println("StorePerson: retrieved object: " + person3);
152 if (person.getAttributes().equals(person3.getAttributes())) {
153 System.out.println(
154 "StorePerson: retrieved person matches original");
155 } else {
156 System.out.println(
157 "StorePerson: retrieved person does NOT match original");
158 }
159 } catch (NamingException e) {
160 System.err.println("StorePerson: error retrieving entry '" +
161 dn + "' " + e);
162 e.printStackTrace();
163 cleanup(ctx, dn, dn2);
164 return;
165 }
166
167 try {
168 Person person4 = (Person) ctx.lookup(dn2);
169 System.out.println("StorePerson: retrieved object: " + person4);
170 if (person2.getAttributes().equals(person4.getAttributes())) {
171 System.out.println(
172 "StorePerson: retrieved person matches original");
173 } else {
174 System.out.println(
175 "StorePerson: retrieved person does NOT match original");
176 }
177 } catch (NamingException e) {
178 System.err.println("StorePerson: error retrieving entry '" +
179 dn2 + "' " + e);
180 e.printStackTrace();
181 cleanup(ctx, dn, dn2);
182 return;
183 }
184
185 cleanup(ctx, dn, dn2);
186 }
187 }
188
189 /*
190 * Remove objects from the LDAP directory
191 */
192 private static void cleanup(DirContext ctx, String... dns)
193 throws NamingException {
194
195 for (String dn : dns) {
196 try {
197 ctx.destroySubcontext(dn);
198 System.out.println("StorePerson: removed entry '" + dn + "'");
199 } catch (NamingException e) {
200 System.err.println("StorePerson: error removing entry '" + dn +
201 "' " + e);
202 }
203 }
204 ctx.close();
205 }
206 }