SessionDescription.java

  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            final String name = entry.getKey();
141            RtpContentMap.DescriptionTransport descriptionTransport = entry.getValue();
142            RtpDescription description = descriptionTransport.description;
143            IceUdpTransportInfo transport = descriptionTransport.transport;
144            final ArrayListMultimap<String, String> mediaAttributes = ArrayListMultimap.create();
145            final String ufrag = transport.getAttribute("ufrag");
146            final String pwd = transport.getAttribute("pwd");
147            if (!Strings.isNullOrEmpty(ufrag)) {
148                mediaAttributes.put("ice-ufrag", ufrag);
149            }
150            if (!Strings.isNullOrEmpty(pwd)) {
151                mediaAttributes.put("ice-pwd", pwd);
152            }
153            mediaAttributes.put("ice-options", HARDCODED_ICE_OPTIONS);
154            final IceUdpTransportInfo.Fingerprint fingerprint = transport.getFingerprint();
155            if (fingerprint != null) {
156                mediaAttributes.put("fingerprint", fingerprint.getHash() + " " + fingerprint.getContent());
157                mediaAttributes.put("setup", fingerprint.getSetup());
158            }
159            final ImmutableList.Builder<Integer> formatBuilder = new ImmutableList.Builder<>();
160            for (RtpDescription.PayloadType payloadType : description.getPayloadTypes()) {
161                final String id = payloadType.getId();
162                if (Strings.isNullOrEmpty(id)) {
163                    throw new IllegalArgumentException("Payload type is missing id");
164                }
165                if (!isInt(id)) {
166                    throw new IllegalArgumentException("Payload id is not numeric");
167                }
168                formatBuilder.add(payloadType.getIntId());
169                mediaAttributes.put("rtpmap", payloadType.toSdpAttribute());
170                List<RtpDescription.Parameter> parameters = payloadType.getParameters();
171                if (parameters.size() > 0) {
172                    mediaAttributes.put("fmtp", RtpDescription.Parameter.toSdpString(id, parameters));
173                }
174                for (RtpDescription.FeedbackNegotiation feedbackNegotiation : payloadType.getFeedbackNegotiations()) {
175                    final String type = feedbackNegotiation.getType();
176                    final String subtype = feedbackNegotiation.getSubType();
177                    if (Strings.isNullOrEmpty(type)) {
178                        throw new IllegalArgumentException("a feedback for payload-type " + id + " negotiation is missing type");
179                    }
180                    mediaAttributes.put("rtcp-fb", id + " " + type + (Strings.isNullOrEmpty(subtype) ? "" : " " + subtype));
181                }
182                for (RtpDescription.FeedbackNegotiationTrrInt feedbackNegotiationTrrInt : payloadType.feedbackNegotiationTrrInts()) {
183                    mediaAttributes.put("rtcp-fb", id + " trr-int " + feedbackNegotiationTrrInt.getValue());
184                }
185            }
186
187            for (RtpDescription.FeedbackNegotiation feedbackNegotiation : description.getFeedbackNegotiations()) {
188                final String type = feedbackNegotiation.getType();
189                final String subtype = feedbackNegotiation.getSubType();
190                if (Strings.isNullOrEmpty(type)) {
191                    throw new IllegalArgumentException("a feedback negotiation is missing type");
192                }
193                mediaAttributes.put("rtcp-fb", "* " + type + (Strings.isNullOrEmpty(subtype) ? "" : " " + subtype));
194            }
195            for (RtpDescription.FeedbackNegotiationTrrInt feedbackNegotiationTrrInt : description.feedbackNegotiationTrrInts()) {
196                mediaAttributes.put("rtcp-fb", "* trr-int " + feedbackNegotiationTrrInt.getValue());
197            }
198            for (RtpDescription.RtpHeaderExtension extension : description.getHeaderExtensions()) {
199                final String id = extension.getId();
200                final String uri = extension.getUri();
201                if (Strings.isNullOrEmpty(id)) {
202                    throw new IllegalArgumentException("A header extension is missing id");
203                }
204                if (Strings.isNullOrEmpty(uri)) {
205                    throw new IllegalArgumentException("A header extension is missing uri");
206                }
207                mediaAttributes.put("extmap", id + " " + uri);
208            }
209            for (RtpDescription.SourceGroup sourceGroup : description.getSourceGroups()) {
210                final String semantics = sourceGroup.getSemantics();
211                final List<String> groups = sourceGroup.getSsrcs();
212                if (Strings.isNullOrEmpty(semantics)) {
213                    throw new IllegalArgumentException("A SSRC group is missing semantics attribute");
214                }
215                if (groups.size() == 0) {
216                    throw new IllegalArgumentException("A SSRC group is missing SSRC ids");
217                }
218                mediaAttributes.put("ssrc-group", String.format("%s %s", semantics, Joiner.on(' ').join(groups)));
219            }
220            for (RtpDescription.Source source : description.getSources()) {
221                for (RtpDescription.Source.Parameter parameter : source.getParameters()) {
222                    final String id = source.getSsrcId();
223                    final String parameterName = parameter.getParameterName();
224                    final String parameterValue = parameter.getParameterValue();
225                    if (Strings.isNullOrEmpty(id)) {
226                        throw new IllegalArgumentException("A source specific media attribute is missing the id");
227                    }
228                    if (Strings.isNullOrEmpty(parameterName)) {
229                        throw new IllegalArgumentException("A source specific media attribute is missing its name");
230                    }
231                    if (Strings.isNullOrEmpty(parameterValue)) {
232                        throw new IllegalArgumentException("A source specific media attribute is missing its value");
233                    }
234                    mediaAttributes.put("ssrc", id + " " + parameter.getParameterName() + ":" + parameter.getParameterValue());
235                }
236            }
237
238            mediaAttributes.put("mid", name);
239
240            //random additional attributes
241            mediaAttributes.put("rtcp", "9 IN IP4 0.0.0.0");
242            mediaAttributes.put("sendrecv", "");
243            mediaAttributes.put("rtcp-mux", "");
244
245            final MediaBuilder mediaBuilder = new MediaBuilder();
246            mediaBuilder.setMedia(description.getMedia().toString().toLowerCase(Locale.ROOT));
247            mediaBuilder.setConnectionData(HARDCODED_CONNECTION);
248            mediaBuilder.setPort(HARDCODED_MEDIA_PORT);
249            mediaBuilder.setProtocol(HARDCODED_MEDIA_PROTOCOL);
250            mediaBuilder.setAttributes(mediaAttributes);
251            mediaBuilder.setFormats(formatBuilder.build());
252            mediaListBuilder.add(mediaBuilder.createMedia());
253
254        }
255        sessionDescriptionBuilder.setVersion(0);
256        sessionDescriptionBuilder.setName("-");
257        sessionDescriptionBuilder.setMedia(mediaListBuilder.build());
258        sessionDescriptionBuilder.setAttributes(attributeMap);
259
260        return sessionDescriptionBuilder.createSessionDescription();
261    }
262
263    public static int ignorantIntParser(final String input) {
264        try {
265            return Integer.parseInt(input);
266        } catch (NumberFormatException e) {
267            return 0;
268        }
269    }
270
271    public static boolean isInt(final String input) {
272        if (input == null) {
273            return false;
274        }
275        try {
276            Integer.parseInt(input);
277            return true;
278        } catch (NumberFormatException e) {
279            return false;
280        }
281    }
282
283    public static Pair<String, String> parseAttribute(final String input) {
284        final String[] pair = input.split(":", 2);
285        if (pair.length == 2) {
286            return new Pair<>(pair[0], pair[1]);
287        } else {
288            return new Pair<>(pair[0], "");
289        }
290    }
291
292    @Override
293    public String toString() {
294        final StringBuilder s = new StringBuilder()
295                .append("v=").append(version).append(LINE_DIVIDER)
296                //TODO randomize or static
297                .append("o=- 8770656990916039506 2 IN IP4 127.0.0.1").append(LINE_DIVIDER) //what ever that means
298                .append("s=").append(name).append(LINE_DIVIDER)
299                .append("t=0 0").append(LINE_DIVIDER);
300        appendAttributes(s, attributes);
301        for (Media media : this.media) {
302            s.append("m=").append(media.media).append(' ').append(media.port).append(' ').append(media.protocol).append(' ').append(Joiner.on(' ').join(media.formats)).append(LINE_DIVIDER);
303            s.append("c=").append(media.connectionData).append(LINE_DIVIDER);
304            appendAttributes(s, media.attributes);
305        }
306        return s.toString();
307    }
308
309    public static class Media {
310        public final String media;
311        public final int port;
312        public final String protocol;
313        public final List<Integer> formats;
314        public final String connectionData;
315        public final ArrayListMultimap<String, String> attributes;
316
317        public Media(String media, int port, String protocol, List<Integer> formats, String connectionData, ArrayListMultimap<String, String> attributes) {
318            this.media = media;
319            this.port = port;
320            this.protocol = protocol;
321            this.formats = formats;
322            this.connectionData = connectionData;
323            this.attributes = attributes;
324        }
325    }
326
327}