Mega Code Archive

 
Categories / Java / J2ME
 

Pass a cookie (stored in rms) between the MIDlet and a Java servlet

/*-------------------------------------------------- * Cookie.java * * Pass a cookie (stored in rms) between the MIDlet * and a Java servlet. The cookie is generated   * by the servlet on the first visit. * * Example from the book:     Core J2ME Technology * Copyright John W. Muchow   http://www.CoreJ2ME.com * You may use/modify for any non-commercial purpose *-------------------------------------------------*/ import javax.microedition.midlet.*; import javax.microedition.lcdui.*; import javax.microedition.rms.*; import javax.microedition.io.*; import java.io.*;    public class Cookie extends MIDlet implements CommandListener {   private Display display;   private TextBox tbMain;   private Form fmMain;   private Command cmExit;   private Command cmLogon;   private String cookie = null;   private RecordStore rs = null;     static final String REC_STORE = "rms_cookie";     private String url = "http://www.mycgiserver.com/servlet/corej2me.CookieServlet";   public Cookie()   {     display = Display.getDisplay(this);     // Create commands     cmExit = new Command("Exit", Command.EXIT, 1);     cmLogon = new Command("Logon", Command.SCREEN, 2);              // Create the form, add commands, listen for events     fmMain = new Form("");     fmMain.addCommand(cmExit);     fmMain.addCommand(cmLogon);     fmMain.setCommandListener(this);     // Read cookie if available     openRecStore();        readCookie();       // System.out.println("Client cookie: " + cookie);           }   public void startApp()   {     display.setCurrent(fmMain);   }       public void pauseApp()   { }   public void destroyApp(boolean unconditional)   {      closeRecStore();  // Close record store   }   public void openRecStore()   {     try     {       // The second parameter indicates that the record store       // should be created if it does not exist       rs = RecordStore.openRecordStore(REC_STORE, true);     }     catch (Exception e)     {       db("open " + e.toString());     }   }          public void closeRecStore()   {     try     {       rs.closeRecordStore();     }     catch (Exception e)     {       db("close " + e.toString());     }   }   /*--------------------------------------------------   * Write cookie to rms   *-------------------------------------------------*/   public void writeRecord(String str)   {     byte[] rec = str.getBytes();     try     {       rs.addRecord(rec, 0, rec.length);     }     catch (Exception e)     {       db("write " + e.toString());     }   }   /*--------------------------------------------------   * Read cookie from rms   *-------------------------------------------------*/   public void readCookie()   {     try     {       byte[] recData = new byte[25];        int len;       if (rs.getNumRecords() > 0)       {         // Only one record will ever be written, safe to use '1'               if (rs.getRecordSize(1) > recData.length)           recData = new byte[rs.getRecordSize(1)];                  len = rs.getRecord(1, recData, 0);         cookie = new String(recData);       }     }     catch (Exception e)     {       db("read " + e.toString());     }   }   /*--------------------------------------------------   * Send client request and recieve server response   *   * Client: If cookie exists, send it to the server   *   * Server: If cookie is sent back, this is the    *         clients first request to the server. In   *         that case, save the cookie. If no cookie   *         sent, display server body (which indicates   *         the last time the MIDlet contacted server).   *-------------------------------------------------*/       private void connect() throws IOException   {     InputStream iStrm = null;     ByteArrayOutputStream bStrm = null;     HttpConnection http = null;              try     {       // Create the connection       http = (HttpConnection) Connector.open(url);       //----------------       // Client Request       //----------------       // 1) Send request method       http.setRequestMethod(HttpConnection.GET);             // If you experience connection/IO problems, try        // removing the comment from the following line       //http.setRequestProperty("Connection", "close");             // 2) Send header information       if (cookie != null)         http.setRequestProperty("cookie", cookie);                  System.out.println("Client cookie: " + cookie);             // 3) Send body/data - No data for this request             //----------------       // Server Response       //----------------       // 1) Get status Line       if (http.getResponseCode() == HttpConnection.HTTP_OK)       {         // 2) Get header information                  String tmpCookie = http.getHeaderField("set-cookie");                    System.out.println("server cookie: " + tmpCookie);                  // Cookie will only be sent back from server only if          // client (us) did not send a cookie in the first place.         // If a cookie is returned, we need to save it to rms         if (tmpCookie != null)         {           writeRecord(tmpCookie);                      // Update the MIDlet cookie variable           cookie = tmpCookie;                      fmMain.append("First visit\n");                     fmMain.append("Client : " + cookie + "\n");         }                 else  // No cookie sent from server         {           // 3) Get data, which is the last time of access           iStrm = http.openInputStream();           int length = (int) http.getLength();           String str;           if (length != -1)           {             byte serverData[] = new byte[length];             iStrm.read(serverData);             str = new String(serverData);           }           else  // Length not available...           {             bStrm = new ByteArrayOutputStream();                             int ch;             while ((ch = iStrm.read()) != -1)               bStrm.write(ch);             str = new String(bStrm.toByteArray());           }                    // Append data to the form                      fmMain.append("Last access:\n" + str + "\n");                            }       }     }     finally     {       // Clean up       if (iStrm != null)         iStrm.close();       if (bStrm != null)         bStrm.close();                       if (http != null)         http.close();     }   }      /*--------------------------------------------------   * Process events   *-------------------------------------------------*/   public void commandAction(Command c, Displayable s)   {     // If the Command button pressed was "Exit"     if (c == cmExit)     {       destroyApp(false);       notifyDestroyed();     }     else if (c == cmLogon)     {       try        {         // Logon to the servlet         connect();            }       catch (Exception e)       {         db("connect " + e.toString());               }     }   }   /*--------------------------------------------------   * Simple message to console for debug/errors   * When used with Exceptions we should handle the    * error in a more appropriate manner.   *-------------------------------------------------*/   private void db(String str)   {     System.err.println("Msg: " + str);   } } /*-------------------------------------------------- * CookieServlet.java * * Use a cookie to identify clients * * Example from the book:     Core J2ME Technology * Copyright John W. Muchow   http://www.CoreJ2ME.com * You may use/modify for any non-commercial purpose *-------------------------------------------------*/ //package corej2me; // Required for mycgiserver.com import java.util.*; import java.io.*; import javax.servlet.*; import javax.servlet.http.*; import java.sql.*; import java.text.*; public class CookieServlet extends HttpServlet {   // Pool of client ID's   private static int[] clientIDs = {123, 456, 789, 901, 225, 701};      protected void doGet(HttpServletRequest req, HttpServletResponse res)                         throws ServletException, IOException   {     // Get cookie from the header     Cookie[] cookies = req.getCookies();     //-------------------------------------------     // If cookie passed in...         // 1) Lookup the client ID in the database      //    and save the last access date     // 2) Update the last access date in database     // 3) Return to the client date from step 1     //-------------------------------------------         if (cookies != null)     {       // There will be only one cookie       Cookie theCookie = cookies[0];       String id = theCookie.getValue();       // Lookup client ID and get last access date       String strLastAccess = lookupLastAccessDate(Integer.parseInt(id));              // Update database with current date       updateLastAccessDate(Integer.parseInt(id));              // Send back the last access date to the client       res.setContentType("text/plain");           PrintWriter out = res.getWriter();       out.print(strLastAccess);       out.close();     }     else  // No Cookie     {       //-------------------------------------------       // Generate a client ID. To keep the database       // from growing out of control, this will not        // generate a new ID for each client.        // Instead, grab a random ID from the array         // clientID's[]. The end result is the same       // as far as the client is concerned.       //-------------------------------------------                    // Random value between 0 and the number of       // entries in the client list array       int random = (int) Math.round(clientIDs.length * Math.random());              // Get the client ID to send in the cookie       int ID = clientIDs[random];       // Update database with current date       updateLastAccessDate(ID);       // Create new cookie and send ID in the header       Cookie cookie = new Cookie("ID", Integer.toString(ID));       res.addCookie(cookie);        }       }     /*--------------------------------------------------   * Update database with last access date for client ID   *-------------------------------------------------*/    private void updateLastAccessDate(int ID)    {     Connection con = null;     Statement st = null;     StringBuffer msgb = new StringBuffer("");     try     {       // These will vary depending on your server/database             Class.forName("sun.jdbc.odbc.JdbcOdbcDriver");         con = DriverManager.getConnection("jdbc:odbc:acctInfo");       Statement stmt = con.createStatement();       // Create a date format           SimpleDateFormat format =            new SimpleDateFormat ("MMM dd-hh:mm aa");             String strDate = format.format(new java.util.Date());              ResultSet rs = stmt.executeQuery("UPDATE clientInfo set lastAccess = '" +                                          strDate + "' where clientID = " + ID);     }     catch (Exception e)     { }   }      /*--------------------------------------------------   * Lookup the client ID in database and get the    * last access date   *-------------------------------------------------*/   private String lookupLastAccessDate(int id)   {     Connection con = null;     Statement st = null;     StringBuffer msgb = new StringBuffer("");     try     {       // These will vary depending on your server/database                   Class.forName("sun.jdbc.odbc.JdbcOdbcDriver");         con = DriverManager.getConnection("jdbc:odbc:acctInfo");       Statement stmt = con.createStatement();       ResultSet rs = stmt.executeQuery("Select lastAccess from clientInfo where clientID = " + id);               if (rs.next())         return rs.getString(1);       else         return null;     }     catch (Exception e)     {       return e.toString();     }   }      /*--------------------------------------------------   * Information about servlet   *-------------------------------------------------*/        public String getServletInfo()   {     return "CookieServlet 1.0 - John W. Muchow - www.corej2me.com";   } }