1package eu.siacs.conversations.xmpp.jingle;
2
3import android.util.Log;
4import android.util.Pair;
5
6import com.google.common.base.Joiner;
7import com.google.common.base.Strings;
8import com.google.common.collect.ArrayListMultimap;
9import com.google.common.collect.ImmutableList;
10
11import java.util.List;
12import java.util.Locale;
13import java.util.Map;
14
15import eu.siacs.conversations.Config;
16import eu.siacs.conversations.xmpp.jingle.stanzas.Group;
17import eu.siacs.conversations.xmpp.jingle.stanzas.IceUdpTransportInfo;
18import eu.siacs.conversations.xmpp.jingle.stanzas.RtpDescription;
19
20public class SessionDescription {
21
22 private final static String LINE_DIVIDER = "\r\n";
23 private final static String HARDCODED_MEDIA_PROTOCOL = "UDP/TLS/RTP/SAVPF"; //probably only true for DTLS-SRTP aka when we have a fingerprint
24 private final static int HARDCODED_MEDIA_PORT = 9;
25 private final static String HARDCODED_ICE_OPTIONS = "trickle renomination";
26 private final static String HARDCODED_CONNECTION = "IN IP4 0.0.0.0";
27
28 public final int version;
29 public final String name;
30 public final String connectionData;
31 public final ArrayListMultimap<String, String> attributes;
32 public final List<Media> media;
33
34
35 public SessionDescription(int version, String name, String connectionData, ArrayListMultimap<String, String> attributes, List<Media> media) {
36 this.version = version;
37 this.name = name;
38 this.connectionData = connectionData;
39 this.attributes = attributes;
40 this.media = media;
41 }
42
43 private static void appendAttributes(StringBuilder s, ArrayListMultimap<String, String> attributes) {
44 for (Map.Entry<String, String> attribute : attributes.entries()) {
45 final String key = attribute.getKey();
46 final String value = attribute.getValue();
47 s.append("a=").append(key);
48 if (!Strings.isNullOrEmpty(value)) {
49 s.append(':').append(value);
50 }
51 s.append(LINE_DIVIDER);
52 }
53 }
54
55 public static SessionDescription parse(final Map<String, RtpContentMap.DescriptionTransport> contents) {
56 final SessionDescriptionBuilder sessionDescriptionBuilder = new SessionDescriptionBuilder();
57 return sessionDescriptionBuilder.createSessionDescription();
58 }
59
60 public static SessionDescription parse(final String input) {
61 final SessionDescriptionBuilder sessionDescriptionBuilder = new SessionDescriptionBuilder();
62 MediaBuilder currentMediaBuilder = null;
63 ArrayListMultimap<String, String> attributeMap = ArrayListMultimap.create();
64 ImmutableList.Builder<Media> mediaBuilder = new ImmutableList.Builder<>();
65 for (final String line : input.split(LINE_DIVIDER)) {
66 final String[] pair = line.trim().split("=", 2);
67 if (pair.length < 2 || pair[0].length() != 1) {
68 Log.d(Config.LOGTAG, "skipping sdp parsing on line " + line);
69 continue;
70 }
71 final char key = pair[0].charAt(0);
72 final String value = pair[1];
73 switch (key) {
74 case 'v':
75 sessionDescriptionBuilder.setVersion(ignorantIntParser(value));
76 break;
77 case 'c':
78 if (currentMediaBuilder != null) {
79 currentMediaBuilder.setConnectionData(value);
80 } else {
81 sessionDescriptionBuilder.setConnectionData(value);
82 }
83 break;
84 case 's':
85 sessionDescriptionBuilder.setName(value);
86 break;
87 case 'a':
88 final Pair<String, String> attribute = parseAttribute(value);
89 attributeMap.put(attribute.first, attribute.second);
90 break;
91 case 'm':
92 if (currentMediaBuilder == null) {
93 sessionDescriptionBuilder.setAttributes(attributeMap);
94 ;
95 } else {
96 currentMediaBuilder.setAttributes(attributeMap);
97 mediaBuilder.add(currentMediaBuilder.createMedia());
98 }
99 attributeMap = ArrayListMultimap.create();
100 currentMediaBuilder = new MediaBuilder();
101 final String[] parts = value.split(" ");
102 if (parts.length >= 3) {
103 currentMediaBuilder.setMedia(parts[0]);
104 currentMediaBuilder.setPort(ignorantIntParser(parts[1]));
105 currentMediaBuilder.setProtocol(parts[2]);
106 ImmutableList.Builder<Integer> formats = new ImmutableList.Builder<>();
107 for (int i = 3; i < parts.length; ++i) {
108 formats.add(ignorantIntParser(parts[i]));
109 }
110 currentMediaBuilder.setFormats(formats.build());
111 } else {
112 Log.d(Config.LOGTAG, "skipping media line " + line);
113 }
114 break;
115 }
116
117 }
118 if (currentMediaBuilder != null) {
119 currentMediaBuilder.setAttributes(attributeMap);
120 mediaBuilder.add(currentMediaBuilder.createMedia());
121 } else {
122 sessionDescriptionBuilder.setAttributes(attributeMap);
123 }
124 sessionDescriptionBuilder.setMedia(mediaBuilder.build());
125 return sessionDescriptionBuilder.createSessionDescription();
126 }
127
128 public static SessionDescription of(final RtpContentMap contentMap) {
129 final SessionDescriptionBuilder sessionDescriptionBuilder = new SessionDescriptionBuilder();
130 final ArrayListMultimap<String, String> attributeMap = ArrayListMultimap.create();
131 final ImmutableList.Builder<Media> mediaListBuilder = new ImmutableList.Builder<>();
132 final Group group = contentMap.group;
133 if (group != null) {
134 attributeMap.put("group", group.getSemantics() + " " + Joiner.on(' ').join(group.getIdentificationTags()));
135 }
136
137 attributeMap.put("msid-semantic", " WMS my-media-stream");
138
139 for (Map.Entry<String, RtpContentMap.DescriptionTransport> entry : contentMap.contents.entrySet()) {
140
141 //TODO sprinkle in a few noWhiteSpaces checks into various parameters and types
142
143 final String name = entry.getKey();
144 RtpContentMap.DescriptionTransport descriptionTransport = entry.getValue();
145 RtpDescription description = descriptionTransport.description;
146 IceUdpTransportInfo transport = descriptionTransport.transport;
147 final ArrayListMultimap<String, String> mediaAttributes = ArrayListMultimap.create();
148 final String ufrag = transport.getAttribute("ufrag");
149 final String pwd = transport.getAttribute("pwd");
150 if (!Strings.isNullOrEmpty(ufrag)) {
151 mediaAttributes.put("ice-ufrag", ufrag);
152 }
153 if (!Strings.isNullOrEmpty(pwd)) {
154 mediaAttributes.put("ice-pwd", pwd);
155 }
156 mediaAttributes.put("ice-options", HARDCODED_ICE_OPTIONS);
157 final IceUdpTransportInfo.Fingerprint fingerprint = transport.getFingerprint();
158 if (fingerprint != null) {
159 mediaAttributes.put("fingerprint", fingerprint.getHash() + " " + fingerprint.getContent());
160 mediaAttributes.put("setup", fingerprint.getSetup());
161 }
162 final ImmutableList.Builder<Integer> formatBuilder = new ImmutableList.Builder<>();
163 for (RtpDescription.PayloadType payloadType : description.getPayloadTypes()) {
164 final String id = payloadType.getId();
165 if (Strings.isNullOrEmpty(id)) {
166 throw new IllegalArgumentException("Payload type is missing id");
167 }
168 if (!isInt(id)) {
169 throw new IllegalArgumentException("Payload id is not numeric");
170 }
171 formatBuilder.add(payloadType.getIntId());
172 mediaAttributes.put("rtpmap", payloadType.toSdpAttribute());
173 List<RtpDescription.Parameter> parameters = payloadType.getParameters();
174 if (parameters.size() > 0) {
175 mediaAttributes.put("fmtp", RtpDescription.Parameter.toSdpString(id, parameters));
176 }
177 for (RtpDescription.FeedbackNegotiation feedbackNegotiation : payloadType.getFeedbackNegotiations()) {
178 final String type = feedbackNegotiation.getType();
179 final String subtype = feedbackNegotiation.getSubType();
180 if (Strings.isNullOrEmpty(type)) {
181 throw new IllegalArgumentException("a feedback for payload-type " + id + " negotiation is missing type");
182 }
183 mediaAttributes.put("rtcp-fb", id + " " + type + (Strings.isNullOrEmpty(subtype) ? "" : " " + subtype));
184 }
185 for (RtpDescription.FeedbackNegotiationTrrInt feedbackNegotiationTrrInt : payloadType.feedbackNegotiationTrrInts()) {
186 mediaAttributes.put("rtcp-fb", id + " trr-int " + feedbackNegotiationTrrInt.getValue());
187 }
188 }
189
190 for (RtpDescription.FeedbackNegotiation feedbackNegotiation : description.getFeedbackNegotiations()) {
191 final String type = feedbackNegotiation.getType();
192 final String subtype = feedbackNegotiation.getSubType();
193 if (Strings.isNullOrEmpty(type)) {
194 throw new IllegalArgumentException("a feedback negotiation is missing type");
195 }
196 mediaAttributes.put("rtcp-fb", "* " + type + (Strings.isNullOrEmpty(subtype) ? "" : " " + subtype));
197 }
198 for (RtpDescription.FeedbackNegotiationTrrInt feedbackNegotiationTrrInt : description.feedbackNegotiationTrrInts()) {
199 mediaAttributes.put("rtcp-fb", "* trr-int " + feedbackNegotiationTrrInt.getValue());
200 }
201 for (RtpDescription.RtpHeaderExtension extension : description.getHeaderExtensions()) {
202 final String id = extension.getId();
203 final String uri = extension.getUri();
204 if (Strings.isNullOrEmpty(id)) {
205 throw new IllegalArgumentException("A header extension is missing id");
206 }
207 if (Strings.isNullOrEmpty(uri)) {
208 throw new IllegalArgumentException("A header extension is missing uri");
209 }
210 mediaAttributes.put("extmap", id + " " + uri);
211 }
212 for (RtpDescription.SourceGroup sourceGroup : description.getSourceGroups()) {
213 final String semantics = sourceGroup.getSemantics();
214 final List<String> groups = sourceGroup.getSsrcs();
215 if (Strings.isNullOrEmpty(semantics)) {
216 throw new IllegalArgumentException("A SSRC group is missing semantics attribute");
217 }
218 if (groups.size() == 0) {
219 throw new IllegalArgumentException("A SSRC group is missing SSRC ids");
220 }
221 mediaAttributes.put("ssrc-group", String.format("%s %s", semantics, Joiner.on(' ').join(groups)));
222 }
223 for (RtpDescription.Source source : description.getSources()) {
224 for (RtpDescription.Source.Parameter parameter : source.getParameters()) {
225 final String id = source.getSsrcId();
226 final String parameterName = parameter.getParameterName();
227 final String parameterValue = parameter.getParameterValue();
228 if (Strings.isNullOrEmpty(id)) {
229 throw new IllegalArgumentException("A source specific media attribute is missing the id");
230 }
231 if (Strings.isNullOrEmpty(parameterName)) {
232 throw new IllegalArgumentException("A source specific media attribute is missing its name");
233 }
234 if (Strings.isNullOrEmpty(parameterValue)) {
235 throw new IllegalArgumentException("A source specific media attribute is missing its value");
236 }
237 mediaAttributes.put("ssrc", id + " " + parameter.getParameterName() + ":" + parameter.getParameterValue());
238 }
239 }
240
241 mediaAttributes.put("mid", name);
242
243 //random additional attributes
244 mediaAttributes.put("rtcp", "9 IN IP4 0.0.0.0");
245 mediaAttributes.put("sendrecv", "");
246 mediaAttributes.put("rtcp-mux", "");
247
248 final MediaBuilder mediaBuilder = new MediaBuilder();
249 mediaBuilder.setMedia(description.getMedia().toString().toLowerCase(Locale.ROOT));
250 mediaBuilder.setConnectionData(HARDCODED_CONNECTION);
251 mediaBuilder.setPort(HARDCODED_MEDIA_PORT);
252 mediaBuilder.setProtocol(HARDCODED_MEDIA_PROTOCOL);
253 mediaBuilder.setAttributes(mediaAttributes);
254 mediaBuilder.setFormats(formatBuilder.build());
255 mediaListBuilder.add(mediaBuilder.createMedia());
256
257 }
258 sessionDescriptionBuilder.setVersion(0);
259 sessionDescriptionBuilder.setName("-");
260 sessionDescriptionBuilder.setMedia(mediaListBuilder.build());
261 sessionDescriptionBuilder.setAttributes(attributeMap);
262
263 return sessionDescriptionBuilder.createSessionDescription();
264 }
265
266 public static int ignorantIntParser(final String input) {
267 try {
268 return Integer.parseInt(input);
269 } catch (NumberFormatException e) {
270 return 0;
271 }
272 }
273
274 public static boolean isInt(final String input) {
275 if (input == null) {
276 return false;
277 }
278 try {
279 Integer.parseInt(input);
280 return true;
281 } catch (NumberFormatException e) {
282 return false;
283 }
284 }
285
286 public static Pair<String, String> parseAttribute(final String input) {
287 final String[] pair = input.split(":", 2);
288 if (pair.length == 2) {
289 return new Pair<>(pair[0], pair[1]);
290 } else {
291 return new Pair<>(pair[0], "");
292 }
293 }
294
295 @Override
296 public String toString() {
297 final StringBuilder s = new StringBuilder()
298 .append("v=").append(version).append(LINE_DIVIDER)
299 //TODO randomize or static
300 .append("o=- 8770656990916039506 2 IN IP4 127.0.0.1").append(LINE_DIVIDER) //what ever that means
301 .append("s=").append(name).append(LINE_DIVIDER)
302 .append("t=0 0").append(LINE_DIVIDER);
303 appendAttributes(s, attributes);
304 for (Media media : this.media) {
305 s.append("m=").append(media.media).append(' ').append(media.port).append(' ').append(media.protocol).append(' ').append(Joiner.on(' ').join(media.formats)).append(LINE_DIVIDER);
306 s.append("c=").append(media.connectionData).append(LINE_DIVIDER);
307 appendAttributes(s, media.attributes);
308 }
309 return s.toString();
310 }
311
312 public static class Media {
313 public final String media;
314 public final int port;
315 public final String protocol;
316 public final List<Integer> formats;
317 public final String connectionData;
318 public final ArrayListMultimap<String, String> attributes;
319
320 public Media(String media, int port, String protocol, List<Integer> formats, String connectionData, ArrayListMultimap<String, String> attributes) {
321 this.media = media;
322 this.port = port;
323 this.protocol = protocol;
324 this.formats = formats;
325 this.connectionData = connectionData;
326 this.attributes = attributes;
327 }
328 }
329
330}