appserver/tests/appserv-tests/util/webrunner/webtest/SendRequest.java

/*
 * Copyright (c) 2017, 2018 Oracle and/or its affiliates. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0, which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the
 * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
 * version 2 with the GNU Classpath Exception, which is available at
 * https://www.gnu.org/software/classpath/license.html.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
 */

package com.sun.ejte.ccl.webrunner.webtest;

import java.net.*;
import java.util.*;
import java.io.*;
import java.lang.*;
import java.util.logging.*;
import com.sun.ejte.ccl.reporter.*;

/**
 *This class is called from WebTest.
 *
 * @author       Deepa Singh (deepa.singh@sun.com)
 *Company       Sun Microsystems Inc.
 *
 */
public class SendRequest {

    private int nThreads;
    private int id ;
    private byte[] filebuf;
    private String website;
    private String cookie_name;
    private String cookie_value;
    private String cookie_path;
    Logger logger = Logger.getLogger("bank.admin");
    ConsoleHandler ch = new ConsoleHandler();
    private String TEST_SUITE_ID = "WEBCLIENT";

    private static String CRLF="\r\n";

    int status,OK=0;
    int notOK=0;
    int url_size;
    double start,end;
    boolean ifdone=false;
    boolean setCookie=false;
    boolean serverCookieSet=false;
    FileOutputStream errout=null;
    //int port=8000;
    private int m_port=80;
    int port;
    private String m_host=new String("local");
    boolean serverSet=false;
    private static int buffer_size=8192;
    private String ws_root;
    Reporter reporter=null;


    public SendRequest(String ws_root,String testsuite_id)
    {
        this.ws_root=ws_root;
        this.TEST_SUITE_ID=testsuite_id;
        reporter = Reporter.getInstance(ws_root);
    }





    public void setServerProperties(String host,int port) {
        this.m_host=host;
        this.m_port=port;
        serverSet=true;
    }



    /**This is a thread safe method so that multiple session of web application can be generated simulataneosly.
     *It parses the byte array and extract individual requests.Uses MimeHeader class for parsing MIME headers.Handles session management by checking if server has set "Set-Cookie" in it's response header.
     *@param host String host name of the web server where web application is to be run e.g. dsingh.sfbay.sun.com
     *@param filebuf byte[] byte array which contains script file.
     *@return int Status Code of web server response.
     *
     */
    public synchronized int processUrl(byte[] filebuf) {
            int responseCode=0;
            int nRequest=0;
            int code=0;
            reporter.setTestSuite(TEST_SUITE_ID, "J2EE Web Client","This test suite integrates web and ejb");


            try {
                    Socket server=null;
                    InetAddress serverAddress=null;
                    BufferedReader bufferedStream=null;
                    DataOutputStream server_out= null;

                    int pos=0;
                    int retVal=0;
                    double currentstart,currentstop;
                    double temp=0;
                    double time=0;
                    char c;
                    String hostName = new String("");

                    int i=0,j=0;
                    MimeHeader mh;
                    while(i<filebuf.length) {
                            c=(char)filebuf[i];

                            switch(c) {
                                    case '!':

                                            if(filebuf[i+1] =='\n' && filebuf[i+2]=='!') {
                                                    nRequest=nRequest+1;
                                                    String temp1=new String(filebuf,j,i-1-j);
                                                    mh=new MimeHeader(temp1);
                                                    if(setCookie) {
                                                            //Cookie: JSESSIONID=4659B62637AC12324972EA5072064469
                                                            //String full_cookie_value="JSESSIONID="+cookie_value+";Path="+cookie_path;
                                                            //String full_cookie_value="JSESSIONID="+cookie_value;
                                                            String full_cookie_value=cookie_name+"="+cookie_value;
                                                            full_cookie_value=full_cookie_value.trim();
                                                            String oldcookie=mh.get("Cookie");
                                                            //  System.out.println("Old Cookie value"+oldcookie);
                                                            mh.put("Cookie",full_cookie_value);
                                                    }

                                                    int stIdx = temp1.indexOf("Host: ");
                                                    stIdx+=5;
                                                    String tp = temp1.substring(stIdx);

                                                    int endIdx = tp.indexOf('\n');
                                                    endIdx+=stIdx;
                                                    hostName = temp1.substring(stIdx, endIdx);

                                                    int portIdx=hostName.indexOf(':');
                                                    String strport=new String();
                                                    if(portIdx > 0){
                                                    strport=hostName.substring(portIdx+1).trim();
                                                            hostName=hostName.substring(0,portIdx);
                                                    //System.out.println("port is"+strport);
}
                                                    hostName = hostName.trim();
                                                    port=new Integer(strport).intValue();
                                                    // if serverSet is true,then use m_host and m_port from the commandline
                                                    //otherwise read host and port from the script.txt
                                                    if(serverSet){
                                                    System.out.println("HTTP port :"+m_port);
                                                    System.out.println("HTTP hostname :"+m_host);
                                                    server=new Socket(m_host,m_port);
                                                    }
                                                    else
                                                        server=new Socket(hostName,port);
                                                    serverAddress=server.getInetAddress();
                                                    bufferedStream=new BufferedReader(new InputStreamReader(server.getInputStream()));
                                                    server_out= new DataOutputStream(server.getOutputStream());
                                                    String req=new String();
                                                    if(mh.ifPOSTRequest()) {
                                                            //System.out.println("POST REQUEST");
                                                            req=mh.getRequestHeader()+ CRLF + mh+CRLF+ mh.getPostData()+CRLF+CRLF;
                                                            //String post="POST" + " " + server_url+ " " + "HTTP/1.0" + CRLF + inmh
                                                            //+ CRLF + postdata+ CRLF+CRLF;

                                                    }
                                                    else
                                                            req=mh.getRequestHeader()+ CRLF + mh + CRLF + CRLF;

                                                    String requestLine=mh.getRequestHeader();
                                                    int fsp=requestLine.indexOf(' ');
                                                    int nsp=requestLine.indexOf(' ',fsp+1);
                                                    int eol=requestLine.indexOf('\n');
                                                    requestLine=requestLine.substring(fsp+1,nsp);

                                                    //Following are debug output statements
                                                    //System.out.println("%%%%%%%%%%%%%%%%%%%%%%Request String Sent to Server%%%%%%%%%%%%%%%%%%%");
                                                    //System.out.println(mh.getRequestHeader());
                                                    //System.out.println(mh);

                                                    try {
                                                            if(server_out!=null)
                                                                    server_out.write(req.getBytes());
                                                            else System.out.println("Server_out is null");
                                                    }
                                                    catch(IOException e) {
                                                            String errMsg=e.getMessage();
                                                            System.out.println(e.getMessage());
                                                    }

                                                    code = readHeader(bufferedStream);
                                                    reporter.addTest(TEST_SUITE_ID,requestLine );
                                                    reporter.addTestCase(TEST_SUITE_ID,requestLine, requestLine );


                                                    if(code==500 || code==404) {
                                                            if(code==404){
                                                                    System.out.println("WebServer returned error code 404");
                                                                    System.out.println("Request Resource not available");
                                                                    System.out.println("There is some deployment error");
                                                            }
                                                            System.out.println("!!Server returned error..Application Exiting");
                                                            reporter.setTestCaseStatus(TEST_SUITE_ID,requestLine,requestLine,ReporterConstants.FAIL);
                                                            //stat.addStatus("WEBCLIENT "+ requestLine,stat.FAIL);
                                                            System.out.println(TEST_SUITE_ID+"\t"+requestLine+"\t FAIL");
                                                            reporter.generateValidReport();
                                                            return 0;
                                                    }
                                                    if (code==200){
                                                            reporter.setTestCaseStatus(TEST_SUITE_ID,requestLine,requestLine,ReporterConstants.PASS);
                                                            //stat.addStatus("WEBCLIENT"+" "+ requestLine,stat.PASS);
                                                            System.out.println(TEST_SUITE_ID+"\t"+requestLine+"\t"+"PASS");
                                                    }

                                            }
                                            i=i+3;
                                            j=i;

                                    default:

                                            i=i+1;
                            }
                            if(time>=temp)
                                    temp=time;

                    }
                    // System.out.println("while loop ended");
                    bufferedStream.close();
                    server_out.close();
                    reporter.generateValidReport();
                    return code;
            }
            catch(Exception e) {
                    e.printStackTrace();
                    String msg=e.getMessage();
            }
            return code;
    }

    public synchronized int readHeader(BufferedReader inStream) {
        int status=0;
        String reasonPhrase = new String("");
        String st=null;
        try {
            StringBuffer result=new StringBuffer();
            st=inStream.readLine();
                /*if(st!=null)
                System.out.println("*****Server Response Starts *****\n"+st );*/

            //st should be of form HTTP/1.1 200 OK
            int fsp=st.indexOf(' ');
            int nsp=st.indexOf(' ',fsp+1);
            int eol=st.indexOf('\n');
            status=Integer.parseInt(st.substring(fsp+1,nsp));
            String line;
            while((line=inStream.readLine())!=null) {

                //Trying to parse
                //Set-Cookie: JSESSIONID=E649A94960EB0304CFA985067EF90893;Path=/bookstore1
                if(line.startsWith("Set-Cookie:") && !setCookie) {
                    if(line!=null) {
                        //   System.out.println("Setting cookie from server");
                        //  System.out.println("COOKIES\t"+line);
                    }

                    int colon=line.indexOf(':');
                    int firstEquals=line.indexOf('=');
                    int semicolon=line.indexOf(';');
                    cookie_name=line.substring(colon+1,firstEquals).trim(); //gives JSESSIONID
                    cookie_value=line.substring(firstEquals+1,semicolon).trim();//gives SessionID
                    cookie_path=line.substring(line.indexOf("Path")+5).trim();
                    setCookie=true;
                }
                if(line!=null) {
                    StringBuffer server_response=new StringBuffer();
                    server_response.append(line);
                    String response=server_response.toString();

                    //System.out.println(line);
                }


            }
            //System.out.println("************Server Response Ends*****************");

        }
        catch(Exception e) {
            System.err.println("-------------------- Read Header Fail---------------------------");
            System.err.println("Error details:" + e.getMessage());
            e.printStackTrace();

            return status;
        }

        return status;
    }

}