test.sh

#!/bin/bash

# 获取所有 httpd 进程的 PID（通过 /opt/lampp/bin/httpd 匹配）
HTTPD_PIDS=$(pgrep -f "/opt/lampp/bin/httpd")

# 获取所有 mysqld 进程的 PID
MYSQLD_PIDS=$(pgrep mysqld)

# 转换为数组
HTTPD_PIDS_ARRAY=($HTTPD_PIDS)
MYSQLD_PIDS_ARRAY=($MYSQLD_PIDS)

echo "追踪的 httpd 进程 PIDs: $HTTPD_PIDS"
echo "追踪的 mysqld 进程 PIDs: $MYSQLD_PIDS"

# 构建 PID 过滤条件（返回不带结束/的过滤器，以便组合）
build_pid_filter() {
    local pids=("$@")
    if [ ${#pids[@]} -eq 0 ]; then
        echo "/false/"
        return
    fi

    local filter="/(pid == ${pids[0]}"
    for ((i=1; i<${#pids[@]}; i++)); do
        filter="$filter || pid == ${pids[$i]}"
    done
    echo "$filter)/"
}

# 构建不带斜杠的 PID 条件（用于组合其他条件）
build_pid_condition() {
    local pids=("$@")
    if [ ${#pids[@]} -eq 0 ]; then
        echo "false"
        return
    fi

    local condition="(pid == ${pids[0]}"
    for ((i=1; i<${#pids[@]}; i++)); do
        condition="$condition || pid == ${pids[$i]}"
    done
    echo "$condition)"
}

# 生成 httpd 和 mysqld 的过滤条件
HTTPD_FILTER=$(build_pid_filter "${HTTPD_PIDS_ARRAY[@]}")
MYSQLD_FILTER=$(build_pid_filter "${MYSQLD_PIDS_ARRAY[@]}")

# 生成不带斜杠的条件（用于 sys_exit 探针）
HTTPD_CONDITION=$(build_pid_condition "${HTTPD_PIDS_ARRAY[@]}")
MYSQLD_CONDITION=$(build_pid_condition "${MYSQLD_PIDS_ARRAY[@]}")

echo "HTTPD 过滤条件: $HTTPD_FILTER"
echo "MYSQLD 过滤条件: $MYSQLD_FILTER"

# 生成动态磁盘追踪脚本
cat > /tmp/trace_disk_dynamic.bt <<EOF
#!/usr/bin/env bpftrace

BEGIN {
    printf("开始追踪磁盘操作...\n");
}

tracepoint:block:block_rq_issue
$HTTPD_FILTER
{
    @disk_ops_httpd[comm, pid] = count();
    @disk_bytes_httpd[comm, pid] = sum(args->bytes);
    @total_disk_ops_httpd = @total_disk_ops_httpd + 1;
    @total_disk_bytes_httpd = @total_disk_bytes_httpd + args->bytes;
}

tracepoint:block:block_rq_issue
$MYSQLD_FILTER
{
    @disk_ops_mysqld[comm, pid] = count();
    @disk_bytes_mysqld[comm, pid] = sum(args->bytes);
    @total_disk_ops_mysqld = @total_disk_ops_mysqld + 1;
    @total_disk_bytes_mysqld = @total_disk_bytes_mysqld + args->bytes;
}

// 每15秒输出一次统计结果
interval:s:15
{
    printf("\n=== 磁盘操作统计 [%s] ===\n", strftime("%H:%M:%S", nsecs));
    printf("\nHTTPD 磁盘操作次数（最近15秒）:\n");
    print(@disk_ops_httpd);
    printf("\nHTTPD 磁盘操作字节（最近15秒）:\n");
    print(@disk_bytes_httpd);
    printf("\nMYSQLD 磁盘操作次数（最近15秒）:\n");
    print(@disk_ops_mysqld);
    printf("\nMYSQLD 磁盘操作字节（最近15秒）:\n");
    print(@disk_bytes_mysqld);
    printf("==============================\n");

    clear(@disk_ops_httpd);
    clear(@disk_bytes_httpd);
    clear(@disk_ops_mysqld);
    clear(@disk_bytes_mysqld);
}

END {
    printf("\n=== 最终磁盘操作统计 ===\n");

    printf("\nHTTPD 各进程磁盘操作次数:\n");
    print(@disk_ops_httpd);
    printf("\nHTTPD 各进程磁盘操作字节:\n");
    print(@disk_bytes_httpd);

    printf("\nMYSQLD 各进程磁盘操作次数:\n");
    print(@disk_ops_mysqld);
    printf("\nMYSQLD 各进程磁盘操作字节:\n");
    print(@disk_bytes_mysqld);

    printf("\n--- 汇总统计 ---\n");
    printf("所有 HTTPD 进程磁盘操作总次数: %lld\n", @total_disk_ops_httpd);
    printf("所有 HTTPD 进程磁盘操作总字节: %lld\n", @total_disk_bytes_httpd);
    printf("所有 MYSQLD 进程磁盘操作总次数: %lld\n", @total_disk_ops_mysqld);
    printf("所有 MYSQLD 进程磁盘操作总字节: %lld\n", @total_disk_bytes_mysqld);
}
EOF

# 生成动态网络追踪脚本（追踪网卡驱动层）
cat > /tmp/trace_network_dynamic.bt <<EOF
#!/usr/bin/env bpftrace

BEGIN {
    printf("开始追踪网络操作（网卡驱动层）...\n");
}

// 追踪发送数据包（网卡驱动层）
tracepoint:net:net_dev_queue
$HTTPD_FILTER
{
    @net_send_httpd[comm, pid] = count();
    @net_send_bytes_httpd[comm, pid] = sum(args->len);
    @total_net_send_httpd = @total_net_send_httpd + 1;
    @total_net_send_bytes_httpd = @total_net_send_bytes_httpd + args->len;
}

tracepoint:net:net_dev_queue
$MYSQLD_FILTER
{
    @net_send_mysqld[comm, pid] = count();
    @net_send_bytes_mysqld[comm, pid] = sum(args->len);
    @total_net_send_mysqld = @total_net_send_mysqld + 1;
    @total_net_send_bytes_mysqld = @total_net_send_bytes_mysqld + args->len;
}

// 追踪接收数据包（网卡驱动层）
tracepoint:net:netif_receive_skb
$HTTPD_FILTER
{
    @net_recv_httpd[comm, pid] = count();
    @net_recv_bytes_httpd[comm, pid] = sum(args->len);
    @total_net_recv_httpd = @total_net_recv_httpd + 1;
    @total_net_recv_bytes_httpd = @total_net_recv_bytes_httpd + args->len;
}

tracepoint:net:netif_receive_skb
$MYSQLD_FILTER
{
    @net_recv_mysqld[comm, pid] = count();
    @net_recv_bytes_mysqld[comm, pid] = sum(args->len);
    @total_net_recv_mysqld = @total_net_recv_mysqld + 1;
    @total_net_recv_bytes_mysqld = @total_net_recv_bytes_mysqld + args->len;
}

// 每15秒输出一次统计结果
interval:s:15
{
    printf("\n=== 网络操作统计 [%s] ===\n", strftime("%H:%M:%S", nsecs));
    printf("\nHTTPD 发送数据包次数（最近15秒）:\n");
    print(@net_send_httpd);
    printf("\nHTTPD 发送字节数（最近15秒）:\n");
    print(@net_send_bytes_httpd);
    printf("\nHTTPD 接收数据包次数（最近15秒）:\n");
    print(@net_recv_httpd);
    printf("\nHTTPD 接收字节数（最近15秒）:\n");
    print(@net_recv_bytes_httpd);

    printf("\nMYSQLD 发送数据包次数（最近15秒）:\n");
    print(@net_send_mysqld);
    printf("\nMYSQLD 发送字节数（最近15秒）:\n");
    print(@net_send_bytes_mysqld);
    printf("\nMYSQLD 接收数据包次数（最近15秒）:\n");
    print(@net_recv_mysqld);
    printf("\nMYSQLD 接收字节数（最近15秒）:\n");
    print(@net_recv_bytes_mysqld);
    printf("==============================\n");

    clear(@net_send_httpd);
    clear(@net_send_bytes_httpd);
    clear(@net_recv_httpd);
    clear(@net_recv_bytes_httpd);
    clear(@net_send_mysqld);
    clear(@net_send_bytes_mysqld);
    clear(@net_recv_mysqld);
    clear(@net_recv_bytes_mysqld);
}

END {
    printf("\n=== 最终网络操作统计 ===\n");

    printf("\nHTTPD 各进程发送数据包次数:\n");
    print(@net_send_httpd);
    printf("\nHTTPD 各进程发送字节数:\n");
    print(@net_send_bytes_httpd);
    printf("\nHTTPD 各进程接收数据包次数:\n");
    print(@net_recv_httpd);
    printf("\nHTTPD 各进程接收字节数:\n");
    print(@net_recv_bytes_httpd);

    printf("\nMYSQLD 各进程发送数据包次数:\n");
    print(@net_send_mysqld);
    printf("\nMYSQLD 各进程发送字节数:\n");
    print(@net_send_bytes_mysqld);
    printf("\nMYSQLD 各进程接收数据包次数:\n");
    print(@net_recv_mysqld);
    printf("\nMYSQLD 各进程接收字节数:\n");
    print(@net_recv_bytes_mysqld);

    printf("\n--- 汇总统计 ---\n");
    printf("所有 HTTPD 进程发送数据包总次数: %lld\n", @total_net_send_httpd);
    printf("所有 HTTPD 进程发送总字节数: %lld\n", @total_net_send_bytes_httpd);
    printf("所有 HTTPD 进程接收数据包总次数: %lld\n", @total_net_recv_httpd);
    printf("所有 HTTPD 进程接收总字节数: %lld\n", @total_net_recv_bytes_httpd);
    printf("所有 MYSQLD 进程发送数据包总次数: %lld\n", @total_net_send_mysqld);
    printf("所有 MYSQLD 进程发送总字节数: %lld\n", @total_net_send_bytes_mysqld);
    printf("所有 MYSQLD 进程接收数据包总次数: %lld\n", @total_net_recv_mysqld);
    printf("所有 MYSQLD 进程接收总字节数: %lld\n", @total_net_recv_bytes_mysqld);
}
EOF

echo "动态脚本已生成："
echo "  - /tmp/trace_disk_dynamic.bt"
echo "  - /tmp/trace_network_dynamic.bt"

# 启动磁盘追踪
sudo BPFTRACE_STACK_SIZE=1024 bpftrace /tmp/trace_disk_dynamic.bt > disk_trace.log 2>&1 &
DISK_TRACE_PID=$!

# 启动网络追踪
sudo BPFTRACE_STACK_SIZE=1024 bpftrace /tmp/trace_network_dynamic.bt > network_trace.log 2>&1 &
NETWORK_TRACE_PID=$!

# 等待 bpftrace 启动
sleep 2

echo "追踪已启动，PID: disk=$DISK_TRACE_PID, network=$NETWORK_TRACE_PID"

# 运行测试（取消注释以启用）
/home/mangp/miniconda3/bin/python3 wordpress_concurrent_test.py --users=100

echo "停止追踪..."
# 停止两个 bpftrace 进程
sudo kill -INT $DISK_TRACE_PID
sudo kill -INT $NETWORK_TRACE_PID

# 等待进程结束
wait $DISK_TRACE_PID 2>/dev/null
wait $NETWORK_TRACE_PID 2>/dev/null

echo "追踪完成！"
echo "磁盘追踪日志: disk_trace.log"
echo "网络追踪日志: network_trace.log"

# 显示统计结果
echo -e "\n=== 磁盘操作统计 ==="
grep -E "(HTTPD|MYSQLD) 磁盘" disk_trace.log

echo -e "\n=== 网络操作统计 ==="
grep -E "(HTTPD|MYSQLD) 网络" network_trace.log