1 /*
2  * Copyright (C) 2021 Huawei Device Co., Ltd.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at
6  *
7  *     http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15 
16 #ifndef RECORDER_SERVICE_STUB_H
17 #define RECORDER_SERVICE_STUB_H
18 
19 #include <map>
20 #include <set>
21 #include "i_standard_recorder_service.h"
22 #include "i_standard_recorder_listener.h"
23 #include "media_death_recipient.h"
24 #include "recorder_server.h"
25 #include "nocopyable.h"
26 #include "monitor_server_object.h"
27 
28 namespace OHOS {
29 namespace Media {
30 using RecorderStubFunc = std::function<int32_t(MessageParcel &, MessageParcel &)>;
31 class RecorderServiceStub : public IRemoteStub<IStandardRecorderService>,
32     public MonitorServerObject, public NoCopyable {
33 public:
34     static sptr<RecorderServiceStub> Create();
35     virtual ~RecorderServiceStub();
36 
37     int OnRemoteRequest(uint32_t code, MessageParcel &data, MessageParcel &reply, MessageOption &option) override;
38     int32_t SetListenerObject(const sptr<IRemoteObject> &object) override;
39     int32_t SetVideoSource(VideoSourceType source, int32_t &sourceId) override;
40     int32_t SetVideoEncoder(int32_t sourceId, VideoCodecFormat encoder) override;
41     int32_t SetVideoSize(int32_t sourceId, int32_t width, int32_t height) override;
42     int32_t SetVideoFrameRate(int32_t sourceId, int32_t frameRate) override;
43     int32_t SetVideoEncodingBitRate(int32_t sourceId, int32_t rate) override;
44     int32_t SetVideoIsHdr(int32_t sourceId, bool isHdr) override;
45     int32_t SetVideoEnableTemporalScale(int32_t sourceId, bool enableTemporalScale) override;
46     int32_t SetMetaConfigs(int32_t sourceId) override;
47     int32_t SetMetaSource(MetaSourceType source, int32_t &sourceId) override;
48     int32_t SetMetaMimeType(int32_t sourceId, const std::string_view &type) override;
49     int32_t SetMetaTimedKey(int32_t sourceId, const std::string_view &timedKey) override;
50     int32_t SetMetaSourceTrackMime(int32_t sourceId, const std::string_view &srcTrackMime) override;
51     sptr<OHOS::Surface> GetSurface(int32_t sourceId) override;
52     sptr<OHOS::Surface> GetMetaSurface(int32_t sourceId) override;
53     int32_t SetAudioSource(AudioSourceType source, int32_t &sourceId) override;
54     int32_t SetAudioEncoder(int32_t sourceId, AudioCodecFormat encoder) override;
55     int32_t SetAudioSampleRate(int32_t sourceId, int32_t rate) override;
56     int32_t SetAudioChannels(int32_t sourceId, int32_t num) override;
57     int32_t SetAudioEncodingBitRate(int32_t sourceId, int32_t bitRate) override;
58     int32_t SetDataSource(DataSourceType dataType, int32_t &sourceId) override;
59     int32_t SetUserCustomInfo(Meta &userCustomInfo) override;
60     int32_t SetGenre(std::string &genre) override;
61     int32_t SetMaxDuration(int32_t duration) override;
62     int32_t SetOutputFormat(OutputFormatType format) override;
63     int32_t SetOutputFile(int32_t fd) override;
64     int32_t SetFileGenerationMode(FileGenerationMode mode) override;
65     int32_t SetLocation(float latitude, float longitude) override;
66     int32_t SetOrientationHint(int32_t rotation) override;
67     int32_t Prepare() override;
68     int32_t Start() override;
69     int32_t Pause() override;
70     int32_t Resume() override;
71     int32_t Stop(bool block) override;
72     int32_t Reset() override;
73     int32_t Release() override;
74     int32_t DestroyStub() override;
75     int32_t DumpInfo(int32_t fd);
76     int32_t GetAVRecorderConfig(ConfigMap &configMap) override;
77     int32_t GetLocation(Location &location) override;
78     int32_t GetCurrentCapturerChangeInfo(AudioRecorderChangeInfo &changeInfo) override;
79     int32_t GetAvailableEncoder(std::vector<EncoderCapabilityData> &encoderInfo) override;
80     int32_t GetMaxAmplitude() override;
81     int32_t IsWatermarkSupported(bool &isWatermarkSupported) override;
82     int32_t SetWatermark(std::shared_ptr<AVBuffer> &waterMarkBuffer) override;
83     // MonitorServerObject override
84     int32_t DoIpcAbnormality() override;
85     int32_t DoIpcRecovery(bool fromMonitor) override;
86 
87 private:
88     RecorderServiceStub();
89     int32_t Init();
90     int32_t SetListenerObject(MessageParcel &data, MessageParcel &reply);
91     int32_t SetVideoSource(MessageParcel &data, MessageParcel &reply);
92     int32_t SetVideoEncoder(MessageParcel &data, MessageParcel &reply);
93     int32_t SetVideoSize(MessageParcel &data, MessageParcel &reply);
94     int32_t SetVideoFrameRate(MessageParcel &data, MessageParcel &reply);
95     int32_t SetVideoEncodingBitRate(MessageParcel &data, MessageParcel &reply);
96     int32_t SetVideoIsHdr(MessageParcel &data, MessageParcel &reply);
97     int32_t SetVideoEnableTemporalScale(MessageParcel &data, MessageParcel &reply);
98     int32_t SetMetaConfigs(MessageParcel &data, MessageParcel &reply);
99     int32_t SetMetaSource(MessageParcel &data, MessageParcel &reply);
100     int32_t SetMetaMimeType(MessageParcel &data, MessageParcel &reply);
101     int32_t SetMetaTimedKey(MessageParcel &data, MessageParcel &reply);
102     int32_t SetMetaSourceTrackMime(MessageParcel &data, MessageParcel &reply);
103     int32_t GetSurface(MessageParcel &data, MessageParcel &reply);
104     int32_t GetMetaSurface(MessageParcel &data, MessageParcel &reply);
105     int32_t SetAudioSource(MessageParcel &data, MessageParcel &reply);
106     int32_t SetAudioEncoder(MessageParcel &data, MessageParcel &reply);
107     int32_t SetAudioSampleRate(MessageParcel &data, MessageParcel &reply);
108     int32_t SetAudioChannels(MessageParcel &data, MessageParcel &reply);
109     int32_t SetAudioEncodingBitRate(MessageParcel &data, MessageParcel &reply);
110     int32_t SetDataSource(MessageParcel &data, MessageParcel &reply);
111     int32_t SetMaxDuration(MessageParcel &data, MessageParcel &reply);
112     int32_t SetOutputFormat(MessageParcel &data, MessageParcel &reply);
113     int32_t SetOutputFile(MessageParcel &data, MessageParcel &reply);
114     int32_t SetFileGenerationMode(MessageParcel &data, MessageParcel &reply);
115     int32_t SetLocation(MessageParcel &data, MessageParcel &reply);
116     int32_t SetOrientationHint(MessageParcel &data, MessageParcel &reply);
117     int32_t SetUserCustomInfo(MessageParcel &data, MessageParcel &reply);
118     int32_t SetGenre(MessageParcel &data, MessageParcel &reply);
119     int32_t Prepare(MessageParcel &data, MessageParcel &reply);
120     int32_t Start(MessageParcel &data, MessageParcel &reply);
121     int32_t Pause(MessageParcel &data, MessageParcel &reply);
122     int32_t Resume(MessageParcel &data, MessageParcel &reply);
123     int32_t Stop(MessageParcel &data, MessageParcel &reply);
124     int32_t Reset(MessageParcel &data, MessageParcel &reply);
125     int32_t Release(MessageParcel &data, MessageParcel &reply);
126     int32_t DestroyStub(MessageParcel &data, MessageParcel &reply);
127     int32_t GetAVRecorderConfig(MessageParcel &data, MessageParcel &reply);
128     int32_t GetLocation(MessageParcel &data, MessageParcel &reply);
129     int32_t GetCurrentCapturerChangeInfo(MessageParcel &data, MessageParcel &reply);
130     int32_t GetAvailableEncoder(MessageParcel &data, MessageParcel &reply);
131     int32_t GetMaxAmplitude(MessageParcel &data, MessageParcel &reply);
132     int32_t IsWatermarkSupported(MessageParcel &data, MessageParcel &reply);
133     int32_t SetWatermark(MessageParcel &data, MessageParcel &reply);
134     int32_t CheckPermission();
135     void FillRecFuncPart1();
136     void FillRecFuncPart2();
137 
138     std::shared_ptr<IRecorderService> recorderServer_ = nullptr;
139     std::map<uint32_t, RecorderStubFunc> recFuncs_;
140     std::mutex mutex_;
141     std::mutex stmutex_;
142     int32_t pid_ = 0;
143     AudioSourceType audioSourceType_ = AUDIO_SOURCE_INVALID;
144     bool needAudioPermissionCheck = false;
145     const std::set<uint32_t> AUDIO_REQUEST = {SET_AUDIO_SOURCE, SET_AUDIO_ENCODER, SET_AUDIO_ENCODER,
146         SET_AUDIO_CHANNELS, SET_AUDIO_ENCODING_BIT_RATE};
147     const std::set<uint32_t> COMMON_REQUEST = {SET_LISTENER_OBJ, SET_DATA_SOURCE, SET_MAX_DURATION,
148         SET_OUTPUT_FORMAT, SET_OUTPUT_FILE, SET_LOCATION, SET_ORIENTATION_HINT,
149         PREPARE, START, PAUSE, RESUME, STOP, RESET, RELEASE, DESTROY};
150 };
151 } // namespace Media
152 } // namespace OHOS
153 #endif // RECORDER_SERVICE_STUB_H
154