2 * Copyright 2008 ZXing authors
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
17 package com.google.zxing.web;
19 import com.google.zxing.BarcodeFormat;
20 import com.google.zxing.BinaryBitmap;
21 import com.google.zxing.DecodeHintType;
22 import com.google.zxing.LuminanceSource;
23 import com.google.zxing.MultiFormatReader;
24 import com.google.zxing.Reader;
25 import com.google.zxing.ReaderException;
26 import com.google.zxing.Result;
27 import com.google.zxing.client.j2se.BufferedImageLuminanceSource;
28 import com.google.zxing.client.result.ParsedResult;
29 import com.google.zxing.client.result.ResultParser;
30 import com.google.zxing.common.GlobalHistogramBinarizer;
31 import com.google.zxing.common.HybridBinarizer;
33 import org.apache.commons.fileupload.FileItem;
34 import org.apache.commons.fileupload.FileUploadException;
35 import org.apache.commons.fileupload.disk.DiskFileItemFactory;
36 import org.apache.commons.fileupload.servlet.ServletFileUpload;
37 import org.apache.commons.lang.StringEscapeUtils;
38 import org.apache.http.Header;
39 import org.apache.http.HttpMessage;
40 import org.apache.http.HttpResponse;
41 import org.apache.http.HttpVersion;
42 import org.apache.http.HttpEntity;
43 import org.apache.http.client.HttpClient;
44 import org.apache.http.client.methods.HttpGet;
45 import org.apache.http.client.methods.HttpUriRequest;
46 import org.apache.http.conn.scheme.PlainSocketFactory;
47 import org.apache.http.conn.scheme.Scheme;
48 import org.apache.http.conn.scheme.SchemeRegistry;
49 import org.apache.http.conn.ssl.SSLSocketFactory;
50 import org.apache.http.conn.ClientConnectionManager;
51 import org.apache.http.impl.client.DefaultHttpClient;
52 import org.apache.http.impl.conn.SingleClientConnManager;
53 import org.apache.http.params.BasicHttpParams;
54 import org.apache.http.params.HttpParams;
55 import org.apache.http.params.HttpProtocolParams;
57 import java.awt.image.BufferedImage;
58 import java.io.IOException;
59 import java.io.InputStream;
60 import java.io.OutputStreamWriter;
61 import java.io.Writer;
62 import java.net.SocketException;
64 import java.net.URISyntaxException;
65 import java.net.UnknownHostException;
66 import java.util.Collection;
67 import java.util.Hashtable;
68 import java.util.List;
69 import java.util.Vector;
70 import java.util.logging.Logger;
72 import javax.imageio.ImageIO;
73 import javax.servlet.ServletConfig;
74 import javax.servlet.ServletException;
75 import javax.servlet.ServletRequest;
76 import javax.servlet.http.HttpServlet;
77 import javax.servlet.http.HttpServletRequest;
78 import javax.servlet.http.HttpServletResponse;
81 * {@link HttpServlet} which decodes images containing barcodes. Given a URL, it will
82 * retrieve the image and decode it. It can also process image files uploaded via POST.
86 public final class DecodeServlet extends HttpServlet {
88 private static final long MAX_IMAGE_SIZE = 500000L;
90 private static final Logger log = Logger.getLogger(DecodeServlet.class.getName());
92 static final Hashtable<DecodeHintType, Object> HINTS;
95 HINTS = new Hashtable<DecodeHintType, Object>(5);
96 HINTS.put(DecodeHintType.TRY_HARDER, Boolean.TRUE);
97 Collection<BarcodeFormat> possibleFormats = new Vector<BarcodeFormat>();
98 possibleFormats.add(BarcodeFormat.UPC_A);
99 possibleFormats.add(BarcodeFormat.UPC_E);
100 possibleFormats.add(BarcodeFormat.EAN_8);
101 possibleFormats.add(BarcodeFormat.EAN_13);
102 possibleFormats.add(BarcodeFormat.CODE_39);
103 possibleFormats.add(BarcodeFormat.CODE_128);
104 possibleFormats.add(BarcodeFormat.ITF);
105 possibleFormats.add(BarcodeFormat.RSS14);
106 possibleFormats.add(BarcodeFormat.QR_CODE);
107 possibleFormats.add(BarcodeFormat.DATAMATRIX);
108 possibleFormats.add(BarcodeFormat.PDF417);
109 HINTS.put(DecodeHintType.POSSIBLE_FORMATS, possibleFormats);
112 private HttpParams params;
113 private SchemeRegistry registry;
114 private DiskFileItemFactory diskFileItemFactory;
117 public void init(ServletConfig servletConfig) {
119 Logger logger = Logger.getLogger("com.google.zxing");
120 logger.addHandler(new ServletContextLogHandler(servletConfig.getServletContext()));
122 params = new BasicHttpParams();
123 HttpProtocolParams.setVersion(params, HttpVersion.HTTP_1_1);
125 registry = new SchemeRegistry();
126 registry.register(new Scheme("http", PlainSocketFactory.getSocketFactory(), 80));
127 registry.register(new Scheme("https", SSLSocketFactory.getSocketFactory(), 443));
129 diskFileItemFactory = new DiskFileItemFactory();
131 log.info("DecodeServlet configured");
135 protected void doGet(HttpServletRequest request, HttpServletResponse response)
136 throws ServletException, IOException {
137 String imageURIString = request.getParameter("u");
138 if (imageURIString == null || imageURIString.length() == 0) {
139 response.sendRedirect("badurl.jspx");
143 if (!(imageURIString.startsWith("http://") || imageURIString.startsWith("https://"))) {
144 imageURIString = "http://" + imageURIString;
149 imageURI = new URI(imageURIString);
150 } catch (URISyntaxException urise) {
151 response.sendRedirect("badurl.jspx");
155 ClientConnectionManager connectionManager = new SingleClientConnManager(params, registry);
156 HttpClient client = new DefaultHttpClient(connectionManager, params);
158 HttpUriRequest getRequest = new HttpGet(imageURI);
159 getRequest.addHeader("Connection", "close"); // Avoids CLOSE_WAIT socket issue?
163 HttpResponse getResponse;
165 getResponse = client.execute(getRequest);
166 } catch (IllegalArgumentException iae) {
167 // Thrown if hostname is bad or null
169 response.sendRedirect("badurl.jspx");
171 } catch (SocketException se) {
172 // Thrown if hostname is bad or null
174 response.sendRedirect("badurl.jspx");
176 } catch (UnknownHostException uhe) {
178 response.sendRedirect("badurl.jspx");
182 if (getResponse.getStatusLine().getStatusCode() != HttpServletResponse.SC_OK) {
183 response.sendRedirect("badurl.jspx");
186 if (!isSizeOK(getResponse)) {
187 response.sendRedirect("badimage.jspx");
191 log.info("Decoding " + imageURI);
192 HttpEntity entity = getResponse.getEntity();
193 InputStream is = entity.getContent();
195 processStream(is, request, response);
197 entity.consumeContent();
202 connectionManager.shutdown();
208 protected void doPost(HttpServletRequest request, HttpServletResponse response)
209 throws ServletException, IOException {
211 if (!ServletFileUpload.isMultipartContent(request)) {
212 response.sendRedirect("badimage.jspx");
216 ServletFileUpload upload = new ServletFileUpload(diskFileItemFactory);
217 upload.setFileSizeMax(MAX_IMAGE_SIZE);
221 for (FileItem item : (List<FileItem>) upload.parseRequest(request)) {
222 if (!item.isFormField()) {
223 if (item.getSize() <= MAX_IMAGE_SIZE) {
224 log.info("Decoding uploaded file");
225 InputStream is = item.getInputStream();
227 processStream(is, request, response);
232 response.sendRedirect("badimage.jspx");
237 } catch (FileUploadException fue) {
238 response.sendRedirect("badimage.jspx");
243 private static void processStream(InputStream is, ServletRequest request,
244 HttpServletResponse response) throws ServletException, IOException {
245 BufferedImage image = ImageIO.read(is);
247 response.sendRedirect("badimage.jspx");
251 Reader reader = new MultiFormatReader();
254 LuminanceSource source = new BufferedImageLuminanceSource(image);
255 BinaryBitmap bitmap = new BinaryBitmap(new GlobalHistogramBinarizer(source));
256 result = reader.decode(bitmap, HINTS);
257 } catch (ReaderException re) {
259 // Try again with other binarizer
260 LuminanceSource source = new BufferedImageLuminanceSource(image);
261 BinaryBitmap bitmap = new BinaryBitmap(new HybridBinarizer(source));
262 result = reader.decode(bitmap, HINTS);
263 } catch (ReaderException re2) {
264 log.info("DECODE FAILED: " + re.toString());
265 response.sendRedirect("notfound.jspx");
270 if (request.getParameter("full") == null) {
271 response.setContentType("text/plain");
272 response.setCharacterEncoding("UTF8");
273 Writer out = new OutputStreamWriter(response.getOutputStream(), "UTF8");
275 out.write(result.getText());
280 request.setAttribute("result", result);
281 byte[] rawBytes = result.getRawBytes();
282 if (rawBytes != null) {
283 request.setAttribute("rawBytesString", arrayToString(rawBytes));
285 request.setAttribute("rawBytesString", "(Not applicable)");
287 String text = result.getText();
289 request.setAttribute("text", StringEscapeUtils.escapeXml(text));
291 request.setAttribute("text", "(Not applicable)");
293 ParsedResult parsedResult = ResultParser.parseResult(result);
294 request.setAttribute("parsedResult", parsedResult);
295 String displayResult = parsedResult.getDisplayResult();
296 if (displayResult != null) {
297 request.setAttribute("displayResult", StringEscapeUtils.escapeXml(displayResult));
299 request.setAttribute("displayResult", "(Not applicable)");
301 request.getRequestDispatcher("decoderesult.jspx").forward(request, response);
305 private static boolean isSizeOK(HttpMessage getResponse) {
306 Header lengthHeader = getResponse.getLastHeader("Content-Length");
307 if (lengthHeader != null) {
308 long length = Long.parseLong(lengthHeader.getValue());
309 if (length > MAX_IMAGE_SIZE) {
316 private static String arrayToString(byte[] bytes) {
317 int length = bytes.length;
318 StringBuilder result = new StringBuilder(length << 2);
321 int max = Math.min(i + 8, length);
322 for (int j = i; j < max; j++) {
323 int value = bytes[j] & 0xFF;
324 result.append(Integer.toHexString(value / 16));
325 result.append(Integer.toHexString(value % 16));
331 for (int j = i - 8; j < length; j++) {
332 result.append(Integer.toHexString(bytes[j] & 0xFF));
335 return result.toString();
339 public void destroy() {
340 log.config("DecodeServlet shutting down...");