1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
/*___INFO__MARK_BEGIN__*/
/*************************************************************************
*
* The Contents of this file are made available subject to the terms of
* the Sun Industry Standards Source License Version 1.2
*
* Sun Microsystems Inc., March, 2001
*
*
* Sun Industry Standards Source License Version 1.2
* =================================================
* The contents of this file are subject to the Sun Industry Standards
* Source License Version 1.2 (the "License"); You may not use this file
* except in compliance with the License. You may obtain a copy of the
* License at http://gridengine.sunsource.net/Gridengine_SISSL_license.html
*
* Software provided under this License is provided on an "AS IS" basis,
* WITHOUT WARRANTY OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING,
* WITHOUT LIMITATION, WARRANTIES THAT THE SOFTWARE IS FREE OF DEFECTS,
* MERCHANTABLE, FIT FOR A PARTICULAR PURPOSE, OR NON-INFRINGING.
* See the License for the specific provisions governing your rights and
* obligations concerning the Software.
*
* The Initial Developer of the Original Code is: Sun Microsystems, Inc.
*
* Copyright: 2001 by Sun Microsystems, Inc.
*
* All Rights Reserved.
*
************************************************************************/
/*___INFO__MARK_END__*/
#include "stdafx.h"
#include <windows.h>
#include <iostream>
#include <Userenv.h>
#include "StarterMethodMain.h"
#include "CConfiguration.h"
#include "CShepherdEnvironment.h"
#include "CError.h"
#include "CException.h"
#include "CJob.h"
#include "CProcess.h"
#include "CRegistryKey.h"
#include "CSehTranslator.h"
#include "CShepherd.h"
#include "CTimer.h"
#include "CTrace.h"
using namespace std;
using namespace GridEngine;
int wmain(int argc, LPTSTR argv[], LPTSTR envp[]) {
// enable the SEH to CEH translator
_se_translator_function pOldTranslator;
pOldTranslator = _set_se_translator(TranslateSeToCe);
int nRet = 0;
try {
if (argc == 2) {
CString BasePath = argv[1];
CConfiguration Configuration(BasePath);
CString CommandLine;
// Read Configuration from file
try {
Configuration.ReadConfiguration();
Configuration.ReadOsJobId();
} catch(...) {
CShepherd::WriteExitStatusFile(BasePath, CShepherd::ExitCodes::ReadConfig);
wcerr << L"Could not read configuration" << endl;
return StarterFailed;
}
// Change the current environment
try {
CShepherdEnvironment Environmnet(BasePath);
Environmnet.ReadFromFile();
} catch(...) {
CShepherd::WriteExitStatusFile(BasePath, CShepherd::ExitCodes::ReadConfig);
wcerr << L"Could not read environmnet" << endl;
return StarterFailed;
}
// Shell
CommandLine = CEnvironment::GetExpandedEnvironmentString(Configuration.GetShellPath());
CommandLine += L" ";
// Script
CommandLine += CEnvironment::GetExpandedEnvironmentString(Configuration.GetScriptFile());
// Arguments
for (DWORD dwIndex=0; dwIndex<Configuration.GetJobArgs(); dwIndex++) {
CommandLine += L" ";
CommandLine += Configuration.GetJobArg(dwIndex);
}
// StdOut redirection
CommandLine += L" 1>" + CEnvironment::GetExpandedEnvironmentString(Configuration.GetStdOutPath());
// StdErr redirection
if (Configuration.GetMergeStdErr()) {
CommandLine += L" 2>&1";
} else {
CommandLine += L" 2>" + CEnvironment::GetExpandedEnvironmentString(Configuration.GetStdErrPath());
}
// Start first process of job
try {
CProcess Process(CommandLine, CProcess::Type::INVISIBLE);
Process.Start();
Process.WaitForObject();
return Process.GetExitCode();
} catch (...) {
CShepherd::WriteExitStatusFile(BasePath, CShepherd::ExitCodes::NoShell);
wcerr << L"Could not start job" << endl;
return StarterFailed;
}
}
} catch (CCodineException& ex) {
wcout << ex;
} catch (CException& ex) {
wcout << ex;
} catch (...) {
wcout << L"Caught unknown exception in main()" << endl;
}
// disable translator
_set_se_translator(pOldTranslator);
return 0;
}
|
