mirror of
https://github.com/rutgersc/m3u8-rs.git
synced 2024-11-15 21:21:08 +00:00
SessionData: Must have either VALUE or URI, but not both.
If SessionData has a value, don't write the URI and vice-versa. As per https://tools.ietf.org/html/rfc8216#section-4.3.4.4 EXT-X-SESSION-DATA must have one or the other, not both.
This commit is contained in:
parent
85b0826103
commit
870ca830d3
3 changed files with 49 additions and 22 deletions
|
@ -319,8 +319,9 @@ named!(pub alternative_media_tag<AlternativeMedia>,
|
||||||
);
|
);
|
||||||
|
|
||||||
named!(pub session_data_tag<SessionData>,
|
named!(pub session_data_tag<SessionData>,
|
||||||
do_parse!( tag!("#EXT-X-SESSION-DATA:") >> attributes: key_value_pairs >>
|
do_parse!( tag!("#EXT-X-SESSION-DATA:") >>
|
||||||
( SessionData::from_hashmap(attributes)))
|
session_data: map_res!(key_value_pairs, |attrs| SessionData::from_hashmap(attrs)) >>
|
||||||
|
( session_data))
|
||||||
);
|
);
|
||||||
|
|
||||||
named!(pub session_key_tag<SessionKey>,
|
named!(pub session_key_tag<SessionKey>,
|
||||||
|
|
|
@ -343,6 +343,9 @@ impl fmt::Display for AlternativeMediaType {
|
||||||
|
|
||||||
/// [`#EXT-X-SESSION-KEY:<attribute-list>`]
|
/// [`#EXT-X-SESSION-KEY:<attribute-list>`]
|
||||||
/// (https://tools.ietf.org/html/draft-pantos-http-live-streaming-19#section-4.3.4.5)
|
/// (https://tools.ietf.org/html/draft-pantos-http-live-streaming-19#section-4.3.4.5)
|
||||||
|
/// The EXT-X-SESSION-KEY tag allows encryption keys from Media Playlists
|
||||||
|
/// to be specified in a Master Playlist. This allows the client to
|
||||||
|
/// preload these keys without having to read the Media Playlist(s) first.
|
||||||
#[derive(Debug, Default, PartialEq, Clone)]
|
#[derive(Debug, Default, PartialEq, Clone)]
|
||||||
pub struct SessionKey(pub Key);
|
pub struct SessionKey(pub Key);
|
||||||
|
|
||||||
|
@ -354,34 +357,56 @@ impl SessionKey {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, PartialEq, Clone)]
|
||||||
|
pub enum SessionDataField {
|
||||||
|
Value(String),
|
||||||
|
Uri(String)
|
||||||
|
}
|
||||||
|
|
||||||
/// [`#EXT-X-SESSION-DATA:<attribute-list>`]
|
/// [`#EXT-X-SESSION-DATA:<attribute-list>`]
|
||||||
/// (https://tools.ietf.org/html/draft-pantos-http-live-streaming-19#section-4.3.4.4)
|
/// (https://tools.ietf.org/html/draft-pantos-http-live-streaming-19#section-4.3.4.4)
|
||||||
/// The EXT-X-SESSION-KEY tag allows encryption keys from Media Playlists
|
/// The EXT-X-SESSION-DATA tag allows arbitrary session data to be carried
|
||||||
/// to be specified in a Master Playlist. This allows the client to
|
/// in a Master Playlist.
|
||||||
/// preload these keys without having to read the Media Playlist(s) first.
|
#[derive(Debug, PartialEq, Clone)]
|
||||||
#[derive(Debug, Default, PartialEq, Clone)]
|
|
||||||
pub struct SessionData {
|
pub struct SessionData {
|
||||||
pub data_id: String,
|
pub data_id: String,
|
||||||
pub value: String,
|
pub field: SessionDataField,
|
||||||
pub uri: String,
|
|
||||||
pub language: Option<String>,
|
pub language: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl SessionData {
|
impl SessionData {
|
||||||
pub fn from_hashmap(mut attrs: HashMap<String, String>) -> SessionData {
|
pub fn from_hashmap(mut attrs: HashMap<String, String>) -> Result<SessionData, String> {
|
||||||
SessionData {
|
let data_id = match attrs.remove("DATA-ID") {
|
||||||
data_id: attrs.remove("DATA-ID").unwrap_or_else(String::new),
|
Some(data_id) => data_id,
|
||||||
value: attrs.remove("VALUE").unwrap_or_else(String::new),
|
None => return Err("EXT-X-SESSION-DATA field without DATA-ID".to_string())
|
||||||
uri: attrs.remove("URI").unwrap_or_else(String::new),
|
};
|
||||||
|
|
||||||
|
let value = attrs.remove("VALUE");
|
||||||
|
let uri = attrs.remove("URI");
|
||||||
|
|
||||||
|
// SessionData must contain either a VALUE or a URI,
|
||||||
|
// but not both https://tools.ietf.org/html/rfc8216#section-4.3.4.4
|
||||||
|
let field = match (value, uri) {
|
||||||
|
(Some(value), None) => SessionDataField::Value(value),
|
||||||
|
(None, Some(uri)) => SessionDataField::Uri(uri),
|
||||||
|
(Some(_), Some(_)) => return Err(format!["EXT-X-SESSION-DATA tag {} contains both a value and a uri", data_id]),
|
||||||
|
(None, None) => return Err(format!["EXT-X-SESSION-DATA tag {} must contain either a value or a uri", data_id]),
|
||||||
|
};
|
||||||
|
|
||||||
|
Ok(SessionData {
|
||||||
|
data_id,
|
||||||
|
field,
|
||||||
language: attrs.remove("LANGUAGE"),
|
language: attrs.remove("LANGUAGE"),
|
||||||
}
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn write_to<T: Write>(&self, w: &mut T) -> std::io::Result<()> {
|
pub fn write_to<T: Write>(&self, w: &mut T) -> std::io::Result<()> {
|
||||||
write!(w, "#EXT-X-SESSION-DATA:")?;
|
write!(w, "#EXT-X-SESSION-DATA:")?;
|
||||||
write!(w, "DATA-ID=\"{}\"", self.data_id)?;
|
write!(w, "DATA-ID=\"{}\"", self.data_id)?;
|
||||||
write!(w, ",VALUE=\"{}\"", self.value)?;
|
match &self.field {
|
||||||
write!(w, ",URI=\"{}\"", self.uri)?;
|
SessionDataField::Value(value) => write!(w, ",VALUE=\"{}\"", value)?,
|
||||||
|
SessionDataField::Uri(uri) => write!(w, ",URI=\"{}\"", uri)?,
|
||||||
|
};
|
||||||
write_some_attribute_quoted!(w, ",LANGUAGE", &self.language)?;
|
write_some_attribute_quoted!(w, ",LANGUAGE", &self.language)?;
|
||||||
write!(w, "\n")
|
write!(w, "\n")
|
||||||
}
|
}
|
||||||
|
|
13
tests/lib.rs
13
tests/lib.rs
|
@ -333,12 +333,13 @@ fn create_and_parse_master_playlist_full() {
|
||||||
closed_captions: Some("closed_captions".into()),
|
closed_captions: Some("closed_captions".into()),
|
||||||
}
|
}
|
||||||
],
|
],
|
||||||
session_data: Some(SessionData {
|
session_data: Some(
|
||||||
data_id: "****".into(),
|
SessionData {
|
||||||
value: "%%%%".into(),
|
data_id: "****".into(),
|
||||||
uri: "++++".into(),
|
field: SessionDataField::Value("%%%%".to_string()),
|
||||||
language: Some("SessionDataLanguage".into()),
|
language: Some("SessionDataLanguage".into()),
|
||||||
}),
|
}
|
||||||
|
),
|
||||||
session_key: Some(SessionKey(Key {
|
session_key: Some(SessionKey(Key {
|
||||||
method: "AES-128".into(),
|
method: "AES-128".into(),
|
||||||
uri: Some("https://secure.domain.com".into()),
|
uri: Some("https://secure.domain.com".into()),
|
||||||
|
|
Loading…
Reference in a new issue