123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260 |
- /**
- * Tencent is pleased to support the open source community by making Tars available.
- *
- * Copyright (C) 2016THL A29 Limited, a Tencent company. All rights reserved.
- *
- * Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
- * in compliance with the License. You may obtain a copy of the License at
- *
- * https://opensource.org/licenses/BSD-3-Clause
- *
- * 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.
- */
- #include "util/tc_epoll_server.h"
- #include "util/tc_http.h"
- #include "util/tc_grpc.h"
- #include "servant/AppProtocol.h"
- // #include "servant/TC_Transceiver.h"
- #include "servant/AdapterProxy.h"
- #include "servant/RemoteLogger.h"
- #include "tup/Tars.h"
- #include <iostream>
- #if TARS_HTTP2
- #include "util/tc_http2.h"
- #endif
- namespace tars
- {
- class TC_Transceiver;
- shared_ptr<TC_NetWorkBuffer::Buffer> ProxyProtocol::toBuffer(TarsOutputStream<BufferWriter> &os)
- {
- shared_ptr<TC_NetWorkBuffer::Buffer> buff = std::make_shared<TC_NetWorkBuffer::Buffer>();
- buff->replaceBuffer(os.getBuffer(), os.getLength());
- os._buf = NULL;
- os._buf_len = 0;
- os._len = 0;
- return buff;
- }
- shared_ptr<TC_NetWorkBuffer::Buffer> ProxyProtocol::tarsRequest(RequestPacket& request, TC_Transceiver *)
- {
- TarsOutputStream<BufferWriter> os;
- int iHeaderLen = 0;
- // 先预留4个字节长度
- os.writeBuf((const char *)&iHeaderLen, sizeof(iHeaderLen));
- request.writeTo(os);
- assert(os.getLength() >= 4);
- iHeaderLen = htonl((int)(os.getLength()));
- memcpy((void*)os.getBuffer(), (const char *)&iHeaderLen, sizeof(iHeaderLen));
- return toBuffer(os);
- }
- ////////////////////////////////////////////////////////////////////////////////////
- shared_ptr<TC_NetWorkBuffer::Buffer> ProxyProtocol::http1Request(tars::RequestPacket& request, TC_Transceiver *trans)
- {
- shared_ptr<TC_HttpRequest> &data = *(shared_ptr<TC_HttpRequest>*)request.sBuffer.data();
- shared_ptr<TC_NetWorkBuffer::Buffer> buff = std::make_shared<TC_NetWorkBuffer::Buffer>();
- if(!data->hasHeader("Host")) {
- data->setHost(trans->getEndpoint().getHost());
- }
- data->encode(buff);
- data.reset();
- return buff;
- }
- TC_NetWorkBuffer::PACKET_TYPE ProxyProtocol::http1Response(TC_NetWorkBuffer &in, ResponsePacket& rsp)
- {
- shared_ptr<TC_HttpResponse> *context = (shared_ptr<TC_HttpResponse>*)(in.getContextData());
- if(!context)
- {
- context = new shared_ptr<TC_HttpResponse>();
- *context = std::make_shared<TC_HttpResponse>();
- in.setContextData(context, [](TC_NetWorkBuffer*nb){ shared_ptr<TC_HttpResponse> *p = (shared_ptr<TC_HttpResponse>*)(nb->getContextData()); if(!p) { nb->setContextData(NULL); delete p; }});
- }
- if((*context)->incrementDecode(in))
- {
- rsp.sBuffer.resize(sizeof(shared_ptr<TC_HttpResponse>));
- shared_ptr<TC_HttpResponse> &data = *(shared_ptr<TC_HttpResponse>*)rsp.sBuffer.data();
- data = *context;
- auto ret = TC_NetWorkBuffer::PACKET_FULL;
- if (data->checkHeader("Connection", "keep-alive") || (data->getVersion() == "HTTP/1.1" && !data->hasHeader("Connection")))
- {
- ret = TC_NetWorkBuffer::PACKET_FULL;
- }
- else
- {
- ret = TC_NetWorkBuffer::PACKET_FULL_CLOSE;
- }
- //收取到完整的包, 把当前包释放掉, 下次新包来会新建context
- (*context) = NULL;
- delete context;
- in.setContextData(NULL);
- return ret;
- }
- return TC_NetWorkBuffer::PACKET_LESS;
- }
- /////////////////////////////////////////////////////////////////////////////////////////////////
- #if TARS_HTTP2
- // ENCODE function, called by network thread
- shared_ptr<TC_NetWorkBuffer::Buffer> ProxyProtocol::http2Request(RequestPacket& request, TC_Transceiver *trans)
- {
- TC_Http2Client* session = (TC_Http2Client*)trans->getSendBuffer().getContextData();
- if(session == NULL)
- {
- session = new TC_Http2Client();
- trans->getSendBuffer().setContextData(session, [=](TC_NetWorkBuffer*){delete session;});
- session->settings(3000);
- }
- shared_ptr<TC_HttpRequest> *data = (shared_ptr<TC_HttpRequest>*)request.sBuffer.data();
- request.iRequestId = session->submit(*(*data).get());
- //这里把智能指针释放一次
- (*data).reset();
- shared_ptr<TC_NetWorkBuffer::Buffer> buff = std::make_shared<TC_NetWorkBuffer::Buffer>();
- if (request.iRequestId < 0)
- {
- TLOGERROR("http2Request::Fatal submit error: " << session->getErrMsg() << endl);
- return buff;
- }
- vector<char> out;
- session->swap(out);
- buff->addBuffer(out);
- return buff;
- }
- TC_NetWorkBuffer::PACKET_TYPE ProxyProtocol::http2Response(TC_NetWorkBuffer &in, ResponsePacket& rsp)
- {
- TC_Http2Client* session = (TC_Http2Client*)((TC_Transceiver*)(in.getConnection()))->getSendBuffer().getContextData();
- pair<int, shared_ptr<TC_HttpResponse>> out;
- TC_NetWorkBuffer::PACKET_TYPE flag = session->parseResponse(in, out);
- if(flag == TC_NetWorkBuffer::PACKET_FULL)
- {
- rsp.iRequestId = out.first;
- rsp.sBuffer.resize(sizeof(shared_ptr<TC_HttpResponse>));
- //这里智能指针有一次+1, 后面要自己reset掉
- *(shared_ptr<TC_HttpResponse>*)rsp.sBuffer.data() = out.second;
- }
- return flag;
- }
- // ENCODE function, called by network thread
- shared_ptr<TC_NetWorkBuffer::Buffer> ProxyProtocol::grpcRequest(RequestPacket& request, TC_Transceiver *trans)
- {
- TC_GrpcClient* session = (TC_GrpcClient*)trans->getSendBuffer().getContextData();
- if(session == NULL)
- {
- session = new TC_GrpcClient();
- trans->getSendBuffer().setContextData(session, [=](TC_NetWorkBuffer*){ delete session; });
- session->settings(3000);
- }
- if (session->buffer().size() != 0) {
- //直接发送裸得应答数据,业务层一般不直接使用,仅仅tcp支持
- trans->getSendBuffer().addBuffer(session->buffer());
- trans->doRequest();
- // auto data = trans->getSendBuffer().getBufferPointer();
- // int iRet = trans->send(data.first, (uint32_t) data.second, 0);
- // trans->getSendBuffer().moveHeader(iRet);
- session->buffer().clear();
- }
- shared_ptr<TC_HttpRequest> *data = (shared_ptr<TC_HttpRequest>*)request.sBuffer.data();
- request.iRequestId = session->submit(*(*data).get());
- //这里把智能指针释放一次
- (*data).reset();
- shared_ptr<TC_NetWorkBuffer::Buffer> buff = std::make_shared<TC_NetWorkBuffer::Buffer>();
- if (request.iRequestId < 0)
- {
- TLOGERROR("http2Request::Fatal submit error: " << session->getErrMsg() << endl);
- return buff;
- }
- // cout << "http2Request id:" << request.iRequestId << endl;
- vector<char> out;
- session->swap(out);
- buff->addBuffer(out);
- return buff;
- }
- TC_NetWorkBuffer::PACKET_TYPE ProxyProtocol::grpcResponse(TC_NetWorkBuffer &in, ResponsePacket& rsp)
- {
- TC_GrpcClient* session = (TC_GrpcClient*)((TC_Transceiver*)(in.getConnection()))->getSendBuffer().getContextData();
- pair<int, shared_ptr<TC_HttpResponse>> out;
- TC_NetWorkBuffer::PACKET_TYPE flag = session->parseResponse(in, out);
- if(flag == TC_NetWorkBuffer::PACKET_FULL)
- {
- rsp.iRequestId = out.first;
- rsp.sBuffer.resize(sizeof(shared_ptr<TC_HttpResponse>));
- //这里智能指针有一次+1, 后面要自己reset掉
- *(shared_ptr<TC_HttpResponse>*)rsp.sBuffer.data() = out.second;
- }
- return flag;
- }
- #endif
- }
|