1/*
2 * Copyright (C) 2010 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#ifndef A_RTSP_CONNECTION_H_
18
19#define A_RTSP_CONNECTION_H_
20
21#include <media/stagefright/foundation/AHandler.h>
22#include <media/stagefright/foundation/AString.h>
23
24namespace android {
25
26struct ABuffer;
27
28struct ARTSPResponse : public RefBase {
29    unsigned long mStatusCode;
30    AString mStatusLine;
31    KeyedVector<AString,AString> mHeaders;
32    sp<ABuffer> mContent;
33};
34
35struct ARTSPConnection : public AHandler {
36    ARTSPConnection(bool uidValid = false, uid_t uid = 0);
37
38    void connect(const char *url, const sp<AMessage> &reply);
39    void disconnect(const sp<AMessage> &reply);
40
41    void sendRequest(const char *request, const sp<AMessage> &reply);
42
43    void observeBinaryData(const sp<AMessage> &reply);
44
45    static bool ParseURL(
46            const char *url, AString *host, unsigned *port, AString *path,
47            AString *user, AString *pass);
48
49protected:
50    virtual ~ARTSPConnection();
51    virtual void onMessageReceived(const sp<AMessage> &msg);
52
53private:
54    enum State {
55        DISCONNECTED,
56        CONNECTING,
57        CONNECTED,
58    };
59
60    enum {
61        kWhatConnect            = 'conn',
62        kWhatDisconnect         = 'disc',
63        kWhatCompleteConnection = 'comc',
64        kWhatSendRequest        = 'sreq',
65        kWhatReceiveResponse    = 'rres',
66        kWhatObserveBinaryData  = 'obin',
67    };
68
69    enum AuthType {
70        NONE,
71        BASIC,
72        DIGEST
73    };
74
75    static const int64_t kSelectTimeoutUs;
76
77    static const AString sUserAgent;
78
79    bool mUIDValid;
80    uid_t mUID;
81    State mState;
82    AString mUser, mPass;
83    AuthType mAuthType;
84    AString mNonce;
85    int mSocket;
86    int32_t mConnectionID;
87    int32_t mNextCSeq;
88    bool mReceiveResponseEventPending;
89
90    KeyedVector<int32_t, sp<AMessage> > mPendingRequests;
91
92    sp<AMessage> mObserveBinaryMessage;
93
94    void performDisconnect();
95
96    void onConnect(const sp<AMessage> &msg);
97    void onDisconnect(const sp<AMessage> &msg);
98    void onCompleteConnection(const sp<AMessage> &msg);
99    void onSendRequest(const sp<AMessage> &msg);
100    void onReceiveResponse();
101
102    void flushPendingRequests();
103    void postReceiveReponseEvent();
104
105    // Return false iff something went unrecoverably wrong.
106    bool receiveRTSPReponse();
107    status_t receive(void *data, size_t size);
108    bool receiveLine(AString *line);
109    sp<ABuffer> receiveBinaryData();
110    bool notifyResponseListener(const sp<ARTSPResponse> &response);
111
112    bool parseAuthMethod(const sp<ARTSPResponse> &response);
113    void addAuthentication(AString *request);
114
115    void addUserAgent(AString *request) const;
116
117    status_t findPendingRequest(
118            const sp<ARTSPResponse> &response, ssize_t *index) const;
119
120    bool handleServerRequest(const sp<ARTSPResponse> &request);
121
122    static bool ParseSingleUnsignedLong(
123            const char *from, unsigned long *x);
124
125    DISALLOW_EVIL_CONSTRUCTORS(ARTSPConnection);
126};
127
128}  // namespace android
129
130#endif  // A_RTSP_CONNECTION_H_
131