| /* |
| Copyright The containerd Authors. |
| |
| Licensed 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 ttrpc |
| |
| import ( |
| "context" |
| "net" |
| ) |
| |
| // Handshaker defines the interface for connection handshakes performed on the |
| // server or client when first connecting. |
| type Handshaker interface { |
| // Handshake should confirm or decorate a connection that may be incoming |
| // to a server or outgoing from a client. |
| // |
| // If this returns without an error, the caller should use the connection |
| // in place of the original connection. |
| // |
| // The second return value can contain credential specific data, such as |
| // unix socket credentials or TLS information. |
| // |
| // While we currently only have implementations on the server-side, this |
| // interface should be sufficient to implement similar handshakes on the |
| // client-side. |
| Handshake(ctx context.Context, conn net.Conn) (net.Conn, interface{}, error) |
| } |
| |
| type handshakerFunc func(ctx context.Context, conn net.Conn) (net.Conn, interface{}, error) |
| |
| func (fn handshakerFunc) Handshake(ctx context.Context, conn net.Conn) (net.Conn, interface{}, error) { |
| return fn(ctx, conn) |
| } |
| |
| func noopHandshake(ctx context.Context, conn net.Conn) (net.Conn, interface{}, error) { |
| return conn, nil, nil |
| } |