Escape strings passed to gnuplot.

src/axes2d.rs

@@ -6,7 +6,7 @@ use crate::axes_common::*;
 use crate::coordinates::*;
 use crate::datatype::*;
 use crate::options::*;
-use crate::util::OneWayOwned;
+use crate::util::{escape, OneWayOwned};
 use crate::writer::Writer;
 
 struct LegendData
@@ -83,7 +83,7 @@ impl LegendData
 			Title(ref s) =>
 			{
 				w.write_str(" title \"");
-				w.write_str(s);
+				w.write_str(&escape(s));
 				w.write_str("\"");
 			}
 		}
@@ -92,7 +92,7 @@ impl LegendData
 			Font(ref f, s) =>
 			{
 				w.write_str(" font \"");
-				w.write_str(f);
+				w.write_str(&escape(f));
 				w.write_str(",");
 				w.write_str(&s.to_string()[..]);
 				w.write_str("\"");
@@ -102,7 +102,7 @@ impl LegendData
 			TextColor(ref s) =>
 			{
 				w.write_str(" textcolor rgb \"");
-				w.write_str(&s);
+				w.write_str(&escape(s));
 				w.write_str("\"");
 			}
 		}

src/axes_common.rs

@@ -11,7 +11,7 @@ use crate::coordinates::*;
 
 use crate::datatype::*;
 use crate::options::*;
-use crate::util::OneWayOwned;
+use crate::util::{escape, OneWayOwned};
 use crate::writer::*;
 use std::borrow::Borrow;
 
@@ -419,7 +419,7 @@ impl PlotElement
 		first_opt! {self.options,
 			Caption(ref s) =>
 			{
-				writer.write_str(&s);
+				writer.write_str(&escape(s));
 			}
 		}
 		writer.write_str("\"");
@@ -473,7 +473,7 @@ impl LabelData
 		self.label_type.write_label_str(w);
 
 		w.write_str(" \"");
-		w.write_str(&self.text);
+		w.write_str(&escape(&self.text));
 		w.write_str("\"");
 
 		write_out_label_options(self.label_type, &self.options[..], w);

src/figure.rs

@@ -9,6 +9,7 @@ use crate::axes2d::*;
 use crate::axes3d::*;
 
 use crate::options::{GnuplotVersion, MultiplotFillDirection, MultiplotFillOrder};
+use crate::util::escape;
 use crate::writer::Writer;
 use std::cell::RefCell;
 use std::fs::File;
@@ -508,7 +509,11 @@ impl Figure
 
 		if let Some(ref output_file) = self.output_file
 		{
-			writeln!(w, "set output \"{}\"", output_file.to_str().unwrap());
+			writeln!(
+				w,
+				"set output \"{}\"",
+				escape(&output_file.to_str().unwrap())
+			);
 		}
 
 		writeln!(w, "set termoption dashed");
@@ -551,7 +556,7 @@ impl Figure
 			let title = m
 				.title
 				.as_ref()
-				.map_or("".to_string(), |t| format!(" title \"{}\"", t));
+				.map_or("".to_string(), |t| format!(" title \"{}\"", escape(t)));
 			let scale = m.scale_x.map_or("".to_string(), |s| {
 				format!(" scale {},{}", s, m.scale_y.unwrap())
 			});
@@ -604,8 +609,10 @@ impl Figure
 	}
 }
 
-impl Drop for Figure {
-	fn drop(&mut self) {
+impl Drop for Figure
+{
+	fn drop(&mut self)
+	{
 		self.close();
 	}
 }

src/util.rs

@@ -62,3 +62,19 @@ impl<'l, T: OneWayOwned> OneWayOwned for &'l [T]
 		self.iter().map(|v| v.to_one_way_owned()).collect()
 	}
 }
+
+pub(crate) fn escape(s: &str) -> String
+{
+	let s = s.replace(r"\", r"\\");
+	let s = s.replace(r#"""#, r#"\""#);
+	s
+}
+
+#[test]
+fn escape_test()
+{
+	assert_eq!(r"\\", escape(r"\"));
+	assert_eq!(r"\\\\", escape(r"\\"));
+	assert_eq!(r#"\\\""#, escape(r#"\""#));
+	assert_eq!(r#"\"\""#, escape(r#""""#));
+}