2 * Copyright 2008 Google Inc.
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.DecodeHintType;
20 import com.google.zxing.MultiFormatReader;
21 import com.google.zxing.Reader;
22 import com.google.zxing.ReaderException;
23 import com.google.zxing.Result;
24 import com.google.zxing.client.j2se.BufferedImageMonochromeBitmapSource;
25 import org.apache.commons.fileupload.FileItem;
26 import org.apache.commons.fileupload.FileUploadException;
27 import org.apache.commons.fileupload.disk.DiskFileItemFactory;
28 import org.apache.commons.fileupload.servlet.ServletFileUpload;
29 import org.apache.http.Header;
30 import org.apache.http.HttpException;
31 import org.apache.http.HttpMessage;
32 import org.apache.http.HttpResponse;
33 import org.apache.http.HttpVersion;
34 import org.apache.http.client.HttpClient;
35 import org.apache.http.client.methods.HttpGet;
36 import org.apache.http.conn.PlainSocketFactory;
37 import org.apache.http.conn.Scheme;
38 import org.apache.http.conn.SchemeRegistry;
39 import org.apache.http.conn.ssl.SSLSocketFactory;
40 import org.apache.http.impl.client.DefaultHttpClient;
41 import org.apache.http.impl.conn.tsccm.ThreadSafeClientConnManager;
42 import org.apache.http.params.BasicHttpParams;
43 import org.apache.http.params.HttpParams;
44 import org.apache.http.params.HttpProtocolParams;
46 import javax.imageio.ImageIO;
47 import javax.mail.Authenticator;
48 import javax.servlet.ServletConfig;
49 import javax.servlet.ServletException;
50 import javax.servlet.http.HttpServlet;
51 import javax.servlet.http.HttpServletRequest;
52 import javax.servlet.http.HttpServletResponse;
53 import java.awt.image.BufferedImage;
54 import java.io.IOException;
55 import java.io.InputStream;
56 import java.io.OutputStreamWriter;
57 import java.io.Writer;
59 import java.net.URISyntaxException;
60 import java.util.Hashtable;
61 import java.util.List;
62 import java.util.Timer;
63 import java.util.logging.Logger;
68 public final class DecodeServlet extends HttpServlet {
70 private static final long MAX_IMAGE_SIZE = 500000L;
71 private static final long EMAIL_CHECK_INTERVAL = 60000L;
73 private static final Logger log = Logger.getLogger(DecodeServlet.class.getName());
75 static final Hashtable<DecodeHintType, Object> HINTS;
78 HINTS = new Hashtable<DecodeHintType, Object>(3);
79 HINTS.put(DecodeHintType.TRY_HARDER, Boolean.TRUE);
82 private HttpClient client;
83 private DiskFileItemFactory diskFileItemFactory;
84 private Timer emailTimer;
87 public void init(ServletConfig servletConfig) throws ServletException {
89 Logger logger = Logger.getLogger("com.google.zxing");
90 logger.addHandler(new ServletContextLogHandler(servletConfig.getServletContext()));
92 String emailAddress = servletConfig.getInitParameter("emailAddress");
93 String emailPassword = servletConfig.getInitParameter("emailPassword");
94 if (emailAddress == null || emailPassword == null) {
95 throw new ServletException("emailAddress or emailPassword not specified");
98 HttpParams params = new BasicHttpParams();
99 HttpProtocolParams.setVersion(params, HttpVersion.HTTP_1_1);
101 SchemeRegistry registry = new SchemeRegistry();
102 registry.register(new Scheme("http", PlainSocketFactory.getSocketFactory(), 80));
103 registry.register(new Scheme("https", SSLSocketFactory.getSocketFactory(), 443));
105 client = new DefaultHttpClient(new ThreadSafeClientConnManager(params, registry), params);
107 diskFileItemFactory = new DiskFileItemFactory();
109 Authenticator emailAuthenticator = new EmailAuthenticator(emailAddress, emailPassword);
110 emailTimer = new Timer("Email decoder timer", true);
111 emailTimer.schedule(new DecodeEmailTask(emailAddress, emailAuthenticator), 0L, EMAIL_CHECK_INTERVAL);
113 log.info("DecodeServlet configured");
117 protected void doGet(HttpServletRequest request, HttpServletResponse response) throws IOException {
119 String imageURIString = request.getParameter("u");
120 if (imageURIString == null || imageURIString.length() == 0) {
121 response.sendRedirect("badurl.jspx");
125 if (!(imageURIString.startsWith("http://") || imageURIString.startsWith("https://"))) {
126 imageURIString = "http://" + imageURIString;
131 imageURI = new URI(imageURIString);
132 } catch (URISyntaxException urise) {
133 response.sendRedirect("badurl.jspx");
137 HttpGet getRequest = new HttpGet(imageURI);
140 HttpResponse getResponse = client.execute(getRequest);
141 if (getResponse.getStatusLine().getStatusCode() != HttpServletResponse.SC_OK) {
142 response.sendRedirect("badurl.jspx");
145 if (!isSizeOK(getResponse)) {
146 response.sendRedirect("badimage.jspx");
149 log.info("Decoding " + imageURI);
150 InputStream is = getResponse.getEntity().getContent();
152 processStream(is, response);
156 } catch (InterruptedException ie) {
158 response.sendRedirect("badurl.jspx");
159 } catch (HttpException he) {
161 response.sendRedirect("badurl.jspx");
167 protected void doPost(HttpServletRequest request, HttpServletResponse response)
168 throws ServletException, IOException {
170 if (!ServletFileUpload.isMultipartContent(request)) {
171 response.sendRedirect("badimage.jspx");
175 ServletFileUpload upload = new ServletFileUpload(diskFileItemFactory);
176 upload.setFileSizeMax(MAX_IMAGE_SIZE);
180 for (FileItem item : (List<FileItem>) upload.parseRequest(request)) {
181 if (!item.isFormField()) {
182 if (item.getSize() <= MAX_IMAGE_SIZE) {
183 log.info("Decoding uploaded file");
184 InputStream is = item.getInputStream();
186 processStream(is, response);
191 throw new ServletException("File is too large: " + item.getSize());
196 } catch (FileUploadException fue) {
197 response.sendRedirect("badimage.jspx");
202 private static void processStream(InputStream is, HttpServletResponse response) throws IOException {
203 BufferedImage image = ImageIO.read(is);
205 response.sendRedirect("badimage.jspx");
209 Reader reader = new MultiFormatReader();
212 result = reader.decode(new BufferedImageMonochromeBitmapSource(image), HINTS);
213 } catch (ReaderException re) {
214 log.info("DECODE FAILED: " + re.toString());
215 response.sendRedirect("notfound.jspx");
219 response.setContentType("text/plain");
220 response.setCharacterEncoding("UTF-8");
221 Writer out = new OutputStreamWriter(response.getOutputStream(), "UTF-8");
223 out.write(result.getText());
229 private static boolean isSizeOK(HttpMessage getResponse) {
230 Header lengthHeader = getResponse.getLastHeader("Content-Length");
231 if (lengthHeader != null) {
232 long length = Long.parseLong(lengthHeader.getValue());
233 if (length > MAX_IMAGE_SIZE) {
241 public void destroy() {
242 log.config("DecodeServlet shutting down...");