Commit 6e837153 authored by li.can's avatar li.can Committed by 吴晟
Browse files

Add jdk http plugin (#3171) 

* add http

* update plugin

* Add StaticMethodsAroundInterceptor to be injected.

* move to options plugin

* move folder

* update pom

* support https

* add witness class

* Make JDK9-11 available for bootstrap instrumentation

* Remove temp folder of old bootstrap instrumentation requirement.

* remove witness ,work for jdk6,7,8,9,10,11

* Make JDK 12 works.

* move jdk-http-plugin to bootstrap-plugins folder

* add httpsclient

* update README.md
parent ff0ffb99

apm-protocol/apm-network/src/main/java/org/apache/skywalking/apm/network/trace/component/ComponentsDefine.java

+2 −0
Original line number Diff line number Diff line
@@ -126,6 +126,8 @@ public class ComponentsDefine { 


    public static final OfficialComponent SPRING_ASYNC = new OfficialComponent(65, "SpringAsync");



    public static final OfficialComponent JDK_HTTP = new OfficialComponent(66, "JdkHttp");



    private static ComponentsDefine INSTANCE = new ComponentsDefine();



    private String[] components;

apm-sniffer/bootstrap-plugins/jdk-http-plugin/pom.xml

0 → 100755
+46 −0
Original line number Diff line number Diff line 
<!--

  ~ Licensed to the Apache Software Foundation (ASF) under one or more

  ~ contributor license agreements.  See the NOTICE file distributed with

  ~ this work for additional information regarding copyright ownership.

  ~ The ASF licenses this file to You under the Apache License, Version 2.0

  ~ (the "License"); you may not use this file except in compliance with

  ~ the License.  You may obtain a copy of the License at

  ~

  ~     http://www.apache.org/licenses/LICENSE-2.0

  ~

  ~ Unless required by applicable law or agreed to in writing, software

  ~ distributed under the License is distributed on an "AS IS" BASIS,

  ~ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

  ~ See the License for the specific language governing permissions and

  ~ limitations under the License.

  ~

  -->



<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"

         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">

    <parent>

        <groupId>org.apache.skywalking</groupId>

        <artifactId>bootstrap-plugins</artifactId>

        <version>6.3.0-SNAPSHOT</version>

    </parent>

    <modelVersion>4.0.0</modelVersion>



    <artifactId>apm-jdk-http-plugin</artifactId>

    <packaging>jar</packaging>



    <name>apm-jdk-http-plugin</name>

    <url>http://maven.apache.org</url>



    <properties>

        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>

    </properties>





    <build>

        <plugins>

            <plugin>

                <artifactId>maven-deploy-plugin</artifactId>

            </plugin>

        </plugins>

    </build>

</project>

apm-sniffer/bootstrap-plugins/jdk-http-plugin/src/main/java/org/apache/skywalking/apm/plugin/jdk/http/HttpClientNewInstanceInterceptor.java

0 → 100644
+50 −0
Original line number Diff line number Diff line 
/*

 * Licensed to the Apache Software Foundation (ASF) under one or more

 * contributor license agreements.  See the NOTICE file distributed with

 * this work for additional information regarding copyright ownership.

 * The ASF licenses this file to You under the Apache License, Version 2.0

 * (the "License"); you may not use this file except in compliance with

 * the License.  You may obtain a copy of the License at

 *

 *     http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 *

 */



package org.apache.skywalking.apm.plugin.jdk.http;



import org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.EnhancedInstance;

import org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.MethodInterceptResult;

import org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.StaticMethodsAroundInterceptor;



import java.lang.reflect.Method;



/**

 * @author lican

 */

public class HttpClientNewInstanceInterceptor implements StaticMethodsAroundInterceptor {





    @Override

    public void beforeMethod(Class clazz, Method method, Object[] allArguments, Class<?>[] parameterTypes, MethodInterceptResult result) {



    }



    @Override

    public Object afterMethod(Class clazz, Method method, Object[] allArguments, Class<?>[] parameterTypes, Object ret) {

        if (ret instanceof EnhancedInstance) {

            ((EnhancedInstance) ret).setSkyWalkingDynamicField(allArguments[4]);

        }

        return ret;

    }



    @Override

    public void handleMethodException(Class clazz, Method method, Object[] allArguments, Class<?>[] parameterTypes, Throwable t) {



    }

}
 
 No newline at end of file

apm-sniffer/bootstrap-plugins/jdk-http-plugin/src/main/java/org/apache/skywalking/apm/plugin/jdk/http/HttpClientParseHttpInterceptor.java

0 → 100644
+90 −0
Original line number Diff line number Diff line 
/*

 * Licensed to the Apache Software Foundation (ASF) under one or more

 * contributor license agreements.  See the NOTICE file distributed with

 * this work for additional information regarding copyright ownership.

 * The ASF licenses this file to You under the Apache License, Version 2.0

 * (the "License"); you may not use this file except in compliance with

 * the License.  You may obtain a copy of the License at

 *

 *     http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 *

 */



package org.apache.skywalking.apm.plugin.jdk.http;



import org.apache.skywalking.apm.agent.core.context.ContextManager;

import org.apache.skywalking.apm.agent.core.context.tag.Tags;

import org.apache.skywalking.apm.agent.core.context.trace.AbstractSpan;

import org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.EnhancedInstance;

import org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.InstanceMethodsAroundInterceptor;

import org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.MethodInterceptResult;

import org.apache.skywalking.apm.util.StringUtil;

import sun.net.www.MessageHeader;



import java.lang.reflect.Method;



/**

 * @author lican

 */

public class HttpClientParseHttpInterceptor implements InstanceMethodsAroundInterceptor {





    @Override

    public void beforeMethod(EnhancedInstance objInst, Method method, Object[] allArguments, Class<?>[] argumentsTypes,

                             MethodInterceptResult result) throws Throwable {



    }





    @Override

    public Object afterMethod(EnhancedInstance objInst, Method method, Object[] allArguments, Class<?>[] argumentsTypes,

                              Object ret) throws Throwable {

        MessageHeader responseHeader = (MessageHeader) allArguments[0];

        String statusLine = responseHeader.getValue(0);

        Integer responseCode = parseResponseCode(statusLine);

        if (responseCode >= 400) {

            AbstractSpan span = ContextManager.activeSpan();

            span.errorOccurred();

            Tags.STATUS_CODE.set(span, Integer.toString(responseCode));

        }

        ContextManager.stopSpan();

        return ret;

    }



    @Override

    public void handleMethodException(EnhancedInstance objInst, Method method, Object[] allArguments,

                                      Class<?>[] argumentsTypes, Throwable t) {

        AbstractSpan span = ContextManager.activeSpan();

        span.errorOccurred().log(t);

    }





    /**

     * <PRE>

     * HTTP/1.0 200 OK

     * HTTP/1.0 401 Unauthorized

     * </PRE>

     * It will return 200 and 401 respectively.

     * Returns -1 if no code can be discerned

     */

    private Integer parseResponseCode(String statusLine) {

        if (!StringUtil.isEmpty(statusLine)) {

            String[] results = statusLine.split(" ");

            if (results.length >= 1) {

                try {

                    return Integer.valueOf(results[1]);

                } catch (Exception e) {

                }

            }

        }

        return -1;

    }





}
 
 No newline at end of file

apm-sniffer/bootstrap-plugins/jdk-http-plugin/src/main/java/org/apache/skywalking/apm/plugin/jdk/http/HttpClientWriteRequestInterceptor.java

0 → 100644
+90 −0
Original line number Diff line number Diff line 
/*

 * Licensed to the Apache Software Foundation (ASF) under one or more

 * contributor license agreements.  See the NOTICE file distributed with

 * this work for additional information regarding copyright ownership.

 * The ASF licenses this file to You under the Apache License, Version 2.0

 * (the "License"); you may not use this file except in compliance with

 * the License.  You may obtain a copy of the License at

 *

 *     http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 *

 */



package org.apache.skywalking.apm.plugin.jdk.http;



import org.apache.skywalking.apm.agent.core.context.CarrierItem;

import org.apache.skywalking.apm.agent.core.context.ContextCarrier;

import org.apache.skywalking.apm.agent.core.context.ContextManager;

import org.apache.skywalking.apm.agent.core.context.tag.Tags;

import org.apache.skywalking.apm.agent.core.context.trace.AbstractSpan;

import org.apache.skywalking.apm.agent.core.context.trace.SpanLayer;

import org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.EnhancedInstance;

import org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.InstanceMethodsAroundInterceptor;

import org.apache.skywalking.apm.agent.core.plugin.interceptor.enhance.MethodInterceptResult;

import org.apache.skywalking.apm.network.trace.component.ComponentsDefine;

import org.apache.skywalking.apm.util.StringUtil;

import sun.net.www.MessageHeader;



import java.lang.reflect.Method;

import java.net.HttpURLConnection;

import java.net.URL;



/**

 * @author lican

 */

public class HttpClientWriteRequestInterceptor implements InstanceMethodsAroundInterceptor {





    @Override

    public void beforeMethod(EnhancedInstance objInst, Method method, Object[] allArguments, Class<?>[] argumentsTypes,

                             MethodInterceptResult result) throws Throwable {

        HttpURLConnection connection = (HttpURLConnection) objInst.getSkyWalkingDynamicField();

        MessageHeader headers = (MessageHeader) allArguments[0];

        URL url = connection.getURL();

        ContextCarrier contextCarrier = new ContextCarrier();

        AbstractSpan span = ContextManager.createExitSpan(getPath(url), contextCarrier, getPeer(url));

        span.setComponent(ComponentsDefine.JDK_HTTP);

        Tags.HTTP.METHOD.set(span, connection.getRequestMethod());

        Tags.URL.set(span, url.toString());

        SpanLayer.asHttp(span);

        CarrierItem next = contextCarrier.items();

        while (next.hasNext()) {

            next = next.next();

            headers.add(next.getHeadKey(), next.getHeadValue());

        }

    }





    @Override

    public Object afterMethod(EnhancedInstance objInst, Method method, Object[] allArguments, Class<?>[] argumentsTypes,

                              Object ret) throws Throwable {

        return ret;

    }



    @Override

    public void handleMethodException(EnhancedInstance objInst, Method method, Object[] allArguments,

                                      Class<?>[] argumentsTypes, Throwable t) {

        AbstractSpan span = ContextManager.activeSpan();

        span.errorOccurred().log(t);

    }



    private String getPeer(URL url) {

        String host = url.getHost();

        if (url.getPort() > 0) {

            return host + ":" + url.getPort();

        }

        return host;

    }



    private String getPath(URL url) {

        return StringUtil.isEmpty(url.getPath()) ? "/" : url.getPath();

    }



}