import org.apache.axiom.om.util.UUIDGenerator; //导入依赖的package包/类
/**
 * Create a new axis MessageContext for an incoming message through this transport
 * @return the newly created message context
 */
public MessageContext createMessageContext() {
    MessageContext msgCtx = new MessageContext();
    msgCtx.setConfigurationContext(cfgCtx);

    msgCtx.setIncomingTransportName(getTransportName());
    msgCtx.setTransportOut(transportOut);
    msgCtx.setTransportIn(transportIn);
    msgCtx.setServerSide(true);
    msgCtx.setMessageID(UUIDGenerator.getUUID());

    // There is a discrepency in what I thought, Axis2 spawns a nes threads to
    // send a message is this is TRUE - and I want it to be the other way
    msgCtx.setProperty(MessageContext.CLIENT_API_NON_BLOCKING, Boolean.valueOf(!isNonBlocking));

    // are these relevant?
    //msgCtx.setServiceGroupContextId(UUIDGenerator.getUUID());
    // this is required to support Sandesha 2
    //msgContext.setProperty(RequestResponseTransport.TRANSPORT_CONTROL,
    //        new HttpCoreRequestResponseTransport(msgContext));

    return msgCtx;
}
 

import org.apache.axiom.om.util.UUIDGenerator; //导入依赖的package包/类
public void testSetPropertyInServiceGroupContext2() throws Exception {
    if (!canRunTests) {
        return;
    }

    String sgcID = UUIDGenerator.getUUID();

    ServiceGroupContext serviceGroupContext1 =
            configurationContext1.createServiceGroupContext(serviceGroup1);
    serviceGroupContext1.setId(sgcID);
    configurationContext1.addServiceGroupContextIntoSoapSessionTable(serviceGroupContext1);
    assertNotNull(serviceGroupContext1);

    ServiceGroupContext serviceGroupContext2 =
            configurationContext2.createServiceGroupContext(serviceGroup2);
    serviceGroupContext2.setId(sgcID);
    configurationContext2.addServiceGroupContextIntoSoapSessionTable(serviceGroupContext2);
    assertNotNull(serviceGroupContext2);

    String key1 = "sgCtxKey";
    String val1 = "sgCtxVal1";
    serviceGroupContext1.setProperty(key1, val1);
    ctxMan1.updateContext(serviceGroupContext1);

    assertEquals(val1, serviceGroupContext2.getProperty(key1));
}
 

import org.apache.axiom.om.util.UUIDGenerator; //导入依赖的package包/类
public void testDataSourceWithTXTPlusAttachment() throws Exception {
        Dispatch<DataSource> dispatch = getDispatch();

        Map attachments = new HashMap();
        Map requestContext = dispatch.getRequestContext();

//        requestContext.put(org.apache.axis2.transport.http.HTTPConstants.SO_TIMEOUT , new 
//        Integer(999999));
//        requestContext.put(org.apache.axis2.transport.http.HTTPConstants.CONNECTION_TIMEOUT, new 
//        Integer(999999));

        requestContext.put(MessageContext.OUTBOUND_MESSAGE_ATTACHMENTS,
                attachments);
        attachments.put(UUIDGenerator.getUUID(), new DataHandler(attachmentDS));

        DataSource request = txtDS;
        DataSource response = dispatch.invoke(request);
        assertTrue(response != null);
        assertEquals(response.getContentType(),"text/plain");
        String req = new String(getStreamAsByteArray(request.getInputStream()));
        String res = new String(getStreamAsByteArray(response.getInputStream()));
        assertEquals(req, res);
        Map attachments2 = (Map) dispatch.getResponseContext().get(MessageContext.INBOUND_MESSAGE_ATTACHMENTS);
        assertTrue(attachments2 != null);
        assertEquals(attachments2.size(), 1);
    }
 

import org.apache.axiom.om.util.UUIDGenerator; //导入依赖的package包/类
public void testDataSourceWithImagePlusAttachment() throws Exception {
    Dispatch<DataSource> dispatch = getDispatch();

    Map attachments = new HashMap();
    Map requestContext = dispatch.getRequestContext();

    requestContext.put(MessageContext.OUTBOUND_MESSAGE_ATTACHMENTS,
            attachments);
    attachments.put(UUIDGenerator.getUUID(), new DataHandler(attachmentDS));
    
    DataSource request = imageDS;
    DataSource response = dispatch.invoke(request);
    assertTrue(response != null);
    assertEquals(response.getContentType(),"image/jpeg");
    assertTrue(Arrays.equals(getStreamAsByteArray(request.getInputStream()), 
            getStreamAsByteArray(response.getInputStream())));
    Map attachments2 = (Map) dispatch.getResponseContext().get(MessageContext.INBOUND_MESSAGE_ATTACHMENTS);
    assertTrue(attachments2 != null);
    assertEquals(attachments2.size(), 1);
}
 

import org.apache.axiom.om.util.UUIDGenerator; //导入依赖的package包/类
public void testDataSourceWithTXTPlusTwoAttachments() throws Exception {
    Dispatch<DataSource> dispatch = getDispatch();

    Map attachments = new HashMap();
    Map requestContext = dispatch.getRequestContext();

    requestContext.put(MessageContext.OUTBOUND_MESSAGE_ATTACHMENTS,
            attachments);
    attachments.put(UUIDGenerator.getUUID(), new DataHandler(attachmentDS));
    attachments.put(UUIDGenerator.getUUID(), new DataHandler(imageDS));

    DataSource request = txtDS;
    DataSource response = dispatch.invoke(request);
    assertTrue(response != null);
    assertEquals(response.getContentType(),"text/plain");
    String req = new String(getStreamAsByteArray(request.getInputStream()));
    String res = new String(getStreamAsByteArray(response.getInputStream()));
    assertEquals(req, res);
    Map attachments2 = (Map) dispatch.getResponseContext().get(MessageContext.INBOUND_MESSAGE_ATTACHMENTS);
    assertTrue(attachments2 != null);
    assertEquals(attachments2.size(), 2);
}
 

import org.apache.axiom.om.util.UUIDGenerator; //导入依赖的package包/类
public void addPolicyToAllBindings(AxisService axisService, Policy policy)
        throws ServerException {
    try {
        if (policy.getId() == null) {
            // Generate an ID
            policy.setId(UUIDGenerator.getUUID());
        }
        Map endPointMap = axisService.getEndpoints();
        for (Object o : endPointMap.entrySet()) {
            Map.Entry entry = (Map.Entry) o;
            AxisEndpoint point = (AxisEndpoint) entry.getValue();
            AxisBinding binding = point.getBinding();
            String bindingName = binding.getName().getLocalPart();

            //only UTOverTransport is allowed for HTTP
            if (bindingName.endsWith("HttpBinding") &&
                    (!policy.getAttributes().containsValue("UTOverTransport"))) {
                continue;
            }
            binding.getPolicySubject().attachPolicy(policy);
            // Add the new policy to the registry
        }
    } catch (Exception e) {
        log.error("Error in adding security policy to all bindings", e);
        throw new ServerException("addPoliciesToService", e);
    }
}
 

import org.apache.axiom.om.util.UUIDGenerator; //导入依赖的package包/类
/**
 * This method add Policy to service at the Registry. Does not add the
 * policy to Axis2. To all Bindings available
 *
 * @param axisService Service
 * @param policy      Policy
 * @throws org.wso2.carbon.utils.ServerException se
 */
public void addSecurityPolicyToAllBindings(AxisService axisService, Policy policy)
        throws ServerException {
    try {
        if (policy.getId() == null) {
            policy.setId(UUIDGenerator.getUUID());
        }

        Map endPointMap = axisService.getEndpoints();
        for (Object o : endPointMap.entrySet()) {
            Map.Entry entry = (Map.Entry) o;
            AxisEndpoint point = (AxisEndpoint) entry.getValue();
            AxisBinding binding = point.getBinding();
            String bindingName = binding.getName().getLocalPart();

            //only UTOverTransport is allowed for HTTP
            if (bindingName.endsWith("HttpBinding") &&
                    (!policy.getAttributes().containsValue("UTOverTransport"))) {
                continue;
            }
            binding.getPolicySubject().attachPolicy(policy);

        }
    } catch (Exception e) {
        log.error("Error in adding security policy to all bindings", e);
        throw new ServerException("addPoliciesToService", e);
    }
}
 

import org.apache.axiom.om.util.UUIDGenerator; //导入依赖的package包/类
private MessageContext createMessageContext() throws AxisFault {
    MessageContext msgCtx = createSynapseMessageContext();
    org.apache.axis2.context.MessageContext axis2MsgCtx = ((Axis2MessageContext) msgCtx).getAxis2MessageContext();
    axis2MsgCtx.setServerSide(true);
    axis2MsgCtx.setMessageID(UUIDGenerator.getUUID());
    return msgCtx;
}
 

import org.apache.axiom.om.util.UUIDGenerator; //导入依赖的package包/类
/**
 * Create the message context.
 */
private MessageContext createMessageContext() {
    MessageContext msgCtx = this.synapseEnvironment.createMessageContext();
    org.apache.axis2.context.MessageContext axis2MsgCtx = ((Axis2MessageContext) msgCtx).getAxis2MessageContext();
    axis2MsgCtx.setServerSide(true);
    axis2MsgCtx.setMessageID(UUIDGenerator.getUUID());
    return msgCtx;
}
 

import org.apache.axiom.om.util.UUIDGenerator; //导入依赖的package包/类
/**
 * This method add Policy to service at the Registry. Does not add the
 * policy to Axis2. To all Bindings available
 *
 * @param axisService Service
 * @param policy      Policy
 * @throws org.wso2.carbon.utils.ServerException se
 */
public void addSecurityPolicyToAllBindings(AxisService axisService, Policy policy)
        throws ServerException {
    String serviceGroupId = axisService.getAxisServiceGroup().getServiceGroupName();
    try {
        if (policy.getId() == null) {
            policy.setId(UUIDGenerator.getUUID());
        }

        Map endPointMap = axisService.getEndpoints();
        for (Object o : endPointMap.entrySet()) {
            Map.Entry entry = (Map.Entry) o;
            AxisEndpoint point = (AxisEndpoint) entry.getValue();
            AxisBinding binding = point.getBinding();
            String bindingName = binding.getName().getLocalPart();

            //only UTOverTransport is allowed for HTTP
            if (bindingName.endsWith("HttpBinding") &&
                    (!policy.getAttributes().containsValue("UTOverTransport"))) {
                continue;
            }
            binding.getPolicySubject().attachPolicy(policy);

        }
    } catch (Exception e) {
        log.error("Error in adding security policy to all bindings", e);
        throw new ServerException("addPoliciesToService", e);
    }
}
 

import org.apache.axiom.om.util.UUIDGenerator; //导入依赖的package包/类
@Override
protected void updateMessageID() throws MessagingException {
    // although MailConstants.MAIL_HEADER_X_MESSAGE_ID solves the gmail problem with axis2-axis2
    // invocations it is not a generic solution.
    // we can over come gmail problem by setting the message id as follows with a valid gmail address
    // <[email protected]> this can be achived by appending from address at the end of uuid
 if (getHeader(MailConstants.MAIL_HEADER_MESSAGE_ID) == null) {
        String uuid = "<" + UUIDGenerator.getUUID().replaceAll(":",".") + fromAddress +">";
        setHeader(MailConstants.MAIL_HEADER_MESSAGE_ID, uuid);
    }
}
 

import org.apache.axiom.om.util.UUIDGenerator; //导入依赖的package包/类
protected String sendMessage(ContentType contentType, byte[] message) throws Exception {
    String msgId = UUIDGenerator.getUUID();
    MimeMessage msg = new MimeMessage(session);
    msg.setRecipients(Message.RecipientType.TO, InternetAddress.parse(channel.getRecipient().getAddress()));
    msg.setFrom(new InternetAddress(channel.getSender().getAddress()));
    msg.setSentDate(new Date());
    msg.setHeader(MailConstants.MAIL_HEADER_MESSAGE_ID, msgId);
    msg.setHeader(MailConstants.MAIL_HEADER_X_MESSAGE_ID, msgId);
    DataHandler dh = new DataHandler(new ByteArrayDataSource(message, contentType.toString()));
    layout.setupMessage(msg, dh);
    Transport.send(msg);
    return msgId;
}
 

import org.apache.axiom.om.util.UUIDGenerator; //导入依赖的package包/类
private MessageContext createMessageContext(OperationContext oc, ConfigurationContext cc,
                                                int flowType) throws Exception {
        MessageContext mc = cc.createMessageContext();

        mc.setFLOW(flowType);
        mc.setTransportIn(transportIn);
        mc.setTransportOut(transportOut);

        mc.setServerSide(true);
//        mc.setProperty(MessageContext.TRANSPORT_OUT, System.out);

        SOAPFactory omFac = OMAbstractFactory.getSOAP11Factory();
        mc.setEnvelope(omFac.getDefaultEnvelope());

        AxisOperation axisOperation = oc.getAxisOperation();
        String action = axisOperation.getName().getLocalPart();
        mc.setSoapAction(action);
//        System.out.flush();

        mc.setMessageID(UUIDGenerator.getUUID());

        axisOperation.registerOperationContext(mc, oc);
        mc.setOperationContext(oc);

        ServiceContext sc = oc.getServiceContext();
        mc.setServiceContext(sc);

        mc.setTo(new EndpointReference("axis2/services/NullService"));
        mc.setWSAAction("DummyOp");

        AxisMessage axisMessage = axisOperation.getMessage(WSDLConstants.MESSAGE_LABEL_IN_VALUE);
        mc.setAxisMessage(axisMessage);

        return mc;
    }
 

import org.apache.axiom.om.util.UUIDGenerator; //导入依赖的package包/类
private MessageContext createMessageContext(OperationContext oc) throws Exception {
        MessageContext mc = configurationContext.createMessageContext();
        mc.setTransportIn(transportIn);
        mc.setTransportOut(transportOut);

        mc.setServerSide(true);
//        mc.setProperty(MessageContext.TRANSPORT_OUT, System.out);

        SOAPFactory omFac = OMAbstractFactory.getSOAP11Factory();
        mc.setEnvelope(omFac.getDefaultEnvelope());

        AxisOperation axisOperation = oc.getAxisOperation();
        String action = axisOperation.getName().getLocalPart();
        mc.setSoapAction(action);
//        System.out.flush();

        mc.setMessageID(UUIDGenerator.getUUID());

        axisOperation.registerOperationContext(mc, oc);
        mc.setOperationContext(oc);

        ServiceContext sc = oc.getServiceContext();
        mc.setServiceContext(sc);

        mc.setTo(new EndpointReference("axis2/services/NullService"));
        mc.setWSAAction("DummyOp");

        AxisMessage axisMessage = axisOperation.getMessage(WSDLConstants.MESSAGE_LABEL_IN_VALUE);
        mc.setAxisMessage(axisMessage);

        return mc;
    }
 

import org.apache.axiom.om.util.UUIDGenerator; //导入依赖的package包/类
public void testRemovePropertyFromServiceGroupContext2() throws Exception {
    if (!canRunTests) {
        return;
    }

    // Add the property
    String sgcID = UUIDGenerator.getUUID();

    ServiceGroupContext serviceGroupContext1 =
            configurationContext1.createServiceGroupContext(serviceGroup1);
    serviceGroupContext1.setId(sgcID);
    configurationContext1.addServiceGroupContextIntoSoapSessionTable(serviceGroupContext1);
    assertNotNull(serviceGroupContext1);

    ServiceGroupContext serviceGroupContext2 =
            configurationContext2.createServiceGroupContext(serviceGroup2);
    serviceGroupContext2.setId(sgcID);
    configurationContext2.addServiceGroupContextIntoSoapSessionTable(serviceGroupContext2);
    assertNotNull(serviceGroupContext2);

    String key1 = "sgCtxKey";
    String val1 = "sgCtxVal1";
    serviceGroupContext1.setProperty(key1, val1);
    ctxMan1.updateContext(serviceGroupContext1);

    assertEquals(val1, serviceGroupContext2.getProperty(key1));

    // Remove the property
    serviceGroupContext2.removeProperty(key1);
    assertNull(serviceGroupContext2.getProperty(key1));
    ctxMan2.updateContext(serviceGroupContext2);
    assertNull(serviceGroupContext1.getProperty(key1));
}
 

import org.apache.axiom.om.util.UUIDGenerator; //导入依赖的package包/类
/**
 * @return ContentID
 */
public String getContentID() {
    if (cid == null) {
        cid = UUIDGenerator.getUUID();
        // Per spec, use the partName in the content-id
        // http://www.ws-i.org/Profiles/AttachmentsProfile-1.0.html#Value-space_of_Content-Id_Header
        if (partName != null) {
            cid = partName + "=" + cid;
        }
    }
    return cid;
}
 

import org.apache.axiom.om.util.UUIDGenerator; //导入依赖的package包/类
protected void setUp() throws Exception {
    //org.apache.log4j.BasicConfigurator.configure();

    invokecallcount = 0;

    mc = cfgContext.createMessageContext();
    mc.setTransportIn(transportIn);
    mc.setTransportOut(transportOut);

    mc.setServerSide(true);
    mc.setProperty(MessageContext.TRANSPORT_OUT, System.out);

    SOAPFactory omFac = OMAbstractFactory.getSOAP11Factory();
    mc.setEnvelope(omFac.getDefaultEnvelope());

    phase1 = new Phase("1");
    phase1.addHandler(new TempHandler02(0));
    phase1.addHandler(new TempHandler02(1));

    phase2 = new Phase("2");
    phase2.addHandler(new TempHandler02(2));
    phase2.addHandler(handler02);
    phase2.addHandler(new TempHandler02(3));

    phase3 = new Phase("3");
    phase3.addHandler(new TempHandler02(4));
    phase3.addHandler(subhandler);
    phase3.addHandler(handler02);  // same instance, second insertion
    phase3.addHandler(new TempHandler02(5));

    /*
     * TODO:  WARNING WARNING WARNING
     * Ideally inserting subPhase here would make the axis2 engine call
     * the invoke of nested subhandler.  It does not do this.  Please see the
     * warning at bottom of testPause06 method.
     */
    subPhase = new Phase("sub");
    subPhase.addHandler(subhandler);
    phase3.addHandler(subPhase);
    phase3.addHandler(new TempHandler02(6));
    phase3.addHandler(new TempHandler02(7));

    axisOperation.getRemainingPhasesInFlow().add(phase1);
    axisOperation.getRemainingPhasesInFlow().add(phase2);
    axisOperation.getRemainingPhasesInFlow().add(phase3);


    mc.setMessageID(UUIDGenerator.getUUID());

    //operationContext.addMessageContext(mc);  gets done via the register
    axisOperation.registerOperationContext(mc, operationContext);
    mc.setOperationContext(operationContext);
    mc.setServiceContext(serviceContext);

    mc.setTo(new EndpointReference("axis2/services/NullService"));

    mc.setWSAAction(operationName.getLocalPart());
    mc.setSoapAction(operationName.getLocalPart());

}