/** Get map info operation. */ private long mapInfoOp( final HttpServletRequest request, final HttpServletResponse response, final PersistenceManager pm, final ApiAccount apiAccount) throws IOException { String mapFileName = request.getParameter(PARAM_MAP_FILE_NAME); LOGGER.fine( "API account: " + apiAccount.getUser().getEmail() + ", map file name: " + mapFileName); if (mapFileName == null || mapFileName.isEmpty()) { LOGGER.warning("Missing map file name!"); response.sendError(HttpServletResponse.SC_BAD_REQUEST, "Missing map file name!"); return 0; } if ((mapFileName = ServerUtils.checkMapFileName(mapFileName)) == null) { LOGGER.warning("Invalid map file name: " + mapFileName); response.sendError(HttpServletResponse.SC_BAD_REQUEST, "Invalid map file name!"); return 0; } try { final XmlBuilder xb = new XmlBuilder("1.0"); long opsCharged = 0; final List<Map> mapList = new JQBuilder<>(pm, Map.class).filter("fname==p1", "String p1").get(mapFileName); if (mapList.isEmpty()) { // Register a task to process the map TaskServlet.register_processMapTask(mapFileName); xb.createResultElement(MapInfoResult.PROCESSING); } else { final Map map = mapList.get(0); switch (map.getStatus()) { case Map.STATUS_PROCESSING: xb.createResultElement(MapInfoResult.PROCESSING); break; case Map.STATUS_PARSING_ERROR: xb.createResultElement(MapInfoResult.PARSING_ERROR); break; case Map.STATUS_DL_ERROR: xb.createResultElement(MapInfoResult.DOWNLOAD_ERROR); break; case Map.STATUS_READY: { opsCharged = 1; xb.createResultElement(MapInfoResult.OK); Element element = xb.createElement(XTAG_MAP); element.setAttribute(XATTR_NAME, map.getName()); element.setAttribute(XATTR_WIDTH, Integer.toString(map.getMwidth())); element.setAttribute(XATTR_HEIGHT, Integer.toString(map.getMheight())); element = xb.createElement(XTAG_MAP_IMAGE); element.setAttribute(XATTR_FORMAT, "JPEG"); element.setAttribute(XATTR_SIZE, Integer.toString(map.getSize())); element.setAttribute(XATTR_WIDTH, Integer.toString(map.getWidth())); element.setAttribute(XATTR_HEIGHT, Integer.toString(map.getHeight())); element.setTextContent( javax.xml.bind.DatatypeConverter.printBase64Binary(map.getImage().getBytes())); break; } } } xb.printDocument(response); return opsCharged; } catch (final Exception e) { LOGGER.log(Level.SEVERE, "", e); response.sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR); return 0; } }
@Override protected void doPost(final HttpServletRequest request, final HttpServletResponse response) throws ServletException, IOException { final long startNanoTime = System.nanoTime(); final String operation = checkProtVerAndGetOperation(PROTOCOL_VERSION_1, request, response); if (operation == null) return; final String apiKey = request.getParameter(PARAM_API_KEY); if (apiKey == null || apiKey.isEmpty()) { LOGGER.warning("Missing API key!"); response.sendError(HttpServletResponse.SC_BAD_REQUEST, "Missing API key!"); return; } long opsCharged = 0; ApiAccount apiAccount = null; PersistenceManager pm = null; ResponseWrapper responseWrapper = null; boolean denied = false; try { pm = PMF.get().getPersistenceManager(); // Check API key final List<ApiAccount> apiAccountList = new JQBuilder<>(pm, ApiAccount.class).filter("apiKey==p1", "String p1").get(apiKey); if (apiAccountList.isEmpty()) { LOGGER.warning("Unauthorized access, invalid API Key: " + apiKey); response.sendError( HttpServletResponse.SC_FORBIDDEN, "Unauthorized access, invalid API Key!"); return; } apiAccount = apiAccountList.get(0); responseWrapper = new ResponseWrapper(response); // Check Ops quota final List<ApiCallStat> totalApiCallStatList = new JQBuilder<>(pm, ApiCallStat.class) .filter("ownerKey==p1 && day==p2", "KEY p1, String p2") .get(apiAccount.getKey(), ApiCallStat.DAY_TOTAL); final long totalUsedOps = totalApiCallStatList.isEmpty() ? 0 : totalApiCallStatList.get(0).getUsedOps(); if (!OPERATION_INFO.equals(operation) && totalUsedOps >= apiAccount.getPaidOps()) { denied = true; LOGGER.warning( "Ops quota have been exceeded, serving denied! (API account: " + apiAccount.getUser().getEmail() + ")"); responseWrapper.sendError( HttpServletResponse.SC_PAYMENT_REQUIRED, "Ops quota have been exceeded, serving denied!"); return; } switch (operation) { case OPERATION_INFO: opsCharged = infoOp(request, responseWrapper, pm, apiAccount); break; case OPERATION_MAP_INFO: opsCharged = mapInfoOp(request, responseWrapper, pm, apiAccount); break; case OPERATION_PARSE_REPLAY: opsCharged = parseRepOp(request, responseWrapper, pm, apiAccount); break; case OPERATION_PROFILE_INFO: opsCharged = profInfoOp(request, responseWrapper, pm, apiAccount); break; default: LOGGER.warning( "Invalid Operation! (API account: " + apiAccount.getUser().getEmail() + ")"); responseWrapper.sendError(HttpServletResponse.SC_BAD_REQUEST, "Invalid Operation!"); return; } // Notification available Ops will be checked in the task servlet, update API call stat task } finally { if (apiAccount != null) TaskServlet.register_updateApiCallStat( apiAccount.getKey(), apiAccount.getPaidOps(), apiAccount.getNotificationAvailOps(), opsCharged, (System.nanoTime() - startNanoTime) / 1000000l, denied, responseWrapper == null ? true : responseWrapper.isError(), operation); if (pm != null) pm.close(); } }