#include <Alert.h>
#include <DataIO.h>
#include <Message.h>
#include <Directory.h>
#include <netdb.h>
#include <pwd.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <iomanip>
#include <sstream>
#include "LpsClient.h"
#include "LprSetupDlg.h"
#include "LprTransport.h"
#include "LprDefs.h"
#include "DbgMsg.h"
using namespace std;
LprTransport::LprTransport(BMessage *msg)
:
BDataIO()
{
fServer[0] = '\0';
fQueue[0] = '\0';
fFile[0] = '\0';
fUser[0] = '\0';
fJobId = 0;
fError = false;
struct passwd *pwd = getpwuid(geteuid());
if (pwd != NULL && pwd->pw_name != NULL && pwd->pw_name[0])
strcpy(fUser, pwd->pw_name);
else
strcpy(fUser, "baron");
DUMP_BMESSAGE(msg);
const char *spool_path = msg->FindString(SPOOL_PATH);
if (spool_path && *spool_path) {
BDirectory dir(spool_path);
DUMP_BDIRECTORY(&dir);
dir.ReadAttr(LPR_SERVER_NAME, B_STRING_TYPE, 0, fServer, sizeof(fServer));
if (fServer[0] == '\0') {
LprSetupDlg *dlg = new LprSetupDlg(&dir);
if (dlg->Go() == B_ERROR) {
fError = true;
return;
}
}
dir.ReadAttr(LPR_SERVER_NAME, B_STRING_TYPE, 0, fServer, sizeof(fServer));
dir.ReadAttr(LPR_QUEUE_NAME, B_STRING_TYPE, 0, fQueue, sizeof(fQueue));
dir.ReadAttr(LPR_JOB_ID, B_INT32_TYPE, 0, &fJobId, sizeof(fJobId));
fJobId++;
if (fJobId > 255) {
fJobId = 1;
}
dir.WriteAttr(LPR_JOB_ID, B_INT32_TYPE, 0, &fJobId, sizeof(fJobId));
sprintf(fFile, "%s/%s@ipp.%" B_PRId32, spool_path, fUser, fJobId);
fStream.open(fFile, ios::in | ios::out | ios::binary | ios::trunc);
if (fStream.good()) {
DBGMSG(("spool_file: %s\n", fFile));
return;
}
}
fError = true;
}
LprTransport::~LprTransport()
{
if (!fError)
_SendFile();
if (fFile[0] != '\0')
unlink(fFile);
}
void
LprTransport::_SendFile()
{
char hostname[128];
if (gethostname(hostname, sizeof(hostname)) != B_OK)
strcpy(hostname, "localhost");
ostringstream cfname;
cfname << "cfA" << setw(3) << setfill('0') << fJobId << hostname;
ostringstream dfname;
dfname << "dfA" << setw(3) << setfill('0') << fJobId << hostname;
ostringstream cf;
cf << 'H' << hostname << '\n';
cf << 'P' << fUser << '\n';
cf << 'l' << dfname.str() << '\n';
cf << 'U' << dfname.str() << '\n';
long cfsize = cf.str().length();
long dfsize = fStream.tellg();
fStream.seekg(0, ios::beg);
try {
LpsClient lpr(fServer);
lpr.connect();
lpr.receiveJob(fQueue);
lpr.receiveControlFile(cfsize, cfname.str().c_str());
lpr.transferData(cf.str().c_str(), cfsize);
lpr.endTransfer();
lpr.receiveDataFile(dfsize, dfname.str().c_str());
lpr.transferData(fStream, dfsize);
lpr.endTransfer();
}
catch (LPSException &err) {
DBGMSG(("error: %s\n", err.what()));
BAlert *alert = new BAlert("", err.what(), "OK");
alert->SetFlags(alert->Flags() | B_CLOSE_ON_ESCAPE);
alert->Go();
}
}
ssize_t
LprTransport::Read(void *, size_t)
{
return 0;
}
ssize_t
LprTransport::Write(const void *buffer, size_t size)
{
if (!fStream.write((char *)buffer, size)) {
fError = true;
return 0;
}
return size;
}
