---

yaml
---
r: 22214
b: refs/heads/snap-stage3
c: 9fe40e8
h: refs/heads/master
v: v3

Author: pcwalton

[refs]

@@ -1,7 +1,7 @@
 ---
 refs/heads/master: e430a699f2c60890d9b86069fd0c68a70ece7120
 refs/heads/snap-stage1: e33de59e47c5076a89eadeb38f4934f58a3618a6
-refs/heads/snap-stage3: a4a7df0cccc36ed00b46462050ec522eadc2a0fc
+refs/heads/snap-stage3: 9fe40e8a7b86a530371f086effd1a427064b8a6d
 refs/heads/try: ffbe0e0e00374358b789b0037bcb3a577cd218be
 refs/tags/release-0.1: 1f5c5126e96c79d22cb7862f75304136e204f105
 refs/heads/ndm: f3868061cd7988080c30d6d5bf352a5a5fe2460b

branches/snap-stage3/src/libcore/os.rs

@@ -738,7 +738,7 @@ pub fn set_exit_status(code: int) {
  * Returns a list of the command line arguments.
  */
 #[cfg(target_os = "macos")]
-pub fn args() -> ~[~str] {
+fn real_args() -> ~[~str] {
     unsafe {
         let (argc, argv) = (*_NSGetArgc() as uint, *_NSGetArgv());
         let mut args = ~[];
@@ -749,6 +749,38 @@ pub fn args() -> ~[~str] {
     }
 }
 
+#[cfg(target_os = "linux")]
+fn real_args() -> ~[~str] {
+    fail    // TODO
+}
+
+#[cfg(windows)]
+fn real_args() -> ~[~str] {
+    fail    // TODO
+}
+
+struct OverriddenArgs {
+    val: ~[~str]
+}
+
+fn overridden_arg_key(+v: @OverriddenArgs) {}
+
+pub fn args() -> ~[~str] {
+    unsafe {
+        match task::local_data::local_data_get(overridden_arg_key) {
+            None => real_args(),
+            Some(args) => copy args.val
+        }
+    }
+}
+
+pub fn set_args(+new_args: ~[~str]) {
+    unsafe {
+        let overridden_args = @OverriddenArgs { val: copy new_args };
+        task::local_data::local_data_set(overridden_arg_key, overridden_args);
+    }
+}
+
 #[cfg(target_os = "macos")]
 extern {
     // These functions are in crt_externs.h.