chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,44 @@
|
||||
// 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.
|
||||
|
||||
#ifndef BRPC_TEST_FUZZING_FUZZ_COMMON_H
|
||||
#define BRPC_TEST_FUZZING_FUZZ_COMMON_H
|
||||
|
||||
#include "brpc/socket.h"
|
||||
#include "butil/endpoint.h"
|
||||
|
||||
// Create a valid Socket for use in fuzz harnesses that need a non-NULL Socket*.
|
||||
// Returns a raw Socket* that remains valid for the lifetime of the process
|
||||
// (held by the static SocketUniquePtr).
|
||||
inline brpc::Socket* get_fuzz_socket() {
|
||||
static brpc::SocketId sid = 0;
|
||||
static brpc::SocketUniquePtr sock_ptr;
|
||||
static bool initialized = false;
|
||||
|
||||
if (!initialized) {
|
||||
brpc::SocketOptions options;
|
||||
options.remote_side = butil::EndPoint(butil::IP_ANY, 7777);
|
||||
if (brpc::Socket::Create(options, &sid) == 0 &&
|
||||
brpc::Socket::Address(sid, &sock_ptr) == 0) {
|
||||
initialized = true;
|
||||
}
|
||||
}
|
||||
|
||||
return sock_ptr.get();
|
||||
}
|
||||
|
||||
#endif // BRPC_TEST_FUZZING_FUZZ_COMMON_H
|
||||
Reference in New Issue
Block a user