def conn1(addr, encoding=Dtc.PROTOCOL_BUFFERS, heartbeat_interval=10): """ Primary function to connect and logon to DTC server""" sock = socket.create_connection(addr) # encoding request, construct and send enc_req = Dtc.EncodingRequest() enc_req.Encoding = encoding enc_req.ProtocolType = "DTC" enc_req.ProtocolVersion = Dtc.CURRENT_VERSION send_message(enc_req, Dtc.ENCODING_REQUEST, sock) # send the encoding request # receive encoding response m_type, m_resp = get_message(sock) logMsg.debug("%s, %s", m_type[0], m_resp) logStdout.debug("%s, %s", m_type[0], m_resp) # Logon request, construct and send logon_req = Dtc.LogonRequest() logon_req.ProtocolVersion = Dtc.CURRENT_VERSION # logon_req.Username = "******" # logon_req.Password = "******" logon_req.GeneralTextData = "John's Test" logon_req.HeartbeatIntervalInSeconds = heartbeat_interval logon_req.ClientName = "John_Tester" send_message(logon_req, Dtc.LOGON_REQUEST, sock) # receive logon response m_type, m_resp = get_message(sock) logMsg.debug("%s, %s", m_type[0], m_resp) logStdout.debug("%s, %s", m_type[0], m_resp) return sock
def create_enc_req(encoding): enc_req = Dtc.EncodingRequest() enc_req.Encoding = encoding enc_req.ProtocolType = "DTC" enc_req.ProtocolVersion = Dtc.CURRENT_VERSION return enc_req