Если это зарыто где-то в дебрях замыканий(т.е. вышеприведённый код исполненный
после подключённого чужого не работает), и вывод document.write вообще не нужен то можно сделать так:
<script>
var loc;
document.write = function(old){
function write(arg) {
loc = arg;
}
write.old = old;
return write
}(document.write);
</script>
<script src="/xxx.js"></script>
<script>
document.write = document.write.old;
</script>