|
0
|
1 //
|
|
|
2 // Copyright (C) 2002 Constantin Kaplinsky. All Rights Reserved.
|
|
|
3 //
|
|
|
4 // This is free software; you can redistribute it and/or modify
|
|
|
5 // it under the terms of the GNU General Public License as published by
|
|
|
6 // the Free Software Foundation; either version 2 of the License, or
|
|
|
7 // (at your option) any later version.
|
|
|
8 //
|
|
|
9 // This software is distributed in the hope that it will be useful,
|
|
|
10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
12 // GNU General Public License for more details.
|
|
|
13 //
|
|
|
14 // You should have received a copy of the GNU General Public License
|
|
|
15 // along with this software; if not, write to the Free Software
|
|
|
16 // Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
|
|
|
17 // USA.
|
|
|
18 //
|
|
|
19
|
|
|
20 //
|
|
|
21 // Recording frame. It allows to control recording RFB sessions into
|
|
|
22 // FBS (FrameBuffer Stream) files.
|
|
|
23 //
|
|
|
24
|
|
|
25 import java.io.*;
|
|
|
26 import java.awt.*;
|
|
|
27 import java.awt.event.*;
|
|
|
28
|
|
|
29 class RecordingFrame extends Frame
|
|
|
30 implements WindowListener, ActionListener {
|
|
|
31
|
|
|
32 boolean recording;
|
|
|
33
|
|
|
34 TextField fnameField;
|
|
|
35 Button browseButton;
|
|
|
36
|
|
|
37 Label statusLabel;
|
|
|
38
|
|
|
39 Button recordButton, nextButton, closeButton;
|
|
|
40 VncViewer viewer;
|
|
|
41
|
|
|
42 //
|
|
|
43 // Check if current security manager allows to create a
|
|
|
44 // RecordingFrame object.
|
|
|
45 //
|
|
|
46
|
|
|
47 public static boolean checkSecurity() {
|
|
|
48 SecurityManager security = System.getSecurityManager();
|
|
|
49 if (security != null) {
|
|
|
50 try {
|
|
|
51 security.checkPropertyAccess("user.dir");
|
|
|
52 security.checkPropertyAccess("file.separator");
|
|
|
53 // Work around (rare) checkPropertyAccess bug
|
|
|
54 System.getProperty("user.dir");
|
|
|
55 } catch (SecurityException e) {
|
|
|
56 System.out.println("SecurityManager restricts session recording.");
|
|
|
57 return false;
|
|
|
58 }
|
|
|
59 }
|
|
|
60 return true;
|
|
|
61 }
|
|
|
62
|
|
|
63 //
|
|
|
64 // Constructor.
|
|
|
65 //
|
|
|
66
|
|
|
67 RecordingFrame(VncViewer v) {
|
|
|
68 super("TightVNC Session Recording");
|
|
|
69
|
|
|
70 viewer = v;
|
|
|
71
|
|
|
72 // Determine initial filename for next saved session.
|
|
|
73 // FIXME: Check SecurityManager.
|
|
|
74
|
|
|
75 String fname = nextNewFilename(System.getProperty("user.dir") +
|
|
|
76 System.getProperty("file.separator") +
|
|
|
77 "vncsession.fbs");
|
|
|
78
|
|
|
79 // Construct new panel with file name field and "Browse" button.
|
|
|
80
|
|
|
81 Panel fnamePanel = new Panel();
|
|
|
82 GridBagLayout fnameGridbag = new GridBagLayout();
|
|
|
83 fnamePanel.setLayout(fnameGridbag);
|
|
|
84
|
|
|
85 GridBagConstraints fnameConstraints = new GridBagConstraints();
|
|
|
86 fnameConstraints.gridwidth = GridBagConstraints.RELATIVE;
|
|
|
87 fnameConstraints.fill = GridBagConstraints.BOTH;
|
|
|
88 fnameConstraints.weightx = 4.0;
|
|
|
89
|
|
|
90 fnameField = new TextField(fname, 64);
|
|
|
91 fnameGridbag.setConstraints(fnameField, fnameConstraints);
|
|
|
92 fnamePanel.add(fnameField);
|
|
|
93 fnameField.addActionListener(this);
|
|
|
94
|
|
|
95 fnameConstraints.gridwidth = GridBagConstraints.REMAINDER;
|
|
|
96 fnameConstraints.weightx = 1.0;
|
|
|
97
|
|
|
98 browseButton = new Button("Browse");
|
|
|
99 fnameGridbag.setConstraints(browseButton, fnameConstraints);
|
|
|
100 fnamePanel.add(browseButton);
|
|
|
101 browseButton.addActionListener(this);
|
|
|
102
|
|
|
103 // Construct the frame.
|
|
|
104
|
|
|
105 GridBagLayout gridbag = new GridBagLayout();
|
|
|
106 setLayout(gridbag);
|
|
|
107
|
|
|
108 GridBagConstraints gbc = new GridBagConstraints();
|
|
|
109 gbc.gridwidth = GridBagConstraints.REMAINDER;
|
|
|
110 gbc.fill = GridBagConstraints.BOTH;
|
|
|
111 gbc.weighty = 1.0;
|
|
|
112 gbc.insets = new Insets(10, 0, 0, 0);
|
|
|
113
|
|
|
114 Label helpLabel =
|
|
|
115 new Label("File name to save next recorded session in:", Label.CENTER);
|
|
|
116 gridbag.setConstraints(helpLabel, gbc);
|
|
|
117 add(helpLabel);
|
|
|
118
|
|
|
119 gbc.fill = GridBagConstraints.HORIZONTAL;
|
|
|
120 gbc.weighty = 0.0;
|
|
|
121 gbc.insets = new Insets(0, 0, 0, 0);
|
|
|
122
|
|
|
123 gridbag.setConstraints(fnamePanel, gbc);
|
|
|
124 add(fnamePanel);
|
|
|
125
|
|
|
126 gbc.fill = GridBagConstraints.BOTH;
|
|
|
127 gbc.weighty = 1.0;
|
|
|
128 gbc.insets = new Insets(10, 0, 10, 0);
|
|
|
129
|
|
|
130 statusLabel = new Label("", Label.CENTER);
|
|
|
131 gridbag.setConstraints(statusLabel, gbc);
|
|
|
132 add(statusLabel);
|
|
|
133
|
|
|
134 gbc.fill = GridBagConstraints.HORIZONTAL;
|
|
|
135 gbc.weightx = 1.0;
|
|
|
136 gbc.weighty = 0.0;
|
|
|
137 gbc.gridwidth = 1;
|
|
|
138 gbc.insets = new Insets(0, 0, 0, 0);
|
|
|
139
|
|
|
140 recordButton = new Button("Record");
|
|
|
141 gridbag.setConstraints(recordButton, gbc);
|
|
|
142 add(recordButton);
|
|
|
143 recordButton.addActionListener(this);
|
|
|
144
|
|
|
145 nextButton = new Button("Next file");
|
|
|
146 gridbag.setConstraints(nextButton, gbc);
|
|
|
147 add(nextButton);
|
|
|
148 nextButton.addActionListener(this);
|
|
|
149
|
|
|
150 closeButton = new Button("Close");
|
|
|
151 gridbag.setConstraints(closeButton, gbc);
|
|
|
152 add(closeButton);
|
|
|
153 closeButton.addActionListener(this);
|
|
|
154
|
|
|
155 // Set correct text, font and color for the statusLabel.
|
|
|
156 stopRecording();
|
|
|
157
|
|
|
158 pack();
|
|
|
159
|
|
|
160 addWindowListener(this);
|
|
|
161 }
|
|
|
162
|
|
|
163 //
|
|
|
164 // If the given string ends with ".NNN" where NNN is a decimal
|
|
|
165 // number, increase this number by one. Otherwise, append ".001"
|
|
|
166 // to the given string.
|
|
|
167 //
|
|
|
168
|
|
|
169 protected String nextFilename(String fname) {
|
|
|
170 int len = fname.length();
|
|
|
171 int suffixPos = len;
|
|
|
172 int suffixNum = 1;
|
|
|
173
|
|
|
174 if (len > 4 && fname.charAt(len - 4) == '.') {
|
|
|
175 try {
|
|
|
176 suffixNum = Integer.parseInt(fname.substring(len - 3, len)) + 1;
|
|
|
177 suffixPos = len - 4;
|
|
|
178 } catch (NumberFormatException e) { }
|
|
|
179 }
|
|
|
180
|
|
|
181 char[] zeroes = {'0', '0', '0'};
|
|
|
182 String suffix = String.valueOf(suffixNum);
|
|
|
183 if (suffix.length() < 3) {
|
|
|
184 suffix = new String(zeroes, 0, 3 - suffix.length()) + suffix;
|
|
|
185 }
|
|
|
186
|
|
|
187 return fname.substring(0, suffixPos) + '.' + suffix;
|
|
|
188 }
|
|
|
189
|
|
|
190 //
|
|
|
191 // Find next name of a file which does not exist yet.
|
|
|
192 //
|
|
|
193
|
|
|
194 protected String nextNewFilename(String fname) {
|
|
|
195 String newName = fname;
|
|
|
196 File f;
|
|
|
197 try {
|
|
|
198 do {
|
|
|
199 newName = nextFilename(newName);
|
|
|
200 f = new File(newName);
|
|
|
201 } while (f.exists());
|
|
|
202 } catch (SecurityException e) { }
|
|
|
203
|
|
|
204 return newName;
|
|
|
205 }
|
|
|
206
|
|
|
207 //
|
|
|
208 // Let the user choose a file name showing a FileDialog.
|
|
|
209 //
|
|
|
210
|
|
|
211 protected boolean browseFile() {
|
|
|
212 File currentFile = new File(fnameField.getText());
|
|
|
213
|
|
|
214 FileDialog fd =
|
|
|
215 new FileDialog(this, "Save next session as...", FileDialog.SAVE);
|
|
|
216 fd.setDirectory(currentFile.getParent());
|
|
|
217 fd.setVisible(true);
|
|
|
218 if (fd.getFile() != null) {
|
|
|
219 String newDir = fd.getDirectory();
|
|
|
220 String sep = System.getProperty("file.separator");
|
|
|
221 if (newDir.length() > 0) {
|
|
|
222 if (!sep.equals(newDir.substring(newDir.length() - sep.length())))
|
|
|
223 newDir += sep;
|
|
|
224 }
|
|
|
225 String newFname = newDir + fd.getFile();
|
|
|
226 if (newFname.equals(fnameField.getText())) {
|
|
|
227 fnameField.setText(newFname);
|
|
|
228 return true;
|
|
|
229 }
|
|
|
230 }
|
|
|
231 return false;
|
|
|
232 }
|
|
|
233
|
|
|
234 //
|
|
|
235 // Start recording.
|
|
|
236 //
|
|
|
237
|
|
|
238 public void startRecording() {
|
|
|
239 statusLabel.setText("Status: Recording...");
|
|
|
240 statusLabel.setFont(new Font("Helvetica", Font.BOLD, 12));
|
|
|
241 statusLabel.setForeground(Color.red);
|
|
|
242 recordButton.setLabel("Stop recording");
|
|
|
243
|
|
|
244 recording = true;
|
|
|
245
|
|
|
246 viewer.setRecordingStatus(fnameField.getText());
|
|
|
247 }
|
|
|
248
|
|
|
249 //
|
|
|
250 // Stop recording.
|
|
|
251 //
|
|
|
252
|
|
|
253 public void stopRecording() {
|
|
|
254 statusLabel.setText("Status: Not recording.");
|
|
|
255 statusLabel.setFont(new Font("Helvetica", Font.PLAIN, 12));
|
|
|
256 statusLabel.setForeground(Color.black);
|
|
|
257 recordButton.setLabel("Record");
|
|
|
258
|
|
|
259 recording = false;
|
|
|
260
|
|
|
261 viewer.setRecordingStatus(null);
|
|
|
262 }
|
|
|
263
|
|
|
264 //
|
|
|
265 // Close our window properly.
|
|
|
266 //
|
|
|
267
|
|
|
268 public void windowClosing(WindowEvent evt) {
|
|
|
269 setVisible(false);
|
|
|
270 }
|
|
|
271
|
|
|
272 //
|
|
|
273 // Ignore window events we're not interested in.
|
|
|
274 //
|
|
|
275
|
|
|
276 public void windowActivated(WindowEvent evt) {}
|
|
|
277 public void windowDeactivated (WindowEvent evt) {}
|
|
|
278 public void windowOpened(WindowEvent evt) {}
|
|
|
279 public void windowClosed(WindowEvent evt) {}
|
|
|
280 public void windowIconified(WindowEvent evt) {}
|
|
|
281 public void windowDeiconified(WindowEvent evt) {}
|
|
|
282
|
|
|
283
|
|
|
284 //
|
|
|
285 // Respond to button presses
|
|
|
286 //
|
|
|
287
|
|
|
288 public void actionPerformed(ActionEvent evt) {
|
|
|
289 if (evt.getSource() == browseButton) {
|
|
|
290 if (browseFile() && recording)
|
|
|
291 startRecording();
|
|
|
292
|
|
|
293 } else if (evt.getSource() == recordButton) {
|
|
|
294 if (!recording) {
|
|
|
295 startRecording();
|
|
|
296 } else {
|
|
|
297 stopRecording();
|
|
|
298 fnameField.setText(nextNewFilename(fnameField.getText()));
|
|
|
299 }
|
|
|
300
|
|
|
301 } else if (evt.getSource() == nextButton) {
|
|
|
302 fnameField.setText(nextNewFilename(fnameField.getText()));
|
|
|
303 if (recording)
|
|
|
304 startRecording();
|
|
|
305
|
|
|
306 } else if (evt.getSource() == closeButton) {
|
|
|
307 setVisible(false);
|
|
|
308
|
|
|
309 }
|
|
|
310 }
|
|
|
311 }
|
